[
  {
    "stage": "pretrain",
    "step": 1,
    "loss": 80.38571166992188,
    "text_loss": 80.38571166992188,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 20,
    "loss": 2.5035173892974854,
    "text_loss": 2.5035173892974854,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 40,
    "loss": 1.3843837976455688,
    "text_loss": 1.3843837976455688,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 60,
    "loss": 0.8159888386726379,
    "text_loss": 0.8159888386726379,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 80,
    "loss": 0.19992661476135254,
    "text_loss": 0.19992661476135254,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 100,
    "loss": 0.05571526661515236,
    "text_loss": 0.05571526661515236,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 120,
    "loss": 0.005689455661922693,
    "text_loss": 0.005689455661922693,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 140,
    "loss": 0.00010565178672550246,
    "text_loss": 0.00010565178672550246,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 160,
    "loss": 4.098461431567557e-05,
    "text_loss": 4.098461431567557e-05,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 180,
    "loss": 2.716351991693955e-05,
    "text_loss": 2.716351991693955e-05,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 200,
    "loss": 2.2414897102862597e-05,
    "text_loss": 2.2414897102862597e-05,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 220,
    "loss": 1.9143404642818496e-05,
    "text_loss": 1.9143404642818496e-05,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 240,
    "loss": 1.6641104593873024e-05,
    "text_loss": 1.6641104593873024e-05,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 260,
    "loss": 1.4626367374148685e-05,
    "text_loss": 1.4626367374148685e-05,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 280,
    "loss": 1.2977556252735667e-05,
    "text_loss": 1.2977556252735667e-05,
    "flow_loss": null
  },
  {
    "stage": "pretrain",
    "step": 300,
    "loss": 1.16078899736749e-05,
    "text_loss": 1.16078899736749e-05,
    "flow_loss": null
  },
  {
    "stage": "posttrain",
    "step": 1,
    "loss": 3.8964109420776367,
    "text_loss": 1.1545746929186862e-05,
    "flow_loss": 1.94819974899292
  },
  {
    "stage": "posttrain",
    "step": 20,
    "loss": 4.304055690765381,
    "text_loss": 2.3414058887283318e-05,
    "flow_loss": 2.1520161628723145
  },
  {
    "stage": "posttrain",
    "step": 40,
    "loss": 2.925170660018921,
    "text_loss": 2.142304765584413e-06,
    "flow_loss": 1.4625842571258545
  },
  {
    "stage": "posttrain",
    "step": 60,
    "loss": 1.7996485233306885,
    "text_loss": 5.1970469030493405e-06,
    "flow_loss": 0.8998216390609741
  },
  {
    "stage": "posttrain",
    "step": 80,
    "loss": 2.0647871494293213,
    "text_loss": 3.251374073443003e-05,
    "flow_loss": 1.0323773622512817
  },
  {
    "stage": "posttrain",
    "step": 100,
    "loss": 2.3402059078216553,
    "text_loss": 2.50748803409806e-06,
    "flow_loss": 1.1701016426086426
  },
  {
    "stage": "posttrain",
    "step": 120,
    "loss": 1.4503896236419678,
    "text_loss": 6.828942105130409e-07,
    "flow_loss": 0.7251944541931152
  },
  {
    "stage": "posttrain",
    "step": 140,
    "loss": 0.23238907754421234,
    "text_loss": 1.3709052382182563e-07,
    "flow_loss": 0.1161944717168808
  },
  {
    "stage": "posttrain",
    "step": 160,
    "loss": 0.46604833006858826,
    "text_loss": 2.1542784622852196e-07,
    "flow_loss": 0.23302406072616577
  },
  {
    "stage": "posttrain",
    "step": 180,
    "loss": 0.169603168964386,
    "text_loss": 2.418236704215815e-07,
    "flow_loss": 0.08480146527290344
  },
  {
    "stage": "posttrain",
    "step": 200,
    "loss": 0.7555881142616272,
    "text_loss": 8.855541011598689e-08,
    "flow_loss": 0.3777940273284912
  },
  {
    "stage": "posttrain",
    "step": 220,
    "loss": 0.10211063176393509,
    "text_loss": 1.3623892414216243e-07,
    "flow_loss": 0.05105524882674217
  },
  {
    "stage": "posttrain",
    "step": 240,
    "loss": 0.4649242162704468,
    "text_loss": 2.2819931189133058e-07,
    "flow_loss": 0.23246198892593384
  },
  {
    "stage": "posttrain",
    "step": 260,
    "loss": 0.280495822429657,
    "text_loss": 1.3198149417803506e-07,
    "flow_loss": 0.14024785161018372
  },
  {
    "stage": "posttrain",
    "step": 280,
    "loss": 0.5054064989089966,
    "text_loss": 9.451582627661992e-08,
    "flow_loss": 0.2527031898498535
  },
  {
    "stage": "posttrain",
    "step": 300,
    "loss": 0.19181671738624573,
    "text_loss": 4.453279984772962e-07,
    "flow_loss": 0.09590813517570496
  },
  {
    "stage": "posttrain",
    "step": 320,
    "loss": 0.20589734613895416,
    "text_loss": 2.8524956974251836e-07,
    "flow_loss": 0.10294853150844574
  },
  {
    "stage": "posttrain",
    "step": 340,
    "loss": 0.09927964955568314,
    "text_loss": 1.0132762895409542e-07,
    "flow_loss": 0.04963977262377739
  },
  {
    "stage": "posttrain",
    "step": 360,
    "loss": 0.2213846743106842,
    "text_loss": 1.2942695093443035e-07,
    "flow_loss": 0.11069227010011673
  },
  {
    "stage": "posttrain",
    "step": 380,
    "loss": 0.9335335493087769,
    "text_loss": 1.3964488232431904e-07,
    "flow_loss": 0.46676671504974365
  },
  {
    "stage": "posttrain",
    "step": 400,
    "loss": 0.12127648293972015,
    "text_loss": 7.578287153364727e-08,
    "flow_loss": 0.06063820421695709
  },
  {
    "stage": "posttrain",
    "step": 420,
    "loss": 0.1503632515668869,
    "text_loss": 4.0871746875836834e-08,
    "flow_loss": 0.07518160343170166
  },
  {
    "stage": "posttrain",
    "step": 440,
    "loss": 0.15399529039859772,
    "text_loss": 2.0435869885204738e-08,
    "flow_loss": 0.07699763774871826
  },
  {
    "stage": "posttrain",
    "step": 460,
    "loss": 0.18908466398715973,
    "text_loss": 4.172323286866231e-08,
    "flow_loss": 0.09454230964183807
  },
  {
    "stage": "posttrain",
    "step": 480,
    "loss": 0.2329331636428833,
    "text_loss": 2.8099325533048614e-08,
    "flow_loss": 0.11646656692028046
  },
  {
    "stage": "posttrain",
    "step": 500,
    "loss": 0.1929057091474533,
    "text_loss": 2.784368575703411e-07,
    "flow_loss": 0.09645271301269531
  }
]