[
  {
    "stage": "base",
    "step": 1,
    "loss": 70.63429260253906,
    "text_loss": 68.23114776611328,
    "flow_loss": 1.2015743255615234
  },
  {
    "stage": "base",
    "step": 20,
    "loss": 3.8942408561706543,
    "text_loss": 1.7884641885757446,
    "flow_loss": 1.0528883934020996
  },
  {
    "stage": "base",
    "step": 40,
    "loss": 3.1883139610290527,
    "text_loss": 0.8112223148345947,
    "flow_loss": 1.188545823097229
  },
  {
    "stage": "base",
    "step": 60,
    "loss": 2.8662467002868652,
    "text_loss": 0.5043408274650574,
    "flow_loss": 1.1809529066085815
  },
  {
    "stage": "base",
    "step": 80,
    "loss": 2.9415433406829834,
    "text_loss": 0.2767218053340912,
    "flow_loss": 1.3324108123779297
  },
  {
    "stage": "base",
    "step": 100,
    "loss": 1.23435640335083,
    "text_loss": 0.013344191014766693,
    "flow_loss": 0.6105061173439026
  },
  {
    "stage": "base",
    "step": 120,
    "loss": 0.6844875812530518,
    "text_loss": 0.1804233342409134,
    "flow_loss": 0.2520321309566498
  },
  {
    "stage": "base",
    "step": 140,
    "loss": 1.1563541889190674,
    "text_loss": 6.113824201747775e-05,
    "flow_loss": 0.5781465172767639
  },
  {
    "stage": "base",
    "step": 160,
    "loss": 0.49136143922805786,
    "text_loss": 0.00014380548964254558,
    "flow_loss": 0.24560882151126862
  },
  {
    "stage": "base",
    "step": 180,
    "loss": 0.5463516712188721,
    "text_loss": 1.483415599068394e-05,
    "flow_loss": 0.2731684148311615
  },
  {
    "stage": "base",
    "step": 200,
    "loss": 0.21821065247058868,
    "text_loss": 4.533280844043475e-06,
    "flow_loss": 0.10910306125879288
  },
  {
    "stage": "base",
    "step": 220,
    "loss": 0.2135496884584427,
    "text_loss": 3.412749947528937e-06,
    "flow_loss": 0.10677313804626465
  },
  {
    "stage": "base",
    "step": 240,
    "loss": 0.15570451319217682,
    "text_loss": 3.027882712558494e-06,
    "flow_loss": 0.07785074412822723
  },
  {
    "stage": "base",
    "step": 260,
    "loss": 0.17475822567939758,
    "text_loss": 2.816716460074531e-06,
    "flow_loss": 0.08737770467996597
  },
  {
    "stage": "base",
    "step": 280,
    "loss": 0.12843455374240875,
    "text_loss": 2.6055499802168924e-06,
    "flow_loss": 0.06421597301959991
  },
  {
    "stage": "base",
    "step": 300,
    "loss": 0.3167646825313568,
    "text_loss": 2.387570930295624e-06,
    "flow_loss": 0.1583811491727829
  },
  {
    "stage": "value",
    "step": 1,
    "loss": 5.466859817504883,
    "prediction": -0.5042269229888916,
    "target": 0.0
  },
  {
    "stage": "value",
    "step": 20,
    "loss": 3.274851083755493,
    "prediction": -0.3775767683982849,
    "target": -0.005210627801716328
  },
  {
    "stage": "value",
    "step": 40,
    "loss": 2.3394997119903564,
    "prediction": -0.2888708710670471,
    "target": -0.00456627644598484
  },
  {
    "stage": "value",
    "step": 60,
    "loss": 0.9264469146728516,
    "prediction": -0.2154604196548462,
    "target": -0.0022692831698805094
  },
  {
    "stage": "value",
    "step": 80,
    "loss": 0.756015956401825,
    "prediction": -0.19315782189369202,
    "target": -0.00012316892389208078
  },
  {
    "stage": "value",
    "step": 100,
    "loss": 0.6257156729698181,
    "prediction": -0.18343542516231537,
    "target": 0.0
  },
  {
    "stage": "value",
    "step": 120,
    "loss": 1.3138372898101807,
    "prediction": -0.37475356459617615,
    "target": -0.005220414604991674
  },
  {
    "stage": "value",
    "step": 140,
    "loss": 1.4211702346801758,
    "prediction": -0.1804044544696808,
    "target": -0.0051329550333321095
  },
  {
    "stage": "value",
    "step": 160,
    "loss": 1.448256254196167,
    "prediction": -0.2506459355354309,
    "target": -0.003942344803363085
  },
  {
    "stage": "value",
    "step": 180,
    "loss": 0.33143094182014465,
    "prediction": -0.08220178633928299,
    "target": -0.0013850623508915305
  },
  {
    "stage": "value",
    "step": 200,
    "loss": 0.2722547948360443,
    "prediction": -0.06349469721317291,
    "target": 0.0
  },
  {
    "stage": "value",
    "step": 220,
    "loss": 0.22779540717601776,
    "prediction": -0.05417971312999725,
    "target": 0.0
  },
  {
    "stage": "value",
    "step": 240,
    "loss": 1.5308982133865356,
    "prediction": -0.3449951708316803,
    "target": -0.005219103302806616
  },
  {
    "stage": "value",
    "step": 260,
    "loss": 1.1563409566879272,
    "prediction": -0.2026708871126175,
    "target": -0.004918968304991722
  },
  {
    "stage": "value",
    "step": 280,
    "loss": 1.2716314792633057,
    "prediction": -0.07276482880115509,
    "target": -0.003156546037644148
  },
  {
    "stage": "value",
    "step": 300,
    "loss": 0.3126322329044342,
    "prediction": -0.07691694051027298,
    "target": -0.0006413022056221962
  },
  {
    "stage": "value",
    "step": 320,
    "loss": 0.1044526994228363,
    "prediction": -0.03439018875360489,
    "target": 0.0
  },
  {
    "stage": "value",
    "step": 340,
    "loss": 0.1302744299173355,
    "prediction": -0.03440919518470764,
    "target": 0.0
  },
  {
    "stage": "value",
    "step": 360,
    "loss": 1.5437527894973755,
    "prediction": -0.24599501490592957,
    "target": -0.005210627801716328
  },
  {
    "stage": "value",
    "step": 380,
    "loss": 1.4080818891525269,
    "prediction": -0.3180582821369171,
    "target": -0.00456627644598484
  },
  {
    "stage": "value",
    "step": 400,
    "loss": 0.8151843547821045,
    "prediction": -0.07982967793941498,
    "target": -0.0022692831698805094
  },
  {
    "stage": "mixed",
    "step": 1,
    "loss": 0.05100901424884796,
    "text_loss": 2.3773534394422313e-06,
    "flow_loss": 0.02550331875681877
  },
  {
    "stage": "mixed",
    "step": 20,
    "loss": 3.583022117614746,
    "text_loss": 2.2782084941864014,
    "flow_loss": 0.6524068117141724
  },
  {
    "stage": "mixed",
    "step": 40,
    "loss": 2.2450661659240723,
    "text_loss": 1.1263315677642822,
    "flow_loss": 0.5593672394752502
  },
  {
    "stage": "mixed",
    "step": 60,
    "loss": 2.665435552597046,
    "text_loss": 0.9095665812492371,
    "flow_loss": 0.8779345154762268
  },
  {
    "stage": "mixed",
    "step": 80,
    "loss": 0.9744815826416016,
    "text_loss": 0.7057558298110962,
    "flow_loss": 0.1343628615140915
  },
  {
    "stage": "mixed",
    "step": 100,
    "loss": 2.257481575012207,
    "text_loss": 0.7136633992195129,
    "flow_loss": 0.7719091176986694
  },
  {
    "stage": "mixed",
    "step": 120,
    "loss": 1.557675838470459,
    "text_loss": 0.6338224411010742,
    "flow_loss": 0.46192666888237
  },
  {
    "stage": "mixed",
    "step": 140,
    "loss": 1.2216718196868896,
    "text_loss": 0.7448893785476685,
    "flow_loss": 0.2383912056684494
  },
  {
    "stage": "mixed",
    "step": 160,
    "loss": 0.5911489725112915,
    "text_loss": 0.27954256534576416,
    "flow_loss": 0.15580320358276367
  },
  {
    "stage": "mixed",
    "step": 180,
    "loss": 2.7055318355560303,
    "text_loss": 0.25718608498573303,
    "flow_loss": 1.224172830581665
  },
  {
    "stage": "mixed",
    "step": 200,
    "loss": 1.3802378177642822,
    "text_loss": 0.32982689142227173,
    "flow_loss": 0.5252054929733276
  },
  {
    "stage": "mixed",
    "step": 220,
    "loss": 1.1629283428192139,
    "text_loss": 0.26602569222450256,
    "flow_loss": 0.44845134019851685
  },
  {
    "stage": "mixed",
    "step": 240,
    "loss": 0.5630282759666443,
    "text_loss": 0.2572731077671051,
    "flow_loss": 0.1528775840997696
  },
  {
    "stage": "mixed",
    "step": 260,
    "loss": 1.1026716232299805,
    "text_loss": 0.15438926219940186,
    "flow_loss": 0.4741412103176117
  },
  {
    "stage": "mixed",
    "step": 280,
    "loss": 1.9438565969467163,
    "text_loss": 0.18404552340507507,
    "flow_loss": 0.8799055218696594
  },
  {
    "stage": "mixed",
    "step": 300,
    "loss": 0.8524055480957031,
    "text_loss": 0.1450427919626236,
    "flow_loss": 0.35368138551712036
  },
  {
    "stage": "recap",
    "step": 1,
    "loss": 10.587621688842773,
    "text_loss": 9.856688499450684,
    "flow_loss": 0.3654666244983673
  },
  {
    "stage": "recap",
    "step": 20,
    "loss": 3.289060354232788,
    "text_loss": 2.240044355392456,
    "flow_loss": 0.524507999420166
  },
  {
    "stage": "recap",
    "step": 40,
    "loss": 2.791560173034668,
    "text_loss": 1.4383518695831299,
    "flow_loss": 0.676604151725769
  },
  {
    "stage": "recap",
    "step": 60,
    "loss": 1.8424029350280762,
    "text_loss": 1.1395618915557861,
    "flow_loss": 0.351420521736145
  },
  {
    "stage": "recap",
    "step": 80,
    "loss": 1.8679100275039673,
    "text_loss": 0.9174721837043762,
    "flow_loss": 0.47521892189979553
  },
  {
    "stage": "recap",
    "step": 100,
    "loss": 2.929778814315796,
    "text_loss": 0.9240415096282959,
    "flow_loss": 1.00286865234375
  },
  {
    "stage": "recap",
    "step": 120,
    "loss": 1.4325169324874878,
    "text_loss": 0.7645139694213867,
    "flow_loss": 0.33400148153305054
  },
  {
    "stage": "recap",
    "step": 140,
    "loss": 1.4750593900680542,
    "text_loss": 0.7536309957504272,
    "flow_loss": 0.3607141971588135
  },
  {
    "stage": "recap",
    "step": 160,
    "loss": 1.5970324277877808,
    "text_loss": 1.017958164215088,
    "flow_loss": 0.28953713178634644
  },
  {
    "stage": "recap",
    "step": 180,
    "loss": 1.6522973775863647,
    "text_loss": 0.922364354133606,
    "flow_loss": 0.3649665117263794
  },
  {
    "stage": "recap",
    "step": 200,
    "loss": 1.3916192054748535,
    "text_loss": 0.9831637740135193,
    "flow_loss": 0.2042277306318283
  },
  {
    "stage": "recap",
    "step": 220,
    "loss": 1.138975739479065,
    "text_loss": 0.8730899691581726,
    "flow_loss": 0.13294290006160736
  },
  {
    "stage": "recap",
    "step": 240,
    "loss": 1.502185344696045,
    "text_loss": 0.8015594482421875,
    "flow_loss": 0.3503129780292511
  },
  {
    "stage": "recap",
    "step": 260,
    "loss": 0.8756294846534729,
    "text_loss": 0.31519001722335815,
    "flow_loss": 0.2802197337150574
  },
  {
    "stage": "recap",
    "step": 280,
    "loss": 0.779049277305603,
    "text_loss": 0.3690524101257324,
    "flow_loss": 0.2049984186887741
  },
  {
    "stage": "recap",
    "step": 300,
    "loss": 0.7172976732254028,
    "text_loss": 0.23720531165599823,
    "flow_loss": 0.2400461882352829
  },
  {
    "stage": "recap",
    "step": 320,
    "loss": 2.9586782455444336,
    "text_loss": 0.7825477719306946,
    "flow_loss": 1.088065266609192
  },
  {
    "stage": "recap",
    "step": 340,
    "loss": 1.745682954788208,
    "text_loss": 0.43472984433174133,
    "flow_loss": 0.6554765701293945
  },
  {
    "stage": "recap",
    "step": 360,
    "loss": 1.5331687927246094,
    "text_loss": 0.7225565314292908,
    "flow_loss": 0.4053061604499817
  },
  {
    "stage": "recap",
    "step": 380,
    "loss": 1.6411525011062622,
    "text_loss": 0.14736497402191162,
    "flow_loss": 0.7468937635421753
  },
  {
    "stage": "recap",
    "step": 400,
    "loss": 0.5169777870178223,
    "text_loss": 0.24374954402446747,
    "flow_loss": 0.1366141140460968
  },
  {
    "stage": "recap",
    "step": 420,
    "loss": 0.6074326038360596,
    "text_loss": 0.11582576483488083,
    "flow_loss": 0.24580343067646027
  },
  {
    "stage": "recap",
    "step": 440,
    "loss": 0.7017235159873962,
    "text_loss": 0.3753448724746704,
    "flow_loss": 0.16318932175636292
  },
  {
    "stage": "recap",
    "step": 460,
    "loss": 1.029306411743164,
    "text_loss": 0.2204148769378662,
    "flow_loss": 0.40444573760032654
  },
  {
    "stage": "recap",
    "step": 480,
    "loss": 0.9227437376976013,
    "text_loss": 0.21806900203227997,
    "flow_loss": 0.3523373603820801
  },
  {
    "stage": "recap",
    "step": 500,
    "loss": 0.7456694841384888,
    "text_loss": 0.3062073886394501,
    "flow_loss": 0.21973104774951935
  },
  {
    "stage": "recap",
    "step": 520,
    "loss": 1.2862324714660645,
    "text_loss": 0.6478312015533447,
    "flow_loss": 0.3192006051540375
  },
  {
    "stage": "recap",
    "step": 540,
    "loss": 0.7051770687103271,
    "text_loss": 0.23054330050945282,
    "flow_loss": 0.23731689155101776
  },
  {
    "stage": "recap",
    "step": 560,
    "loss": 1.9938247203826904,
    "text_loss": 1.1214075088500977,
    "flow_loss": 0.4362086057662964
  },
  {
    "stage": "recap",
    "step": 580,
    "loss": 1.6423680782318115,
    "text_loss": 1.1029378175735474,
    "flow_loss": 0.2697151303291321
  },
  {
    "stage": "recap",
    "step": 600,
    "loss": 1.3713027238845825,
    "text_loss": 0.1559465080499649,
    "flow_loss": 0.6076781153678894
  },
  {
    "stage": "recap",
    "step": 620,
    "loss": 0.3619869351387024,
    "text_loss": 0.06553713977336884,
    "flow_loss": 0.14822489023208618
  },
  {
    "stage": "recap",
    "step": 640,
    "loss": 0.5116170048713684,
    "text_loss": 0.07361899316310883,
    "flow_loss": 0.2189989984035492
  },
  {
    "stage": "recap",
    "step": 660,
    "loss": 0.8094859719276428,
    "text_loss": 0.07687854766845703,
    "flow_loss": 0.3663037121295929
  },
  {
    "stage": "recap",
    "step": 680,
    "loss": 1.4220974445343018,
    "text_loss": 0.13382284343242645,
    "flow_loss": 0.6441373229026794
  },
  {
    "stage": "recap",
    "step": 700,
    "loss": 0.40469539165496826,
    "text_loss": 0.04384676367044449,
    "flow_loss": 0.18042431771755219
  }
]