{
  "summary": {
    "stage": "stage2",
    "synthetic_smoke": false,
    "steps": 80,
    "model": {
      "parameters": 605371425,
      "trainable_parameters": 605371425
    },
    "last": {
      "step": 80,
      "loss": 8.052010536193848,
      "flow_mse": 7.989875793457031,
      "art_l2": 0.062134984880685806,
      "t_mean": 0.08626709133386612
    }
  },
  "logs": [
    {
      "step": 1,
      "loss": 4.555010795593262,
      "flow_mse": 4.468900680541992,
      "art_l2": 0.08611014485359192,
      "t_mean": 0.5347283482551575
    },
    {
      "step": 2,
      "loss": 8.082441329956055,
      "flow_mse": 8.016830444335938,
      "art_l2": 0.06561049073934555,
      "t_mean": 0.3323965072631836
    },
    {
      "step": 3,
      "loss": 5.238214492797852,
      "flow_mse": 5.118548393249512,
      "art_l2": 0.1196659728884697,
      "t_mean": 0.8121371269226074
    },
    {
      "step": 4,
      "loss": 4.3173136711120605,
      "flow_mse": 4.255227565765381,
      "art_l2": 0.06208610534667969,
      "t_mean": 0.686140239238739
    },
    {
      "step": 5,
      "loss": 6.124124526977539,
      "flow_mse": 6.057380199432373,
      "art_l2": 0.0667441263794899,
      "t_mean": 0.728419840335846
    },
    {
      "step": 6,
      "loss": 3.70115327835083,
      "flow_mse": 3.602912664413452,
      "art_l2": 0.09824050962924957,
      "t_mean": 0.6403265595436096
    },
    {
      "step": 7,
      "loss": 4.561351776123047,
      "flow_mse": 4.482015132904053,
      "art_l2": 0.07933680713176727,
      "t_mean": 0.5700072646141052
    },
    {
      "step": 8,
      "loss": 5.916510105133057,
      "flow_mse": 5.827247142791748,
      "art_l2": 0.08926297724246979,
      "t_mean": 0.655311644077301
    },
    {
      "step": 9,
      "loss": 4.786649227142334,
      "flow_mse": 4.732999801635742,
      "art_l2": 0.05364961549639702,
      "t_mean": 0.6019466519355774
    },
    {
      "step": 10,
      "loss": 3.9152398109436035,
      "flow_mse": 3.8640236854553223,
      "art_l2": 0.05121621862053871,
      "t_mean": 0.7785599827766418
    },
    {
      "step": 11,
      "loss": 7.887813568115234,
      "flow_mse": 7.814604759216309,
      "art_l2": 0.07320895045995712,
      "t_mean": 0.6584867835044861
    },
    {
      "step": 12,
      "loss": 5.762132167816162,
      "flow_mse": 5.6032915115356445,
      "art_l2": 0.15884068608283997,
      "t_mean": 0.2739311456680298
    },
    {
      "step": 13,
      "loss": 2.1992640495300293,
      "flow_mse": 2.105966091156006,
      "art_l2": 0.09329804033041,
      "t_mean": 0.6072238683700562
    },
    {
      "step": 14,
      "loss": 3.140270709991455,
      "flow_mse": 2.9471776485443115,
      "art_l2": 0.19309300184249878,
      "t_mean": 0.6395391821861267
    },
    {
      "step": 15,
      "loss": 6.881434440612793,
      "flow_mse": 6.796713829040527,
      "art_l2": 0.08472083508968353,
      "t_mean": 0.43723514676094055
    },
    {
      "step": 16,
      "loss": 3.5250988006591797,
      "flow_mse": 3.4759371280670166,
      "art_l2": 0.04916170611977577,
      "t_mean": 0.6621525883674622
    },
    {
      "step": 17,
      "loss": 4.099492073059082,
      "flow_mse": 4.049228191375732,
      "art_l2": 0.050263892859220505,
      "t_mean": 0.5240513682365417
    },
    {
      "step": 18,
      "loss": 5.658870220184326,
      "flow_mse": 5.52504825592041,
      "art_l2": 0.13382190465927124,
      "t_mean": 0.5368333458900452
    },
    {
      "step": 19,
      "loss": 9.602043151855469,
      "flow_mse": 9.52139949798584,
      "art_l2": 0.08064349740743637,
      "t_mean": 0.2161000519990921
    },
    {
      "step": 20,
      "loss": 2.9768917560577393,
      "flow_mse": 2.865424871444702,
      "art_l2": 0.11146692931652069,
      "t_mean": 0.5189581513404846
    },
    {
      "step": 21,
      "loss": 4.881039142608643,
      "flow_mse": 4.7817769050598145,
      "art_l2": 0.09926208853721619,
      "t_mean": 0.11379662156105042
    },
    {
      "step": 22,
      "loss": 4.0256853103637695,
      "flow_mse": 3.965487241744995,
      "art_l2": 0.060197893530130386,
      "t_mean": 0.4425402879714966
    },
    {
      "step": 23,
      "loss": 4.597318649291992,
      "flow_mse": 4.480355739593506,
      "art_l2": 0.11696302890777588,
      "t_mean": 0.20127582550048828
    },
    {
      "step": 24,
      "loss": 2.815336227416992,
      "flow_mse": 2.6951727867126465,
      "art_l2": 0.12016341090202332,
      "t_mean": 0.7069639563560486
    },
    {
      "step": 25,
      "loss": 2.6848018169403076,
      "flow_mse": 2.583383321762085,
      "art_l2": 0.10141842067241669,
      "t_mean": 0.6600659489631653
    },
    {
      "step": 26,
      "loss": 4.562182903289795,
      "flow_mse": 4.4232964515686035,
      "art_l2": 0.13888661563396454,
      "t_mean": 0.4758807420730591
    },
    {
      "step": 27,
      "loss": 3.773374319076538,
      "flow_mse": 3.7105894088745117,
      "art_l2": 0.06278500705957413,
      "t_mean": 0.9104493260383606
    },
    {
      "step": 28,
      "loss": 4.5046210289001465,
      "flow_mse": 4.3704094886779785,
      "art_l2": 0.13421154022216797,
      "t_mean": 0.33697742223739624
    },
    {
      "step": 29,
      "loss": 8.195755004882812,
      "flow_mse": 8.11143970489502,
      "art_l2": 0.08431512117385864,
      "t_mean": 0.324869304895401
    },
    {
      "step": 30,
      "loss": 3.397721529006958,
      "flow_mse": 3.2944910526275635,
      "art_l2": 0.10323046147823334,
      "t_mean": 0.2462034672498703
    },
    {
      "step": 31,
      "loss": 5.644656658172607,
      "flow_mse": 5.528888702392578,
      "art_l2": 0.11576806753873825,
      "t_mean": 0.3077169954776764
    },
    {
      "step": 32,
      "loss": 3.7508933544158936,
      "flow_mse": 3.6254053115844727,
      "art_l2": 0.12548798322677612,
      "t_mean": 0.2582360506057739
    },
    {
      "step": 33,
      "loss": 4.49017858505249,
      "flow_mse": 4.3665771484375,
      "art_l2": 0.12360142916440964,
      "t_mean": 0.5150696039199829
    },
    {
      "step": 34,
      "loss": 4.182536602020264,
      "flow_mse": 3.9879045486450195,
      "art_l2": 0.19463200867176056,
      "t_mean": 0.3112204372882843
    },
    {
      "step": 35,
      "loss": 3.290544033050537,
      "flow_mse": 3.160871982574463,
      "art_l2": 0.129672110080719,
      "t_mean": 0.639766275882721
    },
    {
      "step": 36,
      "loss": 3.8543291091918945,
      "flow_mse": 3.7053794860839844,
      "art_l2": 0.14894957840442657,
      "t_mean": 0.6604555249214172
    },
    {
      "step": 37,
      "loss": 4.848496437072754,
      "flow_mse": 4.7336835861206055,
      "art_l2": 0.11481265723705292,
      "t_mean": 0.756874680519104
    },
    {
      "step": 38,
      "loss": 4.920182228088379,
      "flow_mse": 4.772139549255371,
      "art_l2": 0.14804257452487946,
      "t_mean": 0.5363696813583374
    },
    {
      "step": 39,
      "loss": 3.0050387382507324,
      "flow_mse": 2.843062162399292,
      "art_l2": 0.16197659075260162,
      "t_mean": 0.580784797668457
    },
    {
      "step": 40,
      "loss": 5.2633442878723145,
      "flow_mse": 5.194778919219971,
      "art_l2": 0.06856553256511688,
      "t_mean": 0.38077911734580994
    },
    {
      "step": 41,
      "loss": 3.5400869846343994,
      "flow_mse": 3.4967432022094727,
      "art_l2": 0.04334376007318497,
      "t_mean": 0.6592270731925964
    },
    {
      "step": 42,
      "loss": 2.7601735591888428,
      "flow_mse": 2.700631618499756,
      "art_l2": 0.05954183638095856,
      "t_mean": 0.8132764101028442
    },
    {
      "step": 43,
      "loss": 5.247016906738281,
      "flow_mse": 5.192378997802734,
      "art_l2": 0.05463814362883568,
      "t_mean": 0.7951807379722595
    },
    {
      "step": 44,
      "loss": 3.2813501358032227,
      "flow_mse": 3.195319414138794,
      "art_l2": 0.08603078871965408,
      "t_mean": 0.5339460968971252
    },
    {
      "step": 45,
      "loss": 3.4666907787323,
      "flow_mse": 3.2917640209198,
      "art_l2": 0.1749267876148224,
      "t_mean": 0.5418036580085754
    },
    {
      "step": 46,
      "loss": 9.8698148727417,
      "flow_mse": 9.714139938354492,
      "art_l2": 0.15567468106746674,
      "t_mean": 0.1129894033074379
    },
    {
      "step": 47,
      "loss": 8.020530700683594,
      "flow_mse": 7.901943683624268,
      "art_l2": 0.1185869500041008,
      "t_mean": 0.18491406738758087
    },
    {
      "step": 48,
      "loss": 3.3771448135375977,
      "flow_mse": 3.2713968753814697,
      "art_l2": 0.10574790835380554,
      "t_mean": 0.3482275903224945
    },
    {
      "step": 49,
      "loss": 2.650822162628174,
      "flow_mse": 2.5946550369262695,
      "art_l2": 0.05616701766848564,
      "t_mean": 0.7947824001312256
    },
    {
      "step": 50,
      "loss": 3.721256732940674,
      "flow_mse": 3.514672040939331,
      "art_l2": 0.20658467710018158,
      "t_mean": 0.6497474312782288
    },
    {
      "step": 51,
      "loss": 3.4485726356506348,
      "flow_mse": 3.3420963287353516,
      "art_l2": 0.10647626221179962,
      "t_mean": 0.7770744562149048
    },
    {
      "step": 52,
      "loss": 5.627742290496826,
      "flow_mse": 5.568551063537598,
      "art_l2": 0.059191156178712845,
      "t_mean": 0.6030257344245911
    },
    {
      "step": 53,
      "loss": 4.632566452026367,
      "flow_mse": 4.5544023513793945,
      "art_l2": 0.07816386967897415,
      "t_mean": 0.4875844717025757
    },
    {
      "step": 54,
      "loss": 1.6710293292999268,
      "flow_mse": 1.5582162141799927,
      "art_l2": 0.1128130555152893,
      "t_mean": 0.5263466835021973
    },
    {
      "step": 55,
      "loss": 4.572157382965088,
      "flow_mse": 4.443684101104736,
      "art_l2": 0.12847349047660828,
      "t_mean": 0.7458981871604919
    },
    {
      "step": 56,
      "loss": 6.709090232849121,
      "flow_mse": 6.636884689331055,
      "art_l2": 0.07220546156167984,
      "t_mean": 0.5128523707389832
    },
    {
      "step": 57,
      "loss": 3.931116819381714,
      "flow_mse": 3.8281140327453613,
      "art_l2": 0.10300279408693314,
      "t_mean": 0.565317690372467
    },
    {
      "step": 58,
      "loss": 5.995464324951172,
      "flow_mse": 5.857316017150879,
      "art_l2": 0.13814836740493774,
      "t_mean": 0.3758229911327362
    },
    {
      "step": 59,
      "loss": 5.310952663421631,
      "flow_mse": 5.178413391113281,
      "art_l2": 0.13253910839557648,
      "t_mean": 0.32518190145492554
    },
    {
      "step": 60,
      "loss": 8.355796813964844,
      "flow_mse": 8.278560638427734,
      "art_l2": 0.07723631709814072,
      "t_mean": 0.3024666905403137
    },
    {
      "step": 61,
      "loss": 2.6846213340759277,
      "flow_mse": 2.648850440979004,
      "art_l2": 0.035770826041698456,
      "t_mean": 0.6290061473846436
    },
    {
      "step": 62,
      "loss": 4.212559223175049,
      "flow_mse": 4.078648090362549,
      "art_l2": 0.13391119241714478,
      "t_mean": 0.5848183631896973
    },
    {
      "step": 63,
      "loss": 8.528426170349121,
      "flow_mse": 8.443646430969238,
      "art_l2": 0.08478017896413803,
      "t_mean": 0.28910601139068604
    },
    {
      "step": 64,
      "loss": 4.044584274291992,
      "flow_mse": 3.9412622451782227,
      "art_l2": 0.10332194715738297,
      "t_mean": 0.2755487263202667
    },
    {
      "step": 65,
      "loss": 5.061482906341553,
      "flow_mse": 4.980173587799072,
      "art_l2": 0.08130916953086853,
      "t_mean": 0.6276462078094482
    },
    {
      "step": 66,
      "loss": 3.691426992416382,
      "flow_mse": 3.611083507537842,
      "art_l2": 0.08034352958202362,
      "t_mean": 0.7105835676193237
    },
    {
      "step": 67,
      "loss": 3.0004680156707764,
      "flow_mse": 2.8428893089294434,
      "art_l2": 0.15757867693901062,
      "t_mean": 0.5785131454467773
    },
    {
      "step": 68,
      "loss": 2.82769513130188,
      "flow_mse": 2.7895803451538086,
      "art_l2": 0.03811470791697502,
      "t_mean": 0.5499139428138733
    },
    {
      "step": 69,
      "loss": 5.256027698516846,
      "flow_mse": 5.185605049133301,
      "art_l2": 0.07042281329631805,
      "t_mean": 0.6692781448364258
    },
    {
      "step": 70,
      "loss": 2.980633497238159,
      "flow_mse": 2.8571670055389404,
      "art_l2": 0.12346649169921875,
      "t_mean": 0.7672603726387024
    },
    {
      "step": 71,
      "loss": 4.170300483703613,
      "flow_mse": 3.966761350631714,
      "art_l2": 0.2035389244556427,
      "t_mean": 0.7359468936920166
    },
    {
      "step": 72,
      "loss": 5.1769585609436035,
      "flow_mse": 5.109204292297363,
      "art_l2": 0.06775448471307755,
      "t_mean": 0.4078901410102844
    },
    {
      "step": 73,
      "loss": 5.3568339347839355,
      "flow_mse": 5.286825656890869,
      "art_l2": 0.07000840455293655,
      "t_mean": 0.6604650616645813
    },
    {
      "step": 74,
      "loss": 3.8415322303771973,
      "flow_mse": 3.7274580001831055,
      "art_l2": 0.11407420784235,
      "t_mean": 0.5058558583259583
    },
    {
      "step": 75,
      "loss": 6.763068675994873,
      "flow_mse": 6.663381576538086,
      "art_l2": 0.09968724846839905,
      "t_mean": 0.4211902618408203
    },
    {
      "step": 76,
      "loss": 4.013341426849365,
      "flow_mse": 3.8626937866210938,
      "art_l2": 0.15064767003059387,
      "t_mean": 0.5710673928260803
    },
    {
      "step": 77,
      "loss": 5.346048355102539,
      "flow_mse": 5.301741600036621,
      "art_l2": 0.04430694878101349,
      "t_mean": 0.5151851773262024
    },
    {
      "step": 78,
      "loss": 8.244471549987793,
      "flow_mse": 8.110602378845215,
      "art_l2": 0.13386961817741394,
      "t_mean": 0.15806666016578674
    },
    {
      "step": 79,
      "loss": 4.752902030944824,
      "flow_mse": 4.728747844696045,
      "art_l2": 0.024154027923941612,
      "t_mean": 0.6362106204032898
    },
    {
      "step": 80,
      "loss": 8.052010536193848,
      "flow_mse": 7.989875793457031,
      "art_l2": 0.062134984880685806,
      "t_mean": 0.08626709133386612
    }
  ]
}