{
  "args": {
    "model": "trm",
    "ckpt_root": "/home/yurenh2/rrm/trm/checkpoints/Sudoku-extreme-1k-aug-1000-ACT-torch/pretrain_mlp_t_sudoku_singleGPU",
    "ckpt_name": "step_26041",
    "train_steps": 10000,
    "batch_size": 4,
    "train_lr": 1e-05,
    "floss_lr": 0.0001,
    "floss_steps": 500,
    "interfloss_at": "0,500",
    "floss_mode": "engelken_l2",
    "lambda_star": 0.0,
    "k_lyap": 4,
    "lyap_act_steps": 4,
    "seed": 42,
    "eval_every": 1000,
    "eval_n": 512,
    "eval_batch_size": 32,
    "floss_log_every": 10,
    "eval_after_floss": true,
    "kl_beta": 10.0,
    "kl_replay_size": 64,
    "kl_batch_size": 4,
    "kl_temperature": 1.0,
    "out": "step7_D_trm_engelken_interfloss_kl10_26041_k4_batch4_10k.json"
  },
  "initial_acc": 0.59765625,
  "initial_tok_acc": 0.8614245756172839,
  "task_steps": [
    {
      "train_step": 1,
      "sup_loss": 1.0403251460331648
    },
    {
      "train_step": 2,
      "sup_loss": 0.7831296437826808
    },
    {
      "train_step": 3,
      "sup_loss": 1.0634877504769487
    },
    {
      "train_step": 4,
      "sup_loss": 1.2959107786731285
    },
    {
      "train_step": 5,
      "sup_loss": 1.1072553392426363
    },
    {
      "train_step": 6,
      "sup_loss": 1.0962899043973697
    },
    {
      "train_step": 7,
      "sup_loss": 1.0860744925613854
    },
    {
      "train_step": 8,
      "sup_loss": 1.1994162853816248
    },
    {
      "train_step": 9,
      "sup_loss": 0.9035157041547422
    },
    {
      "train_step": 10,
      "sup_loss": 0.9996094115588288
    },
    {
      "train_step": 11,
      "sup_loss": 1.0290821646463766
    },
    {
      "train_step": 12,
      "sup_loss": 0.7494518225882397
    },
    {
      "train_step": 13,
      "sup_loss": 1.1560139502458326
    },
    {
      "train_step": 14,
      "sup_loss": 0.9690593505153827
    },
    {
      "train_step": 15,
      "sup_loss": 1.3867824261765593
    },
    {
      "train_step": 16,
      "sup_loss": 1.1713786937354933
    },
    {
      "train_step": 17,
      "sup_loss": 1.1774686432937695
    },
    {
      "train_step": 18,
      "sup_loss": 0.8454486071166539
    },
    {
      "train_step": 19,
      "sup_loss": 1.1312322225007037
    },
    {
      "train_step": 20,
      "sup_loss": 1.0542340423920271
    },
    {
      "train_step": 21,
      "sup_loss": 1.0167175810221967
    },
    {
      "train_step": 22,
      "sup_loss": 1.0071892558243016
    },
    {
      "train_step": 23,
      "sup_loss": 1.0684235290389776
    },
    {
      "train_step": 24,
      "sup_loss": 1.1429263985473128
    },
    {
      "train_step": 25,
      "sup_loss": 1.0745116926795455
    },
    {
      "train_step": 26,
      "sup_loss": 0.9772747754917347
    },
    {
      "train_step": 27,
      "sup_loss": 0.9166478468663994
    },
    {
      "train_step": 28,
      "sup_loss": 0.9424294562292244
    },
    {
      "train_step": 29,
      "sup_loss": 0.952909371690173
    },
    {
      "train_step": 30,
      "sup_loss": 1.0052657579109436
    },
    {
      "train_step": 31,
      "sup_loss": 0.7639527576803236
    },
    {
      "train_step": 32,
      "sup_loss": 1.1223599455293916
    },
    {
      "train_step": 33,
      "sup_loss": 1.0264735617355565
    },
    {
      "train_step": 34,
      "sup_loss": 0.8757956897861044
    },
    {
      "train_step": 35,
      "sup_loss": 0.7786442508276131
    },
    {
      "train_step": 36,
      "sup_loss": 0.995477942843414
    },
    {
      "train_step": 37,
      "sup_loss": 0.9427741893578253
    },
    {
      "train_step": 38,
      "sup_loss": 1.1073327062625713
    },
    {
      "train_step": 39,
      "sup_loss": 1.0462605874709205
    },
    {
      "train_step": 40,
      "sup_loss": 0.9195322324294424
    },
    {
      "train_step": 41,
      "sup_loss": 0.9688621591271749
    },
    {
      "train_step": 42,
      "sup_loss": 1.0381808244792536
    },
    {
      "train_step": 43,
      "sup_loss": 0.9317817274062592
    },
    {
      "train_step": 44,
      "sup_loss": 0.8641137762698659
    },
    {
      "train_step": 45,
      "sup_loss": 0.8778386385323642
    },
    {
      "train_step": 46,
      "sup_loss": 0.8756965924090999
    },
    {
      "train_step": 47,
      "sup_loss": 0.9835877317249826
    },
    {
      "train_step": 48,
      "sup_loss": 0.917973276496139
    },
    {
      "train_step": 49,
      "sup_loss": 1.0586225699059617
    },
    {
      "train_step": 50,
      "sup_loss": 0.9186028399563532
    },
    {
      "train_step": 51,
      "sup_loss": 0.997310268417226
    },
    {
      "train_step": 52,
      "sup_loss": 0.8757362800822579
    },
    {
      "train_step": 53,
      "sup_loss": 0.894975413841335
    },
    {
      "train_step": 54,
      "sup_loss": 0.8655181461926419
    },
    {
      "train_step": 55,
      "sup_loss": 0.9938269877732189
    },
    {
      "train_step": 56,
      "sup_loss": 0.9523132613785966
    },
    {
      "train_step": 57,
      "sup_loss": 0.920955847875173
    },
    {
      "train_step": 58,
      "sup_loss": 0.8852219640201391
    },
    {
      "train_step": 59,
      "sup_loss": 0.6953053515819522
    },
    {
      "train_step": 60,
      "sup_loss": 0.9443247578591283
    },
    {
      "train_step": 61,
      "sup_loss": 0.9932526365666519
    },
    {
      "train_step": 62,
      "sup_loss": 0.8835446371432072
    },
    {
      "train_step": 63,
      "sup_loss": 0.8185871263272912
    },
    {
      "train_step": 64,
      "sup_loss": 0.9146679174530505
    },
    {
      "train_step": 65,
      "sup_loss": 0.809898350316218
    },
    {
      "train_step": 66,
      "sup_loss": 0.8464172938141878
    },
    {
      "train_step": 67,
      "sup_loss": 0.800917161747725
    },
    {
      "train_step": 68,
      "sup_loss": 0.8115797230221632
    },
    {
      "train_step": 69,
      "sup_loss": 0.7870937382858428
    },
    {
      "train_step": 70,
      "sup_loss": 0.8554718620854829
    },
    {
      "train_step": 71,
      "sup_loss": 0.8829257604058058
    },
    {
      "train_step": 72,
      "sup_loss": 0.9447761411339075
    },
    {
      "train_step": 73,
      "sup_loss": 0.9794465381438661
    },
    {
      "train_step": 74,
      "sup_loss": 0.8242210395814575
    },
    {
      "train_step": 75,
      "sup_loss": 0.8394461331736733
    },
    {
      "train_step": 76,
      "sup_loss": 0.6317516257676259
    },
    {
      "train_step": 77,
      "sup_loss": 0.9171515211576301
    },
    {
      "train_step": 78,
      "sup_loss": 0.8368876896584848
    },
    {
      "train_step": 79,
      "sup_loss": 0.8385987592842671
    },
    {
      "train_step": 80,
      "sup_loss": 0.8796235911469467
    },
    {
      "train_step": 81,
      "sup_loss": 0.9138523332065823
    },
    {
      "train_step": 82,
      "sup_loss": 0.9125149664068208
    },
    {
      "train_step": 83,
      "sup_loss": 0.7715950563745138
    },
    {
      "train_step": 84,
      "sup_loss": 0.7749734022011147
    },
    {
      "train_step": 85,
      "sup_loss": 0.8814164250010413
    },
    {
      "train_step": 86,
      "sup_loss": 0.7311161722505384
    },
    {
      "train_step": 87,
      "sup_loss": 0.8107095987416947
    },
    {
      "train_step": 88,
      "sup_loss": 0.808551656251754
    },
    {
      "train_step": 89,
      "sup_loss": 0.8857164120622939
    },
    {
      "train_step": 90,
      "sup_loss": 0.8683209727454886
    },
    {
      "train_step": 91,
      "sup_loss": 0.6948048476737075
    },
    {
      "train_step": 92,
      "sup_loss": 0.9390856985858043
    },
    {
      "train_step": 93,
      "sup_loss": 0.7698575001660715
    },
    {
      "train_step": 94,
      "sup_loss": 0.898431464638425
    },
    {
      "train_step": 95,
      "sup_loss": 0.935734973197651
    },
    {
      "train_step": 96,
      "sup_loss": 0.9267384745695696
    },
    {
      "train_step": 97,
      "sup_loss": 0.9292768883000153
    },
    {
      "train_step": 98,
      "sup_loss": 0.658922155503981
    },
    {
      "train_step": 99,
      "sup_loss": 0.6706835768368808
    },
    {
      "train_step": 100,
      "sup_loss": 0.7824526514458834
    },
    {
      "train_step": 101,
      "sup_loss": 0.9114505899487146
    },
    {
      "train_step": 102,
      "sup_loss": 0.8479857716103155
    },
    {
      "train_step": 103,
      "sup_loss": 0.7329352180182273
    },
    {
      "train_step": 104,
      "sup_loss": 0.8304630053324847
    },
    {
      "train_step": 105,
      "sup_loss": 0.7696208818926299
    },
    {
      "train_step": 106,
      "sup_loss": 0.8557424740442301
    },
    {
      "train_step": 107,
      "sup_loss": 0.7446803938378213
    },
    {
      "train_step": 108,
      "sup_loss": 0.9370060571055288
    },
    {
      "train_step": 109,
      "sup_loss": 0.8740674768516924
    },
    {
      "train_step": 110,
      "sup_loss": 0.853599195047833
    },
    {
      "train_step": 111,
      "sup_loss": 0.7605807083196832
    },
    {
      "train_step": 112,
      "sup_loss": 0.4742261524337822
    },
    {
      "train_step": 113,
      "sup_loss": 0.580267954906575
    },
    {
      "train_step": 114,
      "sup_loss": 0.8566607965671011
    },
    {
      "train_step": 115,
      "sup_loss": 0.7891956860705989
    },
    {
      "train_step": 116,
      "sup_loss": 0.8801160904818237
    },
    {
      "train_step": 117,
      "sup_loss": 0.763257343111892
    },
    {
      "train_step": 118,
      "sup_loss": 1.1126045708613863
    },
    {
      "train_step": 119,
      "sup_loss": 0.9035010006419081
    },
    {
      "train_step": 120,
      "sup_loss": 0.7154155733855007
    },
    {
      "train_step": 121,
      "sup_loss": 0.8131014528860013
    },
    {
      "train_step": 122,
      "sup_loss": 0.7438626816526027
    },
    {
      "train_step": 123,
      "sup_loss": 0.8958839330203195
    },
    {
      "train_step": 124,
      "sup_loss": 0.9146143944857081
    },
    {
      "train_step": 125,
      "sup_loss": 0.7811402606450091
    },
    {
      "train_step": 126,
      "sup_loss": 0.9053893623482765
    },
    {
      "train_step": 127,
      "sup_loss": 0.8880192626593797
    },
    {
      "train_step": 128,
      "sup_loss": 0.6503626408343862
    },
    {
      "train_step": 129,
      "sup_loss": 0.9948227267969016
    },
    {
      "train_step": 130,
      "sup_loss": 0.7829357231806566
    },
    {
      "train_step": 131,
      "sup_loss": 0.8480959395065004
    },
    {
      "train_step": 132,
      "sup_loss": 0.8672935605507226
    },
    {
      "train_step": 133,
      "sup_loss": 0.8529523401841418
    },
    {
      "train_step": 134,
      "sup_loss": 0.7783885562574718
    },
    {
      "train_step": 135,
      "sup_loss": 0.7867377306263516
    },
    {
      "train_step": 136,
      "sup_loss": 0.8377695459878439
    },
    {
      "train_step": 137,
      "sup_loss": 0.8730860173300804
    },
    {
      "train_step": 138,
      "sup_loss": 0.7468528837179922
    },
    {
      "train_step": 139,
      "sup_loss": 0.8303867201786787
    },
    {
      "train_step": 140,
      "sup_loss": 0.6625073054278932
    },
    {
      "train_step": 141,
      "sup_loss": 0.8526869813689404
    },
    {
      "train_step": 142,
      "sup_loss": 0.8989899503373079
    },
    {
      "train_step": 143,
      "sup_loss": 0.864608430027291
    },
    {
      "train_step": 144,
      "sup_loss": 0.7975706622669969
    },
    {
      "train_step": 145,
      "sup_loss": 0.7558367433802693
    },
    {
      "train_step": 146,
      "sup_loss": 0.802751294148994
    },
    {
      "train_step": 147,
      "sup_loss": 0.7933616380439577
    },
    {
      "train_step": 148,
      "sup_loss": 0.9167511379737193
    },
    {
      "train_step": 149,
      "sup_loss": 0.8820214470710125
    },
    {
      "train_step": 150,
      "sup_loss": 0.7923825326158289
    },
    {
      "train_step": 151,
      "sup_loss": 0.8510975599046237
    },
    {
      "train_step": 152,
      "sup_loss": 0.8085708488045885
    },
    {
      "train_step": 153,
      "sup_loss": 0.6801125997749858
    },
    {
      "train_step": 154,
      "sup_loss": 0.8121890624125726
    },
    {
      "train_step": 155,
      "sup_loss": 0.7309105965360733
    },
    {
      "train_step": 156,
      "sup_loss": 0.7102338846613042
    },
    {
      "train_step": 157,
      "sup_loss": 0.7868492707518421
    },
    {
      "train_step": 158,
      "sup_loss": 0.6040579239105968
    },
    {
      "train_step": 159,
      "sup_loss": 0.8580612107743391
    },
    {
      "train_step": 160,
      "sup_loss": 0.8323699801729904
    },
    {
      "train_step": 161,
      "sup_loss": 0.7295482769673575
    },
    {
      "train_step": 162,
      "sup_loss": 0.80718985330685
    },
    {
      "train_step": 163,
      "sup_loss": 0.9129652264567135
    },
    {
      "train_step": 164,
      "sup_loss": 0.8179038138396516
    },
    {
      "train_step": 165,
      "sup_loss": 0.8544041108297894
    },
    {
      "train_step": 166,
      "sup_loss": 0.8274168663846524
    },
    {
      "train_step": 167,
      "sup_loss": 0.8246761799212001
    },
    {
      "train_step": 168,
      "sup_loss": 0.6510109390081371
    },
    {
      "train_step": 169,
      "sup_loss": 0.9743008087451259
    },
    {
      "train_step": 170,
      "sup_loss": 0.7573899158744798
    },
    {
      "train_step": 171,
      "sup_loss": 0.722216884803475
    },
    {
      "train_step": 172,
      "sup_loss": 0.767935771035319
    },
    {
      "train_step": 173,
      "sup_loss": 0.8898586784094281
    },
    {
      "train_step": 174,
      "sup_loss": 0.5746005100911866
    },
    {
      "train_step": 175,
      "sup_loss": 0.8161249199362637
    },
    {
      "train_step": 176,
      "sup_loss": 0.8011729116341311
    },
    {
      "train_step": 177,
      "sup_loss": 0.7824939138062278
    },
    {
      "train_step": 178,
      "sup_loss": 0.7307860854516013
    },
    {
      "train_step": 179,
      "sup_loss": 0.7329826052135233
    },
    {
      "train_step": 180,
      "sup_loss": 0.8400781923136547
    },
    {
      "train_step": 181,
      "sup_loss": 0.7824997581658905
    },
    {
      "train_step": 182,
      "sup_loss": 0.8408219338487627
    },
    {
      "train_step": 183,
      "sup_loss": 0.7167077363055524
    },
    {
      "train_step": 184,
      "sup_loss": 0.6704387833754892
    },
    {
      "train_step": 185,
      "sup_loss": 0.8308008194070603
    },
    {
      "train_step": 186,
      "sup_loss": 0.8870335675657154
    },
    {
      "train_step": 187,
      "sup_loss": 0.7260865073153637
    },
    {
      "train_step": 188,
      "sup_loss": 0.7834550623565195
    },
    {
      "train_step": 189,
      "sup_loss": 0.8292853078814606
    },
    {
      "train_step": 190,
      "sup_loss": 0.9285900950140835
    },
    {
      "train_step": 191,
      "sup_loss": 0.7277168257174865
    },
    {
      "train_step": 192,
      "sup_loss": 0.824271455754566
    },
    {
      "train_step": 193,
      "sup_loss": 0.7844309347124965
    },
    {
      "train_step": 194,
      "sup_loss": 0.7656195205206948
    },
    {
      "train_step": 195,
      "sup_loss": 0.8586030841497913
    },
    {
      "train_step": 196,
      "sup_loss": 0.8351625777039573
    },
    {
      "train_step": 197,
      "sup_loss": 0.6702027831414685
    },
    {
      "train_step": 198,
      "sup_loss": 0.8956009608837614
    },
    {
      "train_step": 199,
      "sup_loss": 0.7765602425588465
    },
    {
      "train_step": 200,
      "sup_loss": 0.6561283631916148
    },
    {
      "train_step": 201,
      "sup_loss": 0.7082692812126578
    },
    {
      "train_step": 202,
      "sup_loss": 0.7190309607629528
    },
    {
      "train_step": 203,
      "sup_loss": 0.9191018672492027
    },
    {
      "train_step": 204,
      "sup_loss": 0.6852117222192206
    },
    {
      "train_step": 205,
      "sup_loss": 0.6070737985659398
    },
    {
      "train_step": 206,
      "sup_loss": 0.8190872382163707
    },
    {
      "train_step": 207,
      "sup_loss": 0.8872710059020984
    },
    {
      "train_step": 208,
      "sup_loss": 0.8151229668055646
    },
    {
      "train_step": 209,
      "sup_loss": 0.6615562017363438
    },
    {
      "train_step": 210,
      "sup_loss": 0.8631835908403015
    },
    {
      "train_step": 211,
      "sup_loss": 0.8801223139992969
    },
    {
      "train_step": 212,
      "sup_loss": 0.8033514896656448
    },
    {
      "train_step": 213,
      "sup_loss": 0.8244294207335697
    },
    {
      "train_step": 214,
      "sup_loss": 0.7534011321648801
    },
    {
      "train_step": 215,
      "sup_loss": 0.7113679978579055
    },
    {
      "train_step": 216,
      "sup_loss": 0.822839857449938
    },
    {
      "train_step": 217,
      "sup_loss": 0.7423181095130277
    },
    {
      "train_step": 218,
      "sup_loss": 0.7588921087440086
    },
    {
      "train_step": 219,
      "sup_loss": 0.8545417393459072
    },
    {
      "train_step": 220,
      "sup_loss": 0.8351535490552093
    },
    {
      "train_step": 221,
      "sup_loss": 0.8469508595320225
    },
    {
      "train_step": 222,
      "sup_loss": 0.7727575808454215
    },
    {
      "train_step": 223,
      "sup_loss": 0.5678238660087146
    },
    {
      "train_step": 224,
      "sup_loss": 0.819204987209274
    },
    {
      "train_step": 225,
      "sup_loss": 0.7365431531593283
    },
    {
      "train_step": 226,
      "sup_loss": 0.7531656566932515
    },
    {
      "train_step": 227,
      "sup_loss": 0.8816137006965884
    },
    {
      "train_step": 228,
      "sup_loss": 0.8513229152935265
    },
    {
      "train_step": 229,
      "sup_loss": 0.839659225608734
    },
    {
      "train_step": 230,
      "sup_loss": 0.7635257142870089
    },
    {
      "train_step": 231,
      "sup_loss": 0.9038475320576723
    },
    {
      "train_step": 232,
      "sup_loss": 0.7118510614068038
    },
    {
      "train_step": 233,
      "sup_loss": 0.7911748291292174
    },
    {
      "train_step": 234,
      "sup_loss": 0.8159055430901326
    },
    {
      "train_step": 235,
      "sup_loss": 0.7775873595164657
    },
    {
      "train_step": 236,
      "sup_loss": 0.6975001125003434
    },
    {
      "train_step": 237,
      "sup_loss": 0.752510708665402
    },
    {
      "train_step": 238,
      "sup_loss": 0.5948777484799088
    },
    {
      "train_step": 239,
      "sup_loss": 0.8661304764171889
    },
    {
      "train_step": 240,
      "sup_loss": 0.7856358954529367
    },
    {
      "train_step": 241,
      "sup_loss": 0.8180662684948173
    },
    {
      "train_step": 242,
      "sup_loss": 0.6174485741081079
    },
    {
      "train_step": 243,
      "sup_loss": 0.9119526590123433
    },
    {
      "train_step": 244,
      "sup_loss": 0.6914810182229212
    },
    {
      "train_step": 245,
      "sup_loss": 0.7468827090443989
    },
    {
      "train_step": 246,
      "sup_loss": 0.7231899595851682
    },
    {
      "train_step": 247,
      "sup_loss": 0.5260079829070651
    },
    {
      "train_step": 248,
      "sup_loss": 0.8554935479053383
    },
    {
      "train_step": 249,
      "sup_loss": 0.7398348100123856
    },
    {
      "train_step": 250,
      "sup_loss": 0.8302293537728015
    },
    {
      "train_step": 251,
      "sup_loss": 0.6935697048850918
    },
    {
      "train_step": 252,
      "sup_loss": 0.787568911098454
    },
    {
      "train_step": 253,
      "sup_loss": 0.6308542788870278
    },
    {
      "train_step": 254,
      "sup_loss": 0.7003149664378768
    },
    {
      "train_step": 255,
      "sup_loss": 0.4865967863704704
    },
    {
      "train_step": 256,
      "sup_loss": 0.7213234556127427
    },
    {
      "train_step": 257,
      "sup_loss": 0.5767994252292434
    },
    {
      "train_step": 258,
      "sup_loss": 0.8274385791347785
    },
    {
      "train_step": 259,
      "sup_loss": 0.6874704557003654
    },
    {
      "train_step": 260,
      "sup_loss": 0.8379295693078619
    },
    {
      "train_step": 261,
      "sup_loss": 0.7740249973212311
    },
    {
      "train_step": 262,
      "sup_loss": 0.7833419844117303
    },
    {
      "train_step": 263,
      "sup_loss": 0.8460462776681149
    },
    {
      "train_step": 264,
      "sup_loss": 0.8078595963743838
    },
    {
      "train_step": 265,
      "sup_loss": 0.7101444531008596
    },
    {
      "train_step": 266,
      "sup_loss": 0.9420441602752809
    },
    {
      "train_step": 267,
      "sup_loss": 0.9291931574050071
    },
    {
      "train_step": 268,
      "sup_loss": 0.8396544830178145
    },
    {
      "train_step": 269,
      "sup_loss": 0.6138917565336434
    },
    {
      "train_step": 270,
      "sup_loss": 0.7771250385688684
    },
    {
      "train_step": 271,
      "sup_loss": 0.6951820158714968
    },
    {
      "train_step": 272,
      "sup_loss": 0.6881266645114053
    },
    {
      "train_step": 273,
      "sup_loss": 0.8049787698468984
    },
    {
      "train_step": 274,
      "sup_loss": 0.6799253977712233
    },
    {
      "train_step": 275,
      "sup_loss": 0.5936978784642226
    },
    {
      "train_step": 276,
      "sup_loss": 0.7853927152900352
    },
    {
      "train_step": 277,
      "sup_loss": 0.8658565438939828
    },
    {
      "train_step": 278,
      "sup_loss": 0.5931359511074583
    },
    {
      "train_step": 279,
      "sup_loss": 0.9047973497963925
    },
    {
      "train_step": 280,
      "sup_loss": 0.4940111051231307
    },
    {
      "train_step": 281,
      "sup_loss": 0.7698648276189446
    },
    {
      "train_step": 282,
      "sup_loss": 0.7969218674190732
    },
    {
      "train_step": 283,
      "sup_loss": 0.7612385079897929
    },
    {
      "train_step": 284,
      "sup_loss": 0.7693718063231213
    },
    {
      "train_step": 285,
      "sup_loss": 0.8784462758583752
    },
    {
      "train_step": 286,
      "sup_loss": 0.8042525025931359
    },
    {
      "train_step": 287,
      "sup_loss": 0.7476388252304846
    },
    {
      "train_step": 288,
      "sup_loss": 0.850839081196899
    },
    {
      "train_step": 289,
      "sup_loss": 0.6607439725403869
    },
    {
      "train_step": 290,
      "sup_loss": 0.9461000059793634
    },
    {
      "train_step": 291,
      "sup_loss": 0.7482594943766787
    },
    {
      "train_step": 292,
      "sup_loss": 0.7148183828500874
    },
    {
      "train_step": 293,
      "sup_loss": 1.0308918870641386
    },
    {
      "train_step": 294,
      "sup_loss": 0.750070895043814
    },
    {
      "train_step": 295,
      "sup_loss": 0.7473150111229265
    },
    {
      "train_step": 296,
      "sup_loss": 0.810315889764051
    },
    {
      "train_step": 297,
      "sup_loss": 0.7732261887072794
    },
    {
      "train_step": 298,
      "sup_loss": 0.8282944538710076
    },
    {
      "train_step": 299,
      "sup_loss": 0.6636485118025783
    },
    {
      "train_step": 300,
      "sup_loss": 0.6152361666322628
    },
    {
      "train_step": 301,
      "sup_loss": 0.8349078382936757
    },
    {
      "train_step": 302,
      "sup_loss": 0.7067369588446502
    },
    {
      "train_step": 303,
      "sup_loss": 0.842964339993838
    },
    {
      "train_step": 304,
      "sup_loss": 0.8647676471765242
    },
    {
      "train_step": 305,
      "sup_loss": 0.777059629198203
    },
    {
      "train_step": 306,
      "sup_loss": 0.7741471700530058
    },
    {
      "train_step": 307,
      "sup_loss": 0.5337517958664176
    },
    {
      "train_step": 308,
      "sup_loss": 0.7814544827082376
    },
    {
      "train_step": 309,
      "sup_loss": 0.7574368331702099
    },
    {
      "train_step": 310,
      "sup_loss": 0.8336189680284587
    },
    {
      "train_step": 311,
      "sup_loss": 0.8430239212039323
    },
    {
      "train_step": 312,
      "sup_loss": 0.8733229959831527
    },
    {
      "train_step": 313,
      "sup_loss": 0.7644493491288687
    },
    {
      "train_step": 314,
      "sup_loss": 0.8135147000107263
    },
    {
      "train_step": 315,
      "sup_loss": 0.7694379810098073
    },
    {
      "train_step": 316,
      "sup_loss": 0.78245865965757
    },
    {
      "train_step": 317,
      "sup_loss": 0.8868010082278622
    },
    {
      "train_step": 318,
      "sup_loss": 0.7973482005624107
    },
    {
      "train_step": 319,
      "sup_loss": 0.6174839174603401
    },
    {
      "train_step": 320,
      "sup_loss": 0.813265881241614
    },
    {
      "train_step": 321,
      "sup_loss": 0.7418517070695971
    },
    {
      "train_step": 322,
      "sup_loss": 0.7530241219749648
    },
    {
      "train_step": 323,
      "sup_loss": 0.8010058182783759
    },
    {
      "train_step": 324,
      "sup_loss": 0.7674026847939305
    },
    {
      "train_step": 325,
      "sup_loss": 0.6896380966927553
    },
    {
      "train_step": 326,
      "sup_loss": 0.4771472575226372
    },
    {
      "train_step": 327,
      "sup_loss": 0.7285234224308612
    },
    {
      "train_step": 328,
      "sup_loss": 0.8918939402834823
    },
    {
      "train_step": 329,
      "sup_loss": 0.7972328928332433
    },
    {
      "train_step": 330,
      "sup_loss": 0.6957711722833665
    },
    {
      "train_step": 331,
      "sup_loss": 0.8222059383667849
    },
    {
      "train_step": 332,
      "sup_loss": 0.6375401226086145
    },
    {
      "train_step": 333,
      "sup_loss": 0.7195529168976724
    },
    {
      "train_step": 334,
      "sup_loss": 0.7034841433433028
    },
    {
      "train_step": 335,
      "sup_loss": 0.8635961970358468
    },
    {
      "train_step": 336,
      "sup_loss": 0.7152275729267557
    },
    {
      "train_step": 337,
      "sup_loss": 0.6925458841228589
    },
    {
      "train_step": 338,
      "sup_loss": 0.9179764971715563
    },
    {
      "train_step": 339,
      "sup_loss": 0.7955700918081331
    },
    {
      "train_step": 340,
      "sup_loss": 0.8824876322282516
    },
    {
      "train_step": 341,
      "sup_loss": 0.5850570786376014
    },
    {
      "train_step": 342,
      "sup_loss": 0.7845109528894801
    },
    {
      "train_step": 343,
      "sup_loss": 0.7456566622304341
    },
    {
      "train_step": 344,
      "sup_loss": 0.8160518104925468
    },
    {
      "train_step": 345,
      "sup_loss": 0.7961278292385054
    },
    {
      "train_step": 346,
      "sup_loss": 0.42761373765138866
    },
    {
      "train_step": 347,
      "sup_loss": 0.8124156543584328
    },
    {
      "train_step": 348,
      "sup_loss": 0.5576074750835204
    },
    {
      "train_step": 349,
      "sup_loss": 0.6950732169335669
    },
    {
      "train_step": 350,
      "sup_loss": 0.9330023269081885
    },
    {
      "train_step": 351,
      "sup_loss": 0.7170383587109859
    },
    {
      "train_step": 352,
      "sup_loss": 0.45874616579757654
    },
    {
      "train_step": 353,
      "sup_loss": 0.7368434042683527
    },
    {
      "train_step": 354,
      "sup_loss": 0.7012908269685169
    },
    {
      "train_step": 355,
      "sup_loss": 0.7258602358834132
    },
    {
      "train_step": 356,
      "sup_loss": 0.7304435477976459
    },
    {
      "train_step": 357,
      "sup_loss": 0.747962162032262
    },
    {
      "train_step": 358,
      "sup_loss": 0.8621522775909887
    },
    {
      "train_step": 359,
      "sup_loss": 0.673724500761975
    },
    {
      "train_step": 360,
      "sup_loss": 0.8209439660270096
    },
    {
      "train_step": 361,
      "sup_loss": 0.6628775931312387
    },
    {
      "train_step": 362,
      "sup_loss": 0.7339292475252199
    },
    {
      "train_step": 363,
      "sup_loss": 0.778266208499981
    },
    {
      "train_step": 364,
      "sup_loss": 0.8767937210725577
    },
    {
      "train_step": 365,
      "sup_loss": 0.7773407224380915
    },
    {
      "train_step": 366,
      "sup_loss": 0.8444904926940435
    },
    {
      "train_step": 367,
      "sup_loss": 0.6698305338212887
    },
    {
      "train_step": 368,
      "sup_loss": 0.6647030177708653
    },
    {
      "train_step": 369,
      "sup_loss": 0.6076299950019625
    },
    {
      "train_step": 370,
      "sup_loss": 0.7762966664519114
    },
    {
      "train_step": 371,
      "sup_loss": 0.6571720796694812
    },
    {
      "train_step": 372,
      "sup_loss": 0.8280881544622547
    },
    {
      "train_step": 373,
      "sup_loss": 0.6251334571813225
    },
    {
      "train_step": 374,
      "sup_loss": 0.8646749266899206
    },
    {
      "train_step": 375,
      "sup_loss": 0.7396668071549627
    },
    {
      "train_step": 376,
      "sup_loss": 0.7934433511228016
    },
    {
      "train_step": 377,
      "sup_loss": 0.6709531749801004
    },
    {
      "train_step": 378,
      "sup_loss": 0.7470734183643987
    },
    {
      "train_step": 379,
      "sup_loss": 0.8072861109476084
    },
    {
      "train_step": 380,
      "sup_loss": 0.7094951122874882
    },
    {
      "train_step": 381,
      "sup_loss": 0.6585880097796395
    },
    {
      "train_step": 382,
      "sup_loss": 0.7361980972701605
    },
    {
      "train_step": 383,
      "sup_loss": 0.6365713159682634
    },
    {
      "train_step": 384,
      "sup_loss": 0.8178576563322136
    },
    {
      "train_step": 385,
      "sup_loss": 0.6742101291130743
    },
    {
      "train_step": 386,
      "sup_loss": 0.6337617934443734
    },
    {
      "train_step": 387,
      "sup_loss": 0.7591840324878041
    },
    {
      "train_step": 388,
      "sup_loss": 0.669271044675174
    },
    {
      "train_step": 389,
      "sup_loss": 0.4951055076549508
    },
    {
      "train_step": 390,
      "sup_loss": 0.9463676451530997
    },
    {
      "train_step": 391,
      "sup_loss": 0.8460420823905839
    },
    {
      "train_step": 392,
      "sup_loss": 0.4854570400436341
    },
    {
      "train_step": 393,
      "sup_loss": 0.8408586517704173
    },
    {
      "train_step": 394,
      "sup_loss": 0.8567190998588149
    },
    {
      "train_step": 395,
      "sup_loss": 0.6128453284479611
    },
    {
      "train_step": 396,
      "sup_loss": 0.4983362484249844
    },
    {
      "train_step": 397,
      "sup_loss": 0.713517374222208
    },
    {
      "train_step": 398,
      "sup_loss": 0.45810721733499704
    },
    {
      "train_step": 399,
      "sup_loss": 0.6091653883759132
    },
    {
      "train_step": 400,
      "sup_loss": 0.4274157410728424
    },
    {
      "train_step": 401,
      "sup_loss": 0.6910117268630528
    },
    {
      "train_step": 402,
      "sup_loss": 0.707669515326762
    },
    {
      "train_step": 403,
      "sup_loss": 0.6694482547288909
    },
    {
      "train_step": 404,
      "sup_loss": 0.6170301990587689
    },
    {
      "train_step": 405,
      "sup_loss": 0.8263307241978242
    },
    {
      "train_step": 406,
      "sup_loss": 0.8455503934696187
    },
    {
      "train_step": 407,
      "sup_loss": 0.8430409171476563
    },
    {
      "train_step": 408,
      "sup_loss": 0.7106542957149745
    },
    {
      "train_step": 409,
      "sup_loss": 0.7762020693786555
    },
    {
      "train_step": 410,
      "sup_loss": 0.7184843058905601
    },
    {
      "train_step": 411,
      "sup_loss": 0.679364078481895
    },
    {
      "train_step": 412,
      "sup_loss": 0.8097469760740501
    },
    {
      "train_step": 413,
      "sup_loss": 0.9707885894662762
    },
    {
      "train_step": 414,
      "sup_loss": 0.6828876014791201
    },
    {
      "train_step": 415,
      "sup_loss": 0.5161769325705564
    },
    {
      "train_step": 416,
      "sup_loss": 0.6643151022498116
    },
    {
      "train_step": 417,
      "sup_loss": 0.8712681970527381
    },
    {
      "train_step": 418,
      "sup_loss": 0.664998608504142
    },
    {
      "train_step": 419,
      "sup_loss": 0.7119329310252512
    },
    {
      "train_step": 420,
      "sup_loss": 0.830395807195487
    },
    {
      "train_step": 421,
      "sup_loss": 0.8325783546451245
    },
    {
      "train_step": 422,
      "sup_loss": 0.7940284375172748
    },
    {
      "train_step": 423,
      "sup_loss": 0.5437189014510635
    },
    {
      "train_step": 424,
      "sup_loss": 0.8153432526324894
    },
    {
      "train_step": 425,
      "sup_loss": 0.7300773725110508
    },
    {
      "train_step": 426,
      "sup_loss": 0.5598960965332518
    },
    {
      "train_step": 427,
      "sup_loss": 0.76099710607962
    },
    {
      "train_step": 428,
      "sup_loss": 0.6370466529503565
    },
    {
      "train_step": 429,
      "sup_loss": 1.0826465846802071
    },
    {
      "train_step": 430,
      "sup_loss": 0.7794782359142398
    },
    {
      "train_step": 431,
      "sup_loss": 0.6681909341656269
    },
    {
      "train_step": 432,
      "sup_loss": 0.876138592101685
    },
    {
      "train_step": 433,
      "sup_loss": 0.8080900116875075
    },
    {
      "train_step": 434,
      "sup_loss": 0.8234722972302353
    },
    {
      "train_step": 435,
      "sup_loss": 0.6969807923163016
    },
    {
      "train_step": 436,
      "sup_loss": 0.8667475211954121
    },
    {
      "train_step": 437,
      "sup_loss": 0.6121405429545617
    },
    {
      "train_step": 438,
      "sup_loss": 0.6861049284246605
    },
    {
      "train_step": 439,
      "sup_loss": 0.6196656996327223
    },
    {
      "train_step": 440,
      "sup_loss": 0.5750879067070244
    },
    {
      "train_step": 441,
      "sup_loss": 0.7295022227653852
    },
    {
      "train_step": 442,
      "sup_loss": 0.8294053399604144
    },
    {
      "train_step": 443,
      "sup_loss": 0.7007216698253843
    },
    {
      "train_step": 444,
      "sup_loss": 0.8718098317337697
    },
    {
      "train_step": 445,
      "sup_loss": 0.8904019694821813
    },
    {
      "train_step": 446,
      "sup_loss": 0.8226793363966727
    },
    {
      "train_step": 447,
      "sup_loss": 0.865717394201662
    },
    {
      "train_step": 448,
      "sup_loss": 0.5769527715027997
    },
    {
      "train_step": 449,
      "sup_loss": 0.8134025714619227
    },
    {
      "train_step": 450,
      "sup_loss": 0.8642424766138751
    },
    {
      "train_step": 451,
      "sup_loss": 0.828386478299715
    },
    {
      "train_step": 452,
      "sup_loss": 0.7474841869316192
    },
    {
      "train_step": 453,
      "sup_loss": 0.6757787574326388
    },
    {
      "train_step": 454,
      "sup_loss": 0.659319092243089
    },
    {
      "train_step": 455,
      "sup_loss": 0.8585960241433578
    },
    {
      "train_step": 456,
      "sup_loss": 0.7856912928905149
    },
    {
      "train_step": 457,
      "sup_loss": 0.7270817929346173
    },
    {
      "train_step": 458,
      "sup_loss": 0.7067870935384498
    },
    {
      "train_step": 459,
      "sup_loss": 0.7295637816733274
    },
    {
      "train_step": 460,
      "sup_loss": 0.7624484805141457
    },
    {
      "train_step": 461,
      "sup_loss": 0.6855570894818899
    },
    {
      "train_step": 462,
      "sup_loss": 0.7839957994760475
    },
    {
      "train_step": 463,
      "sup_loss": 0.654864589890918
    },
    {
      "train_step": 464,
      "sup_loss": 0.7785825364144048
    },
    {
      "train_step": 465,
      "sup_loss": 0.7490111405526703
    },
    {
      "train_step": 466,
      "sup_loss": 0.6985769304430082
    },
    {
      "train_step": 467,
      "sup_loss": 0.736734499289629
    },
    {
      "train_step": 468,
      "sup_loss": 0.7224172244375839
    },
    {
      "train_step": 469,
      "sup_loss": 0.696947379355761
    },
    {
      "train_step": 470,
      "sup_loss": 0.7692405877637716
    },
    {
      "train_step": 471,
      "sup_loss": 0.6703553230081818
    },
    {
      "train_step": 472,
      "sup_loss": 0.7758435124709874
    },
    {
      "train_step": 473,
      "sup_loss": 0.7375462155375153
    },
    {
      "train_step": 474,
      "sup_loss": 0.8163530303291885
    },
    {
      "train_step": 475,
      "sup_loss": 0.545470970733105
    },
    {
      "train_step": 476,
      "sup_loss": 0.5394959931306468
    },
    {
      "train_step": 477,
      "sup_loss": 0.6207291705285634
    },
    {
      "train_step": 478,
      "sup_loss": 0.7588898250225231
    },
    {
      "train_step": 479,
      "sup_loss": 0.7982188598827833
    },
    {
      "train_step": 480,
      "sup_loss": 0.6400272801650385
    },
    {
      "train_step": 481,
      "sup_loss": 0.7737698981224121
    },
    {
      "train_step": 482,
      "sup_loss": 0.8171729104386785
    },
    {
      "train_step": 483,
      "sup_loss": 0.9170426441984659
    },
    {
      "train_step": 484,
      "sup_loss": 0.7368093418564013
    },
    {
      "train_step": 485,
      "sup_loss": 0.6066802020920007
    },
    {
      "train_step": 486,
      "sup_loss": 0.7991373785037879
    },
    {
      "train_step": 487,
      "sup_loss": 0.8103609587768694
    },
    {
      "train_step": 488,
      "sup_loss": 0.8614287362021075
    },
    {
      "train_step": 489,
      "sup_loss": 0.865897595968441
    },
    {
      "train_step": 490,
      "sup_loss": 0.8412592455997809
    },
    {
      "train_step": 491,
      "sup_loss": 0.7241805462501405
    },
    {
      "train_step": 492,
      "sup_loss": 0.793750957182477
    },
    {
      "train_step": 493,
      "sup_loss": 0.5815066455556088
    },
    {
      "train_step": 494,
      "sup_loss": 0.6475257287120326
    },
    {
      "train_step": 495,
      "sup_loss": 0.7630904388718418
    },
    {
      "train_step": 496,
      "sup_loss": 0.688563395680367
    },
    {
      "train_step": 497,
      "sup_loss": 0.5971330944877936
    },
    {
      "train_step": 498,
      "sup_loss": 0.6886764109097181
    },
    {
      "train_step": 499,
      "sup_loss": 0.6345816933855416
    },
    {
      "train_step": 500,
      "sup_loss": 1.0783274212485245
    },
    {
      "train_step": 501,
      "sup_loss": 1.061793253264621
    },
    {
      "train_step": 502,
      "sup_loss": 0.9273892875715538
    },
    {
      "train_step": 503,
      "sup_loss": 1.0576645366466884
    },
    {
      "train_step": 504,
      "sup_loss": 1.0598184475806445
    },
    {
      "train_step": 505,
      "sup_loss": 1.0484109706927238
    },
    {
      "train_step": 506,
      "sup_loss": 1.0548910990840334
    },
    {
      "train_step": 507,
      "sup_loss": 0.83401827694016
    },
    {
      "train_step": 508,
      "sup_loss": 0.89725003102349
    },
    {
      "train_step": 509,
      "sup_loss": 1.1619796784925405
    },
    {
      "train_step": 510,
      "sup_loss": 0.9095318784385605
    },
    {
      "train_step": 511,
      "sup_loss": 0.966041478498353
    },
    {
      "train_step": 512,
      "sup_loss": 1.1189220202544665
    },
    {
      "train_step": 513,
      "sup_loss": 1.035932116083954
    },
    {
      "train_step": 514,
      "sup_loss": 1.0437585484273846
    },
    {
      "train_step": 515,
      "sup_loss": 0.980463704862809
    },
    {
      "train_step": 516,
      "sup_loss": 1.3212493020220206
    },
    {
      "train_step": 517,
      "sup_loss": 0.8213103831098303
    },
    {
      "train_step": 518,
      "sup_loss": 1.2386827571976753
    },
    {
      "train_step": 519,
      "sup_loss": 1.1252236752351086
    },
    {
      "train_step": 520,
      "sup_loss": 1.1366204530542847
    },
    {
      "train_step": 521,
      "sup_loss": 1.0645472694434874
    },
    {
      "train_step": 522,
      "sup_loss": 1.042761974811604
    },
    {
      "train_step": 523,
      "sup_loss": 1.030931172298077
    },
    {
      "train_step": 524,
      "sup_loss": 1.1898241109630734
    },
    {
      "train_step": 525,
      "sup_loss": 0.9008423148288219
    },
    {
      "train_step": 526,
      "sup_loss": 0.9914200311443666
    },
    {
      "train_step": 527,
      "sup_loss": 0.9501471348239893
    },
    {
      "train_step": 528,
      "sup_loss": 1.210102040204386
    },
    {
      "train_step": 529,
      "sup_loss": 0.9503691814093762
    },
    {
      "train_step": 530,
      "sup_loss": 1.2952529992509425
    },
    {
      "train_step": 531,
      "sup_loss": 0.9526451213099726
    },
    {
      "train_step": 532,
      "sup_loss": 1.1638149448583577
    },
    {
      "train_step": 533,
      "sup_loss": 0.7834956037753618
    },
    {
      "train_step": 534,
      "sup_loss": 0.9697160656413892
    },
    {
      "train_step": 535,
      "sup_loss": 0.9017169488517967
    },
    {
      "train_step": 536,
      "sup_loss": 1.0339850862990343
    },
    {
      "train_step": 537,
      "sup_loss": 0.9403435374529939
    },
    {
      "train_step": 538,
      "sup_loss": 0.976295752279075
    },
    {
      "train_step": 539,
      "sup_loss": 0.9205329688499344
    },
    {
      "train_step": 540,
      "sup_loss": 0.5906608282144038
    },
    {
      "train_step": 541,
      "sup_loss": 0.8114398272554854
    },
    {
      "train_step": 542,
      "sup_loss": 0.9691199872654522
    },
    {
      "train_step": 543,
      "sup_loss": 1.01385770599529
    },
    {
      "train_step": 544,
      "sup_loss": 0.9917936874988572
    },
    {
      "train_step": 545,
      "sup_loss": 1.0012757853478846
    },
    {
      "train_step": 546,
      "sup_loss": 0.9674017941554506
    },
    {
      "train_step": 547,
      "sup_loss": 1.0184248882040252
    },
    {
      "train_step": 548,
      "sup_loss": 1.1157263654048952
    },
    {
      "train_step": 549,
      "sup_loss": 1.0626080546062924
    },
    {
      "train_step": 550,
      "sup_loss": 0.7758036499355534
    },
    {
      "train_step": 551,
      "sup_loss": 0.9328287968973958
    },
    {
      "train_step": 552,
      "sup_loss": 0.8109169984926362
    },
    {
      "train_step": 553,
      "sup_loss": 0.9191372043399058
    },
    {
      "train_step": 554,
      "sup_loss": 0.9197025613203244
    },
    {
      "train_step": 555,
      "sup_loss": 1.0078141530402034
    },
    {
      "train_step": 556,
      "sup_loss": 0.9573241001724352
    },
    {
      "train_step": 557,
      "sup_loss": 0.9128546753105122
    },
    {
      "train_step": 558,
      "sup_loss": 0.8566221225487747
    },
    {
      "train_step": 559,
      "sup_loss": 0.7514424129502936
    },
    {
      "train_step": 560,
      "sup_loss": 0.6370008471893651
    },
    {
      "train_step": 561,
      "sup_loss": 0.960873285353226
    },
    {
      "train_step": 562,
      "sup_loss": 0.8768675360785196
    },
    {
      "train_step": 563,
      "sup_loss": 0.9329882047123916
    },
    {
      "train_step": 564,
      "sup_loss": 1.1095958898517444
    },
    {
      "train_step": 565,
      "sup_loss": 1.082473848929873
    },
    {
      "train_step": 566,
      "sup_loss": 1.0804385911701413
    },
    {
      "train_step": 567,
      "sup_loss": 0.7714419398192458
    },
    {
      "train_step": 568,
      "sup_loss": 0.9686020531201603
    },
    {
      "train_step": 569,
      "sup_loss": 0.9895897589842237
    },
    {
      "train_step": 570,
      "sup_loss": 0.8181438532198195
    },
    {
      "train_step": 571,
      "sup_loss": 1.1803319439794424
    },
    {
      "train_step": 572,
      "sup_loss": 0.9120313765449404
    },
    {
      "train_step": 573,
      "sup_loss": 0.861933596786152
    },
    {
      "train_step": 574,
      "sup_loss": 0.7149161846491531
    },
    {
      "train_step": 575,
      "sup_loss": 0.9190908170714165
    },
    {
      "train_step": 576,
      "sup_loss": 0.914213479875307
    },
    {
      "train_step": 577,
      "sup_loss": 0.8083704003208927
    },
    {
      "train_step": 578,
      "sup_loss": 0.8336312384361061
    },
    {
      "train_step": 579,
      "sup_loss": 0.984203745865913
    },
    {
      "train_step": 580,
      "sup_loss": 0.8831947240496855
    },
    {
      "train_step": 581,
      "sup_loss": 0.8734149913614704
    },
    {
      "train_step": 582,
      "sup_loss": 0.905991172245982
    },
    {
      "train_step": 583,
      "sup_loss": 0.7879134648031507
    },
    {
      "train_step": 584,
      "sup_loss": 0.9829536745206916
    },
    {
      "train_step": 585,
      "sup_loss": 0.8424748029884765
    },
    {
      "train_step": 586,
      "sup_loss": 0.9107218944564719
    },
    {
      "train_step": 587,
      "sup_loss": 0.8196173500633397
    },
    {
      "train_step": 588,
      "sup_loss": 0.9763307749113219
    },
    {
      "train_step": 589,
      "sup_loss": 0.8319259798346647
    },
    {
      "train_step": 590,
      "sup_loss": 0.8809121601350364
    },
    {
      "train_step": 591,
      "sup_loss": 0.9126620240976413
    },
    {
      "train_step": 592,
      "sup_loss": 0.8258547837519559
    },
    {
      "train_step": 593,
      "sup_loss": 0.9015250676244398
    },
    {
      "train_step": 594,
      "sup_loss": 0.7152672900069731
    },
    {
      "train_step": 595,
      "sup_loss": 0.7189491349840869
    },
    {
      "train_step": 596,
      "sup_loss": 0.7948822677037277
    },
    {
      "train_step": 597,
      "sup_loss": 0.9283959515738152
    },
    {
      "train_step": 598,
      "sup_loss": 0.8416088790697316
    },
    {
      "train_step": 599,
      "sup_loss": 0.7228295594847908
    },
    {
      "train_step": 600,
      "sup_loss": 0.8340949972224834
    },
    {
      "train_step": 601,
      "sup_loss": 1.013157924869413
    },
    {
      "train_step": 602,
      "sup_loss": 0.8404986901393708
    },
    {
      "train_step": 603,
      "sup_loss": 0.9690713336940715
    },
    {
      "train_step": 604,
      "sup_loss": 0.8669142495608103
    },
    {
      "train_step": 605,
      "sup_loss": 0.8951627866499122
    },
    {
      "train_step": 606,
      "sup_loss": 0.7990149232018882
    },
    {
      "train_step": 607,
      "sup_loss": 0.8061511219083021
    },
    {
      "train_step": 608,
      "sup_loss": 0.8600210374669025
    },
    {
      "train_step": 609,
      "sup_loss": 0.8415640248400508
    },
    {
      "train_step": 610,
      "sup_loss": 0.9625885708494335
    },
    {
      "train_step": 611,
      "sup_loss": 0.909477797711728
    },
    {
      "train_step": 612,
      "sup_loss": 1.092217123133727
    },
    {
      "train_step": 613,
      "sup_loss": 0.7002219509695824
    },
    {
      "train_step": 614,
      "sup_loss": 0.7693357074272655
    },
    {
      "train_step": 615,
      "sup_loss": 0.9118223802361755
    },
    {
      "train_step": 616,
      "sup_loss": 0.7335624232211267
    },
    {
      "train_step": 617,
      "sup_loss": 0.9589858786898986
    },
    {
      "train_step": 618,
      "sup_loss": 0.696505630697213
    },
    {
      "train_step": 619,
      "sup_loss": 0.84039408310417
    },
    {
      "train_step": 620,
      "sup_loss": 0.9846042627729094
    },
    {
      "train_step": 621,
      "sup_loss": 0.7320437153814799
    },
    {
      "train_step": 622,
      "sup_loss": 0.8381748412062852
    },
    {
      "train_step": 623,
      "sup_loss": 0.8597900391551517
    },
    {
      "train_step": 624,
      "sup_loss": 0.7491367109311456
    },
    {
      "train_step": 625,
      "sup_loss": 0.7966040613593386
    },
    {
      "train_step": 626,
      "sup_loss": 0.925155223383582
    },
    {
      "train_step": 627,
      "sup_loss": 0.7589287883722319
    },
    {
      "train_step": 628,
      "sup_loss": 0.9326436174333298
    },
    {
      "train_step": 629,
      "sup_loss": 1.0319969622157263
    },
    {
      "train_step": 630,
      "sup_loss": 0.7873999547096195
    },
    {
      "train_step": 631,
      "sup_loss": 0.7612873469039444
    },
    {
      "train_step": 632,
      "sup_loss": 0.8924874445458895
    },
    {
      "train_step": 633,
      "sup_loss": 0.8051029925436646
    },
    {
      "train_step": 634,
      "sup_loss": 0.7795808120691067
    },
    {
      "train_step": 635,
      "sup_loss": 0.90221717540175
    },
    {
      "train_step": 636,
      "sup_loss": 0.8672346315006262
    },
    {
      "train_step": 637,
      "sup_loss": 0.8749801067267413
    },
    {
      "train_step": 638,
      "sup_loss": 0.6924558823572767
    },
    {
      "train_step": 639,
      "sup_loss": 0.7834198076164448
    },
    {
      "train_step": 640,
      "sup_loss": 0.8252011013481859
    },
    {
      "train_step": 641,
      "sup_loss": 0.8085792066391558
    },
    {
      "train_step": 642,
      "sup_loss": 0.7261081292422058
    },
    {
      "train_step": 643,
      "sup_loss": 0.9201204131961527
    },
    {
      "train_step": 644,
      "sup_loss": 0.7821284829268083
    },
    {
      "train_step": 645,
      "sup_loss": 0.8142738261340428
    },
    {
      "train_step": 646,
      "sup_loss": 0.8447042986569413
    },
    {
      "train_step": 647,
      "sup_loss": 0.7978870432462308
    },
    {
      "train_step": 648,
      "sup_loss": 0.8019173678608119
    },
    {
      "train_step": 649,
      "sup_loss": 0.9444586235398964
    },
    {
      "train_step": 650,
      "sup_loss": 0.8579840937193824
    },
    {
      "train_step": 651,
      "sup_loss": 0.8729750833049138
    },
    {
      "train_step": 652,
      "sup_loss": 0.8241851838968431
    },
    {
      "train_step": 653,
      "sup_loss": 0.7477181451779283
    },
    {
      "train_step": 654,
      "sup_loss": 0.765398059935555
    },
    {
      "train_step": 655,
      "sup_loss": 0.8228718135527636
    },
    {
      "train_step": 656,
      "sup_loss": 0.8078746486951301
    },
    {
      "train_step": 657,
      "sup_loss": 0.9370939749180099
    },
    {
      "train_step": 658,
      "sup_loss": 0.709898507804359
    },
    {
      "train_step": 659,
      "sup_loss": 0.7581979549730384
    },
    {
      "train_step": 660,
      "sup_loss": 0.7606634224530576
    },
    {
      "train_step": 661,
      "sup_loss": 0.8559437659307267
    },
    {
      "train_step": 662,
      "sup_loss": 0.869683244776515
    },
    {
      "train_step": 663,
      "sup_loss": 0.7955084530024127
    },
    {
      "train_step": 664,
      "sup_loss": 0.8379506465871577
    },
    {
      "train_step": 665,
      "sup_loss": 0.7100428344016746
    },
    {
      "train_step": 666,
      "sup_loss": 0.8389270052229679
    },
    {
      "train_step": 667,
      "sup_loss": 0.9623243789573589
    },
    {
      "train_step": 668,
      "sup_loss": 0.7908603018656413
    },
    {
      "train_step": 669,
      "sup_loss": 0.9945529691615546
    },
    {
      "train_step": 670,
      "sup_loss": 0.9778189571138863
    },
    {
      "train_step": 671,
      "sup_loss": 0.8501210892782077
    },
    {
      "train_step": 672,
      "sup_loss": 0.792934045695444
    },
    {
      "train_step": 673,
      "sup_loss": 0.835580482543365
    },
    {
      "train_step": 674,
      "sup_loss": 0.9665213139303032
    },
    {
      "train_step": 675,
      "sup_loss": 0.8012242274536866
    },
    {
      "train_step": 676,
      "sup_loss": 0.9031400066973612
    },
    {
      "train_step": 677,
      "sup_loss": 0.7482764222921218
    },
    {
      "train_step": 678,
      "sup_loss": 0.653051590286166
    },
    {
      "train_step": 679,
      "sup_loss": 0.9899613628505277
    },
    {
      "train_step": 680,
      "sup_loss": 0.828589722234761
    },
    {
      "train_step": 681,
      "sup_loss": 0.8699305573424532
    },
    {
      "train_step": 682,
      "sup_loss": 0.7570629677455835
    },
    {
      "train_step": 683,
      "sup_loss": 0.8176785422589078
    },
    {
      "train_step": 684,
      "sup_loss": 0.9381653721191423
    },
    {
      "train_step": 685,
      "sup_loss": 0.8454263460332566
    },
    {
      "train_step": 686,
      "sup_loss": 0.8894888594508157
    },
    {
      "train_step": 687,
      "sup_loss": 0.9563193908193306
    },
    {
      "train_step": 688,
      "sup_loss": 0.734786935598246
    },
    {
      "train_step": 689,
      "sup_loss": 0.7724821684190655
    },
    {
      "train_step": 690,
      "sup_loss": 0.8351392797571031
    },
    {
      "train_step": 691,
      "sup_loss": 0.8967172833562812
    },
    {
      "train_step": 692,
      "sup_loss": 0.7661000589974943
    },
    {
      "train_step": 693,
      "sup_loss": 0.8622643128186418
    },
    {
      "train_step": 694,
      "sup_loss": 0.7299523611955733
    },
    {
      "train_step": 695,
      "sup_loss": 0.8413122960562461
    },
    {
      "train_step": 696,
      "sup_loss": 0.838334822159851
    },
    {
      "train_step": 697,
      "sup_loss": 0.8305172367157527
    },
    {
      "train_step": 698,
      "sup_loss": 0.905524810713769
    },
    {
      "train_step": 699,
      "sup_loss": 0.8973955255160286
    },
    {
      "train_step": 700,
      "sup_loss": 0.7852566504535642
    },
    {
      "train_step": 701,
      "sup_loss": 0.7883484498621469
    },
    {
      "train_step": 702,
      "sup_loss": 0.764214729057527
    },
    {
      "train_step": 703,
      "sup_loss": 0.7986249376660737
    },
    {
      "train_step": 704,
      "sup_loss": 0.8147993662554213
    },
    {
      "train_step": 705,
      "sup_loss": 0.9045692282084657
    },
    {
      "train_step": 706,
      "sup_loss": 0.8275585757628616
    },
    {
      "train_step": 707,
      "sup_loss": 0.8789554888511868
    },
    {
      "train_step": 708,
      "sup_loss": 0.7641472191865778
    },
    {
      "train_step": 709,
      "sup_loss": 0.8585815437509176
    },
    {
      "train_step": 710,
      "sup_loss": 0.8347086085579524
    },
    {
      "train_step": 711,
      "sup_loss": 0.8306939096350765
    },
    {
      "train_step": 712,
      "sup_loss": 0.8347842868156083
    },
    {
      "train_step": 713,
      "sup_loss": 0.8244613910762308
    },
    {
      "train_step": 714,
      "sup_loss": 0.8916740736415328
    },
    {
      "train_step": 715,
      "sup_loss": 0.7044565351477057
    },
    {
      "train_step": 716,
      "sup_loss": 0.6318959204149561
    },
    {
      "train_step": 717,
      "sup_loss": 1.143087675915103
    },
    {
      "train_step": 718,
      "sup_loss": 0.7600043497945284
    },
    {
      "train_step": 719,
      "sup_loss": 0.7722398596064453
    },
    {
      "train_step": 720,
      "sup_loss": 0.7808237021874201
    },
    {
      "train_step": 721,
      "sup_loss": 0.7292812568441501
    },
    {
      "train_step": 722,
      "sup_loss": 0.8773712588101752
    },
    {
      "train_step": 723,
      "sup_loss": 0.7997511919216881
    },
    {
      "train_step": 724,
      "sup_loss": 0.7914007900366742
    },
    {
      "train_step": 725,
      "sup_loss": 0.6601889786844682
    },
    {
      "train_step": 726,
      "sup_loss": 0.8751254485336715
    },
    {
      "train_step": 727,
      "sup_loss": 0.746447669043616
    },
    {
      "train_step": 728,
      "sup_loss": 0.7020084651006062
    },
    {
      "train_step": 729,
      "sup_loss": 0.7960287347520294
    },
    {
      "train_step": 730,
      "sup_loss": 0.6873575081599075
    },
    {
      "train_step": 731,
      "sup_loss": 0.8094282696789967
    },
    {
      "train_step": 732,
      "sup_loss": 0.8241549773066035
    },
    {
      "train_step": 733,
      "sup_loss": 0.7089061761404465
    },
    {
      "train_step": 734,
      "sup_loss": 0.7528870828373138
    },
    {
      "train_step": 735,
      "sup_loss": 0.8242891682989304
    },
    {
      "train_step": 736,
      "sup_loss": 0.9171124577060382
    },
    {
      "train_step": 737,
      "sup_loss": 0.7940202217682314
    },
    {
      "train_step": 738,
      "sup_loss": 0.9204892067009086
    },
    {
      "train_step": 739,
      "sup_loss": 0.7441714286962972
    },
    {
      "train_step": 740,
      "sup_loss": 0.7552638474998556
    },
    {
      "train_step": 741,
      "sup_loss": 0.8455108536079053
    },
    {
      "train_step": 742,
      "sup_loss": 0.8357528622293879
    },
    {
      "train_step": 743,
      "sup_loss": 0.9035648232390975
    },
    {
      "train_step": 744,
      "sup_loss": 0.7813318805922956
    },
    {
      "train_step": 745,
      "sup_loss": 0.7741205978215097
    },
    {
      "train_step": 746,
      "sup_loss": 0.7713941961383302
    },
    {
      "train_step": 747,
      "sup_loss": 0.531708712681604
    },
    {
      "train_step": 748,
      "sup_loss": 0.8169877104775177
    },
    {
      "train_step": 749,
      "sup_loss": 0.838869709746542
    },
    {
      "train_step": 750,
      "sup_loss": 0.731198215275585
    },
    {
      "train_step": 751,
      "sup_loss": 0.7359230894217269
    },
    {
      "train_step": 752,
      "sup_loss": 0.6860729310476026
    },
    {
      "train_step": 753,
      "sup_loss": 0.7261074405485938
    },
    {
      "train_step": 754,
      "sup_loss": 0.7111007088976117
    },
    {
      "train_step": 755,
      "sup_loss": 0.7182847672718905
    },
    {
      "train_step": 756,
      "sup_loss": 0.8377120622131509
    },
    {
      "train_step": 757,
      "sup_loss": 0.711147974973033
    },
    {
      "train_step": 758,
      "sup_loss": 0.8321285800967723
    },
    {
      "train_step": 759,
      "sup_loss": 0.6948210308724415
    },
    {
      "train_step": 760,
      "sup_loss": 0.8083162257634422
    },
    {
      "train_step": 761,
      "sup_loss": 0.7833910546120916
    },
    {
      "train_step": 762,
      "sup_loss": 0.814133063484482
    },
    {
      "train_step": 763,
      "sup_loss": 0.8811716291070552
    },
    {
      "train_step": 764,
      "sup_loss": 0.8422314274558469
    },
    {
      "train_step": 765,
      "sup_loss": 0.5785608557680005
    },
    {
      "train_step": 766,
      "sup_loss": 0.8815311289102487
    },
    {
      "train_step": 767,
      "sup_loss": 0.6906019363631031
    },
    {
      "train_step": 768,
      "sup_loss": 0.8825135099911944
    },
    {
      "train_step": 769,
      "sup_loss": 0.7140462166347514
    },
    {
      "train_step": 770,
      "sup_loss": 0.8465704115646698
    },
    {
      "train_step": 771,
      "sup_loss": 0.6208816425732038
    },
    {
      "train_step": 772,
      "sup_loss": 0.9732601832276586
    },
    {
      "train_step": 773,
      "sup_loss": 0.7600441769968714
    },
    {
      "train_step": 774,
      "sup_loss": 0.8435250459771721
    },
    {
      "train_step": 775,
      "sup_loss": 0.8189324190261429
    },
    {
      "train_step": 776,
      "sup_loss": 0.8919682515392497
    },
    {
      "train_step": 777,
      "sup_loss": 0.835930881749864
    },
    {
      "train_step": 778,
      "sup_loss": 0.7807033814477371
    },
    {
      "train_step": 779,
      "sup_loss": 0.8260837949542975
    },
    {
      "train_step": 780,
      "sup_loss": 0.8393078760102507
    },
    {
      "train_step": 781,
      "sup_loss": 0.8294250912671608
    },
    {
      "train_step": 782,
      "sup_loss": 0.8873995792144651
    },
    {
      "train_step": 783,
      "sup_loss": 0.7280213543327948
    },
    {
      "train_step": 784,
      "sup_loss": 0.793382784116092
    },
    {
      "train_step": 785,
      "sup_loss": 0.8757059519544115
    },
    {
      "train_step": 786,
      "sup_loss": 0.8214724821630645
    },
    {
      "train_step": 787,
      "sup_loss": 0.7961574158671795
    },
    {
      "train_step": 788,
      "sup_loss": 0.9250854887238268
    },
    {
      "train_step": 789,
      "sup_loss": 0.7789653640766492
    },
    {
      "train_step": 790,
      "sup_loss": 0.8399638752646053
    },
    {
      "train_step": 791,
      "sup_loss": 0.7875669318696321
    },
    {
      "train_step": 792,
      "sup_loss": 0.9540702818728958
    },
    {
      "train_step": 793,
      "sup_loss": 0.7699589424152765
    },
    {
      "train_step": 794,
      "sup_loss": 0.7116468351903839
    },
    {
      "train_step": 795,
      "sup_loss": 0.8717434281675125
    },
    {
      "train_step": 796,
      "sup_loss": 0.7702380976694727
    },
    {
      "train_step": 797,
      "sup_loss": 0.7934664816056234
    },
    {
      "train_step": 798,
      "sup_loss": 0.8119756401222727
    },
    {
      "train_step": 799,
      "sup_loss": 0.9054787366937045
    },
    {
      "train_step": 800,
      "sup_loss": 0.735533246105228
    },
    {
      "train_step": 801,
      "sup_loss": 0.8198377242368282
    },
    {
      "train_step": 802,
      "sup_loss": 0.8614454660905574
    },
    {
      "train_step": 803,
      "sup_loss": 0.887836080697358
    },
    {
      "train_step": 804,
      "sup_loss": 0.6508245891698607
    },
    {
      "train_step": 805,
      "sup_loss": 0.7775641359404855
    },
    {
      "train_step": 806,
      "sup_loss": 0.683186437674009
    },
    {
      "train_step": 807,
      "sup_loss": 0.7160537154396095
    },
    {
      "train_step": 808,
      "sup_loss": 0.8471835848925007
    },
    {
      "train_step": 809,
      "sup_loss": 0.6919747850365245
    },
    {
      "train_step": 810,
      "sup_loss": 0.7351162870012184
    },
    {
      "train_step": 811,
      "sup_loss": 0.8028173033251021
    },
    {
      "train_step": 812,
      "sup_loss": 0.8173241014936978
    },
    {
      "train_step": 813,
      "sup_loss": 0.8321222285472177
    },
    {
      "train_step": 814,
      "sup_loss": 0.7992785739614807
    },
    {
      "train_step": 815,
      "sup_loss": 0.775434878120938
    },
    {
      "train_step": 816,
      "sup_loss": 0.8828603197651977
    },
    {
      "train_step": 817,
      "sup_loss": 0.6326775819219055
    },
    {
      "train_step": 818,
      "sup_loss": 0.7451985217885032
    },
    {
      "train_step": 819,
      "sup_loss": 0.681227411103593
    },
    {
      "train_step": 820,
      "sup_loss": 0.7805023294657378
    },
    {
      "train_step": 821,
      "sup_loss": 0.7019918623907169
    },
    {
      "train_step": 822,
      "sup_loss": 0.7852457968184339
    },
    {
      "train_step": 823,
      "sup_loss": 0.7929809595560257
    },
    {
      "train_step": 824,
      "sup_loss": 0.6453363819885407
    },
    {
      "train_step": 825,
      "sup_loss": 0.7537857991887501
    },
    {
      "train_step": 826,
      "sup_loss": 0.6720068092054918
    },
    {
      "train_step": 827,
      "sup_loss": 0.8750978260012159
    },
    {
      "train_step": 828,
      "sup_loss": 0.6993876003970312
    },
    {
      "train_step": 829,
      "sup_loss": 0.7836146214903926
    },
    {
      "train_step": 830,
      "sup_loss": 0.7765245574540071
    },
    {
      "train_step": 831,
      "sup_loss": 0.7809825036943804
    },
    {
      "train_step": 832,
      "sup_loss": 0.802543674086078
    },
    {
      "train_step": 833,
      "sup_loss": 0.7114137913912333
    },
    {
      "train_step": 834,
      "sup_loss": 0.7105360416723931
    },
    {
      "train_step": 835,
      "sup_loss": 0.9122872880719115
    },
    {
      "train_step": 836,
      "sup_loss": 0.6499887179312325
    },
    {
      "train_step": 837,
      "sup_loss": 0.8066786891941434
    },
    {
      "train_step": 838,
      "sup_loss": 0.7500302096524264
    },
    {
      "train_step": 839,
      "sup_loss": 0.6428706448331155
    },
    {
      "train_step": 840,
      "sup_loss": 0.6997510793600799
    },
    {
      "train_step": 841,
      "sup_loss": 0.8220832402981995
    },
    {
      "train_step": 842,
      "sup_loss": 0.6332787153915107
    },
    {
      "train_step": 843,
      "sup_loss": 0.8536971979773237
    },
    {
      "train_step": 844,
      "sup_loss": 0.9753287632848917
    },
    {
      "train_step": 845,
      "sup_loss": 0.7786749702695092
    },
    {
      "train_step": 846,
      "sup_loss": 0.7426181065264554
    },
    {
      "train_step": 847,
      "sup_loss": 0.9054637992276541
    },
    {
      "train_step": 848,
      "sup_loss": 0.7741285881143568
    },
    {
      "train_step": 849,
      "sup_loss": 0.8370323095850231
    },
    {
      "train_step": 850,
      "sup_loss": 0.6864129029325992
    },
    {
      "train_step": 851,
      "sup_loss": 0.7790339883207047
    },
    {
      "train_step": 852,
      "sup_loss": 0.6426959325565041
    },
    {
      "train_step": 853,
      "sup_loss": 0.7845067663182929
    },
    {
      "train_step": 854,
      "sup_loss": 0.7468415327587119
    },
    {
      "train_step": 855,
      "sup_loss": 0.8261877315478462
    },
    {
      "train_step": 856,
      "sup_loss": 0.895769007299627
    },
    {
      "train_step": 857,
      "sup_loss": 0.9822195490690911
    },
    {
      "train_step": 858,
      "sup_loss": 0.6281663963861278
    },
    {
      "train_step": 859,
      "sup_loss": 0.7850267882560119
    },
    {
      "train_step": 860,
      "sup_loss": 0.8680366496251938
    },
    {
      "train_step": 861,
      "sup_loss": 0.7790028006828188
    },
    {
      "train_step": 862,
      "sup_loss": 0.6492667003401085
    },
    {
      "train_step": 863,
      "sup_loss": 0.8651981505074751
    },
    {
      "train_step": 864,
      "sup_loss": 0.7783033462152871
    },
    {
      "train_step": 865,
      "sup_loss": 0.77660333949164
    },
    {
      "train_step": 866,
      "sup_loss": 0.796880489670525
    },
    {
      "train_step": 867,
      "sup_loss": 0.7582919727051188
    },
    {
      "train_step": 868,
      "sup_loss": 0.8436285967993721
    },
    {
      "train_step": 869,
      "sup_loss": 0.8752064950730384
    },
    {
      "train_step": 870,
      "sup_loss": 0.8487760897195105
    },
    {
      "train_step": 871,
      "sup_loss": 0.7149503634865111
    },
    {
      "train_step": 872,
      "sup_loss": 0.880125232108216
    },
    {
      "train_step": 873,
      "sup_loss": 0.6285789799360155
    },
    {
      "train_step": 874,
      "sup_loss": 0.7967136399729878
    },
    {
      "train_step": 875,
      "sup_loss": 0.7928809963277273
    },
    {
      "train_step": 876,
      "sup_loss": 0.7525728321872689
    },
    {
      "train_step": 877,
      "sup_loss": 0.8029018437919042
    },
    {
      "train_step": 878,
      "sup_loss": 0.9470639305921726
    },
    {
      "train_step": 879,
      "sup_loss": 0.8836934999547974
    },
    {
      "train_step": 880,
      "sup_loss": 0.7157186477754126
    },
    {
      "train_step": 881,
      "sup_loss": 0.7035641031127056
    },
    {
      "train_step": 882,
      "sup_loss": 0.8450900208875326
    },
    {
      "train_step": 883,
      "sup_loss": 0.7453635771653044
    },
    {
      "train_step": 884,
      "sup_loss": 0.8218945074511953
    },
    {
      "train_step": 885,
      "sup_loss": 0.8018417343365247
    },
    {
      "train_step": 886,
      "sup_loss": 0.7921833471794708
    },
    {
      "train_step": 887,
      "sup_loss": 0.6983153338524339
    },
    {
      "train_step": 888,
      "sup_loss": 0.8546335586473272
    },
    {
      "train_step": 889,
      "sup_loss": 0.833143669068632
    },
    {
      "train_step": 890,
      "sup_loss": 0.7302381812286783
    },
    {
      "train_step": 891,
      "sup_loss": 0.8458410465605842
    },
    {
      "train_step": 892,
      "sup_loss": 0.5941486720779993
    },
    {
      "train_step": 893,
      "sup_loss": 0.706594804633925
    },
    {
      "train_step": 894,
      "sup_loss": 0.7628655821426787
    },
    {
      "train_step": 895,
      "sup_loss": 0.8062913097020242
    },
    {
      "train_step": 896,
      "sup_loss": 0.7456899122774204
    },
    {
      "train_step": 897,
      "sup_loss": 0.8675802384143076
    },
    {
      "train_step": 898,
      "sup_loss": 0.8259626061594755
    },
    {
      "train_step": 899,
      "sup_loss": 0.7898136443078925
    },
    {
      "train_step": 900,
      "sup_loss": 0.7051574130096236
    },
    {
      "train_step": 901,
      "sup_loss": 0.7384120727926906
    },
    {
      "train_step": 902,
      "sup_loss": 0.7938258278444754
    },
    {
      "train_step": 903,
      "sup_loss": 0.941349638665239
    },
    {
      "train_step": 904,
      "sup_loss": 0.8596026114091714
    },
    {
      "train_step": 905,
      "sup_loss": 0.756104518708862
    },
    {
      "train_step": 906,
      "sup_loss": 0.758711603705219
    },
    {
      "train_step": 907,
      "sup_loss": 0.842382694963586
    },
    {
      "train_step": 908,
      "sup_loss": 0.6869706810855312
    },
    {
      "train_step": 909,
      "sup_loss": 0.8235058182562232
    },
    {
      "train_step": 910,
      "sup_loss": 0.9854700105501977
    },
    {
      "train_step": 911,
      "sup_loss": 0.7145966718856053
    },
    {
      "train_step": 912,
      "sup_loss": 0.636089024408931
    },
    {
      "train_step": 913,
      "sup_loss": 0.9988186428218431
    },
    {
      "train_step": 914,
      "sup_loss": 0.8969976936919816
    },
    {
      "train_step": 915,
      "sup_loss": 0.5769937721307377
    },
    {
      "train_step": 916,
      "sup_loss": 0.7882430976998079
    },
    {
      "train_step": 917,
      "sup_loss": 0.8952726311370992
    },
    {
      "train_step": 918,
      "sup_loss": 0.8084228722073523
    },
    {
      "train_step": 919,
      "sup_loss": 0.725338898945055
    },
    {
      "train_step": 920,
      "sup_loss": 0.863871202831549
    },
    {
      "train_step": 921,
      "sup_loss": 0.8329177575661143
    },
    {
      "train_step": 922,
      "sup_loss": 0.7432457438301969
    },
    {
      "train_step": 923,
      "sup_loss": 0.7558095454126863
    },
    {
      "train_step": 924,
      "sup_loss": 0.8269963591616265
    },
    {
      "train_step": 925,
      "sup_loss": 0.7262540535811971
    },
    {
      "train_step": 926,
      "sup_loss": 0.7767909185472389
    },
    {
      "train_step": 927,
      "sup_loss": 0.8171811232764044
    },
    {
      "train_step": 928,
      "sup_loss": 0.7898982033217308
    },
    {
      "train_step": 929,
      "sup_loss": 0.8853014302747314
    },
    {
      "train_step": 930,
      "sup_loss": 0.6214460427309956
    },
    {
      "train_step": 931,
      "sup_loss": 0.7595935968254185
    },
    {
      "train_step": 932,
      "sup_loss": 0.6321039819473425
    },
    {
      "train_step": 933,
      "sup_loss": 0.6917209640159361
    },
    {
      "train_step": 934,
      "sup_loss": 0.7220148049487162
    },
    {
      "train_step": 935,
      "sup_loss": 0.6026401856523899
    },
    {
      "train_step": 936,
      "sup_loss": 0.7206005384905909
    },
    {
      "train_step": 937,
      "sup_loss": 0.6371980889042284
    },
    {
      "train_step": 938,
      "sup_loss": 0.6370708691254059
    },
    {
      "train_step": 939,
      "sup_loss": 0.7071684571895476
    },
    {
      "train_step": 940,
      "sup_loss": 0.8222420933518346
    },
    {
      "train_step": 941,
      "sup_loss": 0.682811318482098
    },
    {
      "train_step": 942,
      "sup_loss": 0.8454517598530775
    },
    {
      "train_step": 943,
      "sup_loss": 0.8527770839862014
    },
    {
      "train_step": 944,
      "sup_loss": 0.8380622142163325
    },
    {
      "train_step": 945,
      "sup_loss": 0.8209457256433784
    },
    {
      "train_step": 946,
      "sup_loss": 0.7858712988140603
    },
    {
      "train_step": 947,
      "sup_loss": 0.7564624734019503
    },
    {
      "train_step": 948,
      "sup_loss": 0.8016516228952196
    },
    {
      "train_step": 949,
      "sup_loss": 0.7117371524837107
    },
    {
      "train_step": 950,
      "sup_loss": 0.8214144931773507
    },
    {
      "train_step": 951,
      "sup_loss": 0.8376572634529013
    },
    {
      "train_step": 952,
      "sup_loss": 0.8226013366248521
    },
    {
      "train_step": 953,
      "sup_loss": 0.8860412965632953
    },
    {
      "train_step": 954,
      "sup_loss": 0.7152867481869806
    },
    {
      "train_step": 955,
      "sup_loss": 0.8333693524521386
    },
    {
      "train_step": 956,
      "sup_loss": 0.7253866078661143
    },
    {
      "train_step": 957,
      "sup_loss": 0.8204319853458908
    },
    {
      "train_step": 958,
      "sup_loss": 0.7237275973452587
    },
    {
      "train_step": 959,
      "sup_loss": 0.7655293802429026
    },
    {
      "train_step": 960,
      "sup_loss": 0.8007543056872884
    },
    {
      "train_step": 961,
      "sup_loss": 0.7497613883344421
    },
    {
      "train_step": 962,
      "sup_loss": 0.80470357007026
    },
    {
      "train_step": 963,
      "sup_loss": 0.871770897629343
    },
    {
      "train_step": 964,
      "sup_loss": 0.7601149198870264
    },
    {
      "train_step": 965,
      "sup_loss": 0.7140907632802842
    },
    {
      "train_step": 966,
      "sup_loss": 0.5500520146611346
    },
    {
      "train_step": 967,
      "sup_loss": 0.696250947272461
    },
    {
      "train_step": 968,
      "sup_loss": 0.8510674987367743
    },
    {
      "train_step": 969,
      "sup_loss": 0.848305753666636
    },
    {
      "train_step": 970,
      "sup_loss": 0.8974720701987753
    },
    {
      "train_step": 971,
      "sup_loss": 0.859722615670687
    },
    {
      "train_step": 972,
      "sup_loss": 0.702934071766219
    },
    {
      "train_step": 973,
      "sup_loss": 0.7620052199707235
    },
    {
      "train_step": 974,
      "sup_loss": 0.6545853676866782
    },
    {
      "train_step": 975,
      "sup_loss": 0.6581970602071382
    },
    {
      "train_step": 976,
      "sup_loss": 0.8614720656833101
    },
    {
      "train_step": 977,
      "sup_loss": 0.7109554934091742
    },
    {
      "train_step": 978,
      "sup_loss": 0.6855331127925868
    },
    {
      "train_step": 979,
      "sup_loss": 0.7288621187962734
    },
    {
      "train_step": 980,
      "sup_loss": 0.8804090043394369
    },
    {
      "train_step": 981,
      "sup_loss": 0.6540989948768411
    },
    {
      "train_step": 982,
      "sup_loss": 0.8077154820741617
    },
    {
      "train_step": 983,
      "sup_loss": 0.6873940079214128
    },
    {
      "train_step": 984,
      "sup_loss": 0.7034000900341022
    },
    {
      "train_step": 985,
      "sup_loss": 0.85245408807605
    },
    {
      "train_step": 986,
      "sup_loss": 0.7617459524642858
    },
    {
      "train_step": 987,
      "sup_loss": 0.7771376356381716
    },
    {
      "train_step": 988,
      "sup_loss": 0.6875554948266085
    },
    {
      "train_step": 989,
      "sup_loss": 0.7199282453990055
    },
    {
      "train_step": 990,
      "sup_loss": 0.7079572929628539
    },
    {
      "train_step": 991,
      "sup_loss": 0.6501357350388333
    },
    {
      "train_step": 992,
      "sup_loss": 0.8752529241945766
    },
    {
      "train_step": 993,
      "sup_loss": 0.9779233456845016
    },
    {
      "train_step": 994,
      "sup_loss": 0.6479247321414052
    },
    {
      "train_step": 995,
      "sup_loss": 0.6556078621605284
    },
    {
      "train_step": 996,
      "sup_loss": 0.8084843179559331
    },
    {
      "train_step": 997,
      "sup_loss": 0.8209057586660993
    },
    {
      "train_step": 998,
      "sup_loss": 0.8106559777381399
    },
    {
      "train_step": 999,
      "sup_loss": 0.6710891821887709
    },
    {
      "train_step": 1000,
      "sup_loss": 0.7146530793714808
    },
    {
      "train_step": 1001,
      "sup_loss": 0.8919560168936975
    },
    {
      "train_step": 1002,
      "sup_loss": 0.7326798095686103
    },
    {
      "train_step": 1003,
      "sup_loss": 0.6053508131194908
    },
    {
      "train_step": 1004,
      "sup_loss": 0.8541945155486854
    },
    {
      "train_step": 1005,
      "sup_loss": 0.9663427008357202
    },
    {
      "train_step": 1006,
      "sup_loss": 0.8743265156073067
    },
    {
      "train_step": 1007,
      "sup_loss": 0.785336106829561
    },
    {
      "train_step": 1008,
      "sup_loss": 0.7981518477426931
    },
    {
      "train_step": 1009,
      "sup_loss": 0.773681019354571
    },
    {
      "train_step": 1010,
      "sup_loss": 0.8502159917053186
    },
    {
      "train_step": 1011,
      "sup_loss": 0.8727325337790474
    },
    {
      "train_step": 1012,
      "sup_loss": 0.6042764626651594
    },
    {
      "train_step": 1013,
      "sup_loss": 0.7241068563177778
    },
    {
      "train_step": 1014,
      "sup_loss": 0.8558753243103642
    },
    {
      "train_step": 1015,
      "sup_loss": 0.7947928489737253
    },
    {
      "train_step": 1016,
      "sup_loss": 0.7504157600627086
    },
    {
      "train_step": 1017,
      "sup_loss": 0.6171756267190929
    },
    {
      "train_step": 1018,
      "sup_loss": 0.7343532042342367
    },
    {
      "train_step": 1019,
      "sup_loss": 0.8447567216823014
    },
    {
      "train_step": 1020,
      "sup_loss": 0.8380597013194928
    },
    {
      "train_step": 1021,
      "sup_loss": 0.80481415320807
    },
    {
      "train_step": 1022,
      "sup_loss": 0.6935430595468705
    },
    {
      "train_step": 1023,
      "sup_loss": 0.7324177033720498
    },
    {
      "train_step": 1024,
      "sup_loss": 0.6633087167908331
    },
    {
      "train_step": 1025,
      "sup_loss": 0.6054491967151145
    },
    {
      "train_step": 1026,
      "sup_loss": 0.8411517574898179
    },
    {
      "train_step": 1027,
      "sup_loss": 0.6300008077401088
    },
    {
      "train_step": 1028,
      "sup_loss": 0.7770725548127658
    },
    {
      "train_step": 1029,
      "sup_loss": 0.9821360731093479
    },
    {
      "train_step": 1030,
      "sup_loss": 0.7597321631535768
    },
    {
      "train_step": 1031,
      "sup_loss": 0.46500004220425417
    },
    {
      "train_step": 1032,
      "sup_loss": 0.9853143734445156
    },
    {
      "train_step": 1033,
      "sup_loss": 0.7775407076068337
    },
    {
      "train_step": 1034,
      "sup_loss": 0.8816272149346293
    },
    {
      "train_step": 1035,
      "sup_loss": 0.7813373241760637
    },
    {
      "train_step": 1036,
      "sup_loss": 0.7195896580175739
    },
    {
      "train_step": 1037,
      "sup_loss": 0.6795048018459736
    },
    {
      "train_step": 1038,
      "sup_loss": 0.5276955904113392
    },
    {
      "train_step": 1039,
      "sup_loss": 0.8556597166326552
    },
    {
      "train_step": 1040,
      "sup_loss": 0.6975681078481363
    },
    {
      "train_step": 1041,
      "sup_loss": 0.8710474292929209
    },
    {
      "train_step": 1042,
      "sup_loss": 0.8969376397628448
    },
    {
      "train_step": 1043,
      "sup_loss": 0.9761017330695999
    },
    {
      "train_step": 1044,
      "sup_loss": 0.7164092601539978
    },
    {
      "train_step": 1045,
      "sup_loss": 0.7941393000946806
    },
    {
      "train_step": 1046,
      "sup_loss": 0.7643819238916815
    },
    {
      "train_step": 1047,
      "sup_loss": 0.6801264365795171
    },
    {
      "train_step": 1048,
      "sup_loss": 0.6212454473822902
    },
    {
      "train_step": 1049,
      "sup_loss": 0.7273808935750465
    },
    {
      "train_step": 1050,
      "sup_loss": 0.8998389694284704
    },
    {
      "train_step": 1051,
      "sup_loss": 0.653755391195829
    },
    {
      "train_step": 1052,
      "sup_loss": 0.5554326756724137
    },
    {
      "train_step": 1053,
      "sup_loss": 0.8545579247091155
    },
    {
      "train_step": 1054,
      "sup_loss": 0.6559559093454305
    },
    {
      "train_step": 1055,
      "sup_loss": 0.6734017256992904
    },
    {
      "train_step": 1056,
      "sup_loss": 0.9586760733166437
    },
    {
      "train_step": 1057,
      "sup_loss": 0.7453895885717052
    },
    {
      "train_step": 1058,
      "sup_loss": 0.5879324848383114
    },
    {
      "train_step": 1059,
      "sup_loss": 0.8589072790341268
    },
    {
      "train_step": 1060,
      "sup_loss": 0.7369985261328603
    },
    {
      "train_step": 1061,
      "sup_loss": 1.1747352674338185
    },
    {
      "train_step": 1062,
      "sup_loss": 0.8692146146453487
    },
    {
      "train_step": 1063,
      "sup_loss": 0.6889629807360983
    },
    {
      "train_step": 1064,
      "sup_loss": 0.823121137548374
    },
    {
      "train_step": 1065,
      "sup_loss": 0.47372272523901643
    },
    {
      "train_step": 1066,
      "sup_loss": 0.7699747575581789
    },
    {
      "train_step": 1067,
      "sup_loss": 0.9070301479154331
    },
    {
      "train_step": 1068,
      "sup_loss": 0.8380360350535934
    },
    {
      "train_step": 1069,
      "sup_loss": 0.6849964574069904
    },
    {
      "train_step": 1070,
      "sup_loss": 0.7156438086300825
    },
    {
      "train_step": 1071,
      "sup_loss": 0.6942423726837942
    },
    {
      "train_step": 1072,
      "sup_loss": 0.7500000580521384
    },
    {
      "train_step": 1073,
      "sup_loss": 0.7957110046932334
    },
    {
      "train_step": 1074,
      "sup_loss": 0.6786435218608069
    },
    {
      "train_step": 1075,
      "sup_loss": 0.852261780663845
    },
    {
      "train_step": 1076,
      "sup_loss": 0.7691669996050831
    },
    {
      "train_step": 1077,
      "sup_loss": 0.8030431875441356
    },
    {
      "train_step": 1078,
      "sup_loss": 0.7368617486732943
    },
    {
      "train_step": 1079,
      "sup_loss": 0.6936693434279635
    },
    {
      "train_step": 1080,
      "sup_loss": 0.7549519092925259
    },
    {
      "train_step": 1081,
      "sup_loss": 0.830818561242737
    },
    {
      "train_step": 1082,
      "sup_loss": 0.5807722919387985
    },
    {
      "train_step": 1083,
      "sup_loss": 0.7495762247886741
    },
    {
      "train_step": 1084,
      "sup_loss": 0.5921945178654507
    },
    {
      "train_step": 1085,
      "sup_loss": 0.7380584248616541
    },
    {
      "train_step": 1086,
      "sup_loss": 0.7311464395035054
    },
    {
      "train_step": 1087,
      "sup_loss": 0.4808251531034366
    },
    {
      "train_step": 1088,
      "sup_loss": 0.8321244764848872
    },
    {
      "train_step": 1089,
      "sup_loss": 0.6742497397085843
    },
    {
      "train_step": 1090,
      "sup_loss": 0.6199748629472274
    },
    {
      "train_step": 1091,
      "sup_loss": 0.6961409289206669
    },
    {
      "train_step": 1092,
      "sup_loss": 0.6745599376130637
    },
    {
      "train_step": 1093,
      "sup_loss": 0.878609226242195
    },
    {
      "train_step": 1094,
      "sup_loss": 0.574071217740307
    },
    {
      "train_step": 1095,
      "sup_loss": 0.8209414884126356
    },
    {
      "train_step": 1096,
      "sup_loss": 0.7095376905026389
    },
    {
      "train_step": 1097,
      "sup_loss": 0.8510817320828921
    },
    {
      "train_step": 1098,
      "sup_loss": 0.601031493937983
    },
    {
      "train_step": 1099,
      "sup_loss": 0.776377469396398
    },
    {
      "train_step": 1100,
      "sup_loss": 0.8315882986930057
    },
    {
      "train_step": 1101,
      "sup_loss": 0.47259055046778364
    },
    {
      "train_step": 1102,
      "sup_loss": 0.7579506308056093
    },
    {
      "train_step": 1103,
      "sup_loss": 0.734049031352446
    },
    {
      "train_step": 1104,
      "sup_loss": 0.688469892112965
    },
    {
      "train_step": 1105,
      "sup_loss": 0.748216634163513
    },
    {
      "train_step": 1106,
      "sup_loss": 0.6169870788632996
    },
    {
      "train_step": 1107,
      "sup_loss": 0.70647691498079
    },
    {
      "train_step": 1108,
      "sup_loss": 0.8980453905170243
    },
    {
      "train_step": 1109,
      "sup_loss": 0.48785198409238184
    },
    {
      "train_step": 1110,
      "sup_loss": 0.9187121072675105
    },
    {
      "train_step": 1111,
      "sup_loss": 0.8020336776791216
    },
    {
      "train_step": 1112,
      "sup_loss": 0.8514800779011138
    },
    {
      "train_step": 1113,
      "sup_loss": 0.7710246877999205
    },
    {
      "train_step": 1114,
      "sup_loss": 0.7206795809183819
    },
    {
      "train_step": 1115,
      "sup_loss": 0.4074149752923208
    },
    {
      "train_step": 1116,
      "sup_loss": 0.802730890322623
    },
    {
      "train_step": 1117,
      "sup_loss": 0.7111928103823615
    },
    {
      "train_step": 1118,
      "sup_loss": 0.6988734193096764
    },
    {
      "train_step": 1119,
      "sup_loss": 0.7970735588886626
    },
    {
      "train_step": 1120,
      "sup_loss": 0.605004741961583
    },
    {
      "train_step": 1121,
      "sup_loss": 0.6419401480786422
    },
    {
      "train_step": 1122,
      "sup_loss": 0.7029604955516625
    },
    {
      "train_step": 1123,
      "sup_loss": 0.7206520698503822
    },
    {
      "train_step": 1124,
      "sup_loss": 0.8367332394782279
    },
    {
      "train_step": 1125,
      "sup_loss": 0.44219145880840655
    },
    {
      "train_step": 1126,
      "sup_loss": 0.7613267901535322
    },
    {
      "train_step": 1127,
      "sup_loss": 0.7009906473783123
    },
    {
      "train_step": 1128,
      "sup_loss": 0.6595574127857622
    },
    {
      "train_step": 1129,
      "sup_loss": 0.771181593264681
    },
    {
      "train_step": 1130,
      "sup_loss": 0.6979468688833499
    },
    {
      "train_step": 1131,
      "sup_loss": 0.4894517669965677
    },
    {
      "train_step": 1132,
      "sup_loss": 0.7544315906401143
    },
    {
      "train_step": 1133,
      "sup_loss": 0.6455615567468153
    },
    {
      "train_step": 1134,
      "sup_loss": 0.5720694489242347
    },
    {
      "train_step": 1135,
      "sup_loss": 0.48059211331781015
    },
    {
      "train_step": 1136,
      "sup_loss": 0.616534392458501
    },
    {
      "train_step": 1137,
      "sup_loss": 0.7875063612690549
    },
    {
      "train_step": 1138,
      "sup_loss": 0.7863603346881829
    },
    {
      "train_step": 1139,
      "sup_loss": 0.6922942401130519
    },
    {
      "train_step": 1140,
      "sup_loss": 0.675839299551325
    },
    {
      "train_step": 1141,
      "sup_loss": 0.8250991972153205
    },
    {
      "train_step": 1142,
      "sup_loss": 0.7635809062667075
    },
    {
      "train_step": 1143,
      "sup_loss": 0.7617035135929867
    },
    {
      "train_step": 1144,
      "sup_loss": 0.6080803910376169
    },
    {
      "train_step": 1145,
      "sup_loss": 0.904580415825553
    },
    {
      "train_step": 1146,
      "sup_loss": 0.6038809530878807
    },
    {
      "train_step": 1147,
      "sup_loss": 0.8155025765992481
    },
    {
      "train_step": 1148,
      "sup_loss": 0.7168632671828494
    },
    {
      "train_step": 1149,
      "sup_loss": 0.7885072456658654
    },
    {
      "train_step": 1150,
      "sup_loss": 0.6693905591172403
    },
    {
      "train_step": 1151,
      "sup_loss": 0.6676563321188634
    },
    {
      "train_step": 1152,
      "sup_loss": 0.8490312558183297
    },
    {
      "train_step": 1153,
      "sup_loss": 0.8224209546260426
    },
    {
      "train_step": 1154,
      "sup_loss": 0.4796757663248137
    },
    {
      "train_step": 1155,
      "sup_loss": 0.6329313031871572
    },
    {
      "train_step": 1156,
      "sup_loss": 0.7449012058901708
    },
    {
      "train_step": 1157,
      "sup_loss": 0.6671388502264722
    },
    {
      "train_step": 1158,
      "sup_loss": 0.6846226627284518
    },
    {
      "train_step": 1159,
      "sup_loss": 0.6914599779380405
    },
    {
      "train_step": 1160,
      "sup_loss": 0.8174282244651456
    },
    {
      "train_step": 1161,
      "sup_loss": 0.6682925302337079
    },
    {
      "train_step": 1162,
      "sup_loss": 0.7513168222467619
    },
    {
      "train_step": 1163,
      "sup_loss": 0.9054794464739955
    },
    {
      "train_step": 1164,
      "sup_loss": 0.6366425908425444
    },
    {
      "train_step": 1165,
      "sup_loss": 0.7837681570224568
    },
    {
      "train_step": 1166,
      "sup_loss": 0.6731866308470346
    },
    {
      "train_step": 1167,
      "sup_loss": 0.6794355826434951
    },
    {
      "train_step": 1168,
      "sup_loss": 0.5982859213888136
    },
    {
      "train_step": 1169,
      "sup_loss": 0.7514954273986335
    },
    {
      "train_step": 1170,
      "sup_loss": 0.83655769399974
    },
    {
      "train_step": 1171,
      "sup_loss": 0.7987795882517923
    },
    {
      "train_step": 1172,
      "sup_loss": 0.7763165082363488
    },
    {
      "train_step": 1173,
      "sup_loss": 0.6408111324512674
    },
    {
      "train_step": 1174,
      "sup_loss": 0.6894077736790725
    },
    {
      "train_step": 1175,
      "sup_loss": 0.7626558627657314
    },
    {
      "train_step": 1176,
      "sup_loss": 0.3964195153767104
    },
    {
      "train_step": 1177,
      "sup_loss": 0.7689237289417199
    },
    {
      "train_step": 1178,
      "sup_loss": 0.8995555229980072
    },
    {
      "train_step": 1179,
      "sup_loss": 0.8682803310432566
    },
    {
      "train_step": 1180,
      "sup_loss": 0.8805719956395734
    },
    {
      "train_step": 1181,
      "sup_loss": 0.6614078096445222
    },
    {
      "train_step": 1182,
      "sup_loss": 0.6863526029694688
    },
    {
      "train_step": 1183,
      "sup_loss": 0.5821854638804793
    },
    {
      "train_step": 1184,
      "sup_loss": 0.6236454550557461
    },
    {
      "train_step": 1185,
      "sup_loss": 0.7659578507126575
    },
    {
      "train_step": 1186,
      "sup_loss": 0.7004943258675325
    },
    {
      "train_step": 1187,
      "sup_loss": 0.9689051289120575
    },
    {
      "train_step": 1188,
      "sup_loss": 0.7829078081161656
    },
    {
      "train_step": 1189,
      "sup_loss": 0.6905372921303192
    },
    {
      "train_step": 1190,
      "sup_loss": 0.8134335820155786
    },
    {
      "train_step": 1191,
      "sup_loss": 0.769004399637515
    },
    {
      "train_step": 1192,
      "sup_loss": 0.6238930419025716
    },
    {
      "train_step": 1193,
      "sup_loss": 0.685752275499808
    },
    {
      "train_step": 1194,
      "sup_loss": 0.8231459826634677
    },
    {
      "train_step": 1195,
      "sup_loss": 0.869850958498706
    },
    {
      "train_step": 1196,
      "sup_loss": 0.6457004100323879
    },
    {
      "train_step": 1197,
      "sup_loss": 0.8035118539628142
    },
    {
      "train_step": 1198,
      "sup_loss": 0.7240728427746865
    },
    {
      "train_step": 1199,
      "sup_loss": 0.7752714236615816
    },
    {
      "train_step": 1200,
      "sup_loss": 0.6779435584628718
    },
    {
      "train_step": 1201,
      "sup_loss": 0.765045544274491
    },
    {
      "train_step": 1202,
      "sup_loss": 0.763481529611792
    },
    {
      "train_step": 1203,
      "sup_loss": 0.7834502757473772
    },
    {
      "train_step": 1204,
      "sup_loss": 0.7518682033932234
    },
    {
      "train_step": 1205,
      "sup_loss": 0.6476752218021354
    },
    {
      "train_step": 1206,
      "sup_loss": 0.6494450997705338
    },
    {
      "train_step": 1207,
      "sup_loss": 0.7616613776287814
    },
    {
      "train_step": 1208,
      "sup_loss": 0.7464508671415464
    },
    {
      "train_step": 1209,
      "sup_loss": 0.739574885768595
    },
    {
      "train_step": 1210,
      "sup_loss": 0.5165049256463211
    },
    {
      "train_step": 1211,
      "sup_loss": 0.6798613337273143
    },
    {
      "train_step": 1212,
      "sup_loss": 0.8406608901268889
    },
    {
      "train_step": 1213,
      "sup_loss": 0.6282270471394651
    },
    {
      "train_step": 1214,
      "sup_loss": 0.5614911898274486
    },
    {
      "train_step": 1215,
      "sup_loss": 0.6946941365449282
    },
    {
      "train_step": 1216,
      "sup_loss": 0.7208074916718901
    },
    {
      "train_step": 1217,
      "sup_loss": 0.7847275068198852
    },
    {
      "train_step": 1218,
      "sup_loss": 0.7180904565407196
    },
    {
      "train_step": 1219,
      "sup_loss": 0.8595291219823227
    },
    {
      "train_step": 1220,
      "sup_loss": 0.8184540361784025
    },
    {
      "train_step": 1221,
      "sup_loss": 0.5200089712423615
    },
    {
      "train_step": 1222,
      "sup_loss": 0.7845901659756165
    },
    {
      "train_step": 1223,
      "sup_loss": 0.7796713204717153
    },
    {
      "train_step": 1224,
      "sup_loss": 0.7776818613972777
    },
    {
      "train_step": 1225,
      "sup_loss": 0.7153077203879883
    },
    {
      "train_step": 1226,
      "sup_loss": 0.6474334496533674
    },
    {
      "train_step": 1227,
      "sup_loss": 0.6822253123787803
    },
    {
      "train_step": 1228,
      "sup_loss": 0.750122286770078
    },
    {
      "train_step": 1229,
      "sup_loss": 0.5931804347037342
    },
    {
      "train_step": 1230,
      "sup_loss": 0.7239734581625829
    },
    {
      "train_step": 1231,
      "sup_loss": 0.7622244506610625
    },
    {
      "train_step": 1232,
      "sup_loss": 0.4822163330369428
    },
    {
      "train_step": 1233,
      "sup_loss": 0.7992543140847909
    },
    {
      "train_step": 1234,
      "sup_loss": 0.6266863374567855
    },
    {
      "train_step": 1235,
      "sup_loss": 0.7526022042626455
    },
    {
      "train_step": 1236,
      "sup_loss": 0.8371377354293416
    },
    {
      "train_step": 1237,
      "sup_loss": 0.7465637821563375
    },
    {
      "train_step": 1238,
      "sup_loss": 0.7791683984944142
    },
    {
      "train_step": 1239,
      "sup_loss": 0.615601322994021
    },
    {
      "train_step": 1240,
      "sup_loss": 0.8180859902066695
    },
    {
      "train_step": 1241,
      "sup_loss": 0.7021405890255038
    },
    {
      "train_step": 1242,
      "sup_loss": 0.6696641893870496
    },
    {
      "train_step": 1243,
      "sup_loss": 0.6338760345096129
    },
    {
      "train_step": 1244,
      "sup_loss": 0.794416060607384
    },
    {
      "train_step": 1245,
      "sup_loss": 0.5773712531496111
    },
    {
      "train_step": 1246,
      "sup_loss": 0.5095560439495094
    },
    {
      "train_step": 1247,
      "sup_loss": 0.7120723666889106
    },
    {
      "train_step": 1248,
      "sup_loss": 0.6813057415486834
    },
    {
      "train_step": 1249,
      "sup_loss": 0.7121067832878495
    },
    {
      "train_step": 1250,
      "sup_loss": 0.653366165035247
    },
    {
      "train_step": 1251,
      "sup_loss": 0.7913586312415807
    },
    {
      "train_step": 1252,
      "sup_loss": 0.636088777436667
    },
    {
      "train_step": 1253,
      "sup_loss": 0.7062481151578625
    },
    {
      "train_step": 1254,
      "sup_loss": 0.6963376347979747
    },
    {
      "train_step": 1255,
      "sup_loss": 0.5375696177876351
    },
    {
      "train_step": 1256,
      "sup_loss": 0.8539263913927567
    },
    {
      "train_step": 1257,
      "sup_loss": 0.8505015111935836
    },
    {
      "train_step": 1258,
      "sup_loss": 0.9094759726554346
    },
    {
      "train_step": 1259,
      "sup_loss": 0.7295404996806433
    },
    {
      "train_step": 1260,
      "sup_loss": 0.8657114143072459
    },
    {
      "train_step": 1261,
      "sup_loss": 0.7448493472105232
    },
    {
      "train_step": 1262,
      "sup_loss": 0.6978891539541573
    },
    {
      "train_step": 1263,
      "sup_loss": 0.7037753407495727
    },
    {
      "train_step": 1264,
      "sup_loss": 0.778782741063979
    },
    {
      "train_step": 1265,
      "sup_loss": 0.7900904526277145
    },
    {
      "train_step": 1266,
      "sup_loss": 0.7740102941376076
    },
    {
      "train_step": 1267,
      "sup_loss": 0.633949630620035
    },
    {
      "train_step": 1268,
      "sup_loss": 0.7310552981353443
    },
    {
      "train_step": 1269,
      "sup_loss": 0.8587440174173002
    },
    {
      "train_step": 1270,
      "sup_loss": 0.5656749914781365
    },
    {
      "train_step": 1271,
      "sup_loss": 0.7561113475541089
    },
    {
      "train_step": 1272,
      "sup_loss": 0.8213182196071614
    },
    {
      "train_step": 1273,
      "sup_loss": 0.5208744778152079
    },
    {
      "train_step": 1274,
      "sup_loss": 0.7865236011797259
    },
    {
      "train_step": 1275,
      "sup_loss": 0.7634245324006532
    },
    {
      "train_step": 1276,
      "sup_loss": 0.8448005120095714
    },
    {
      "train_step": 1277,
      "sup_loss": 0.7013570842104189
    },
    {
      "train_step": 1278,
      "sup_loss": 0.7013416898051669
    },
    {
      "train_step": 1279,
      "sup_loss": 0.6581341086191058
    },
    {
      "train_step": 1280,
      "sup_loss": 0.7546555188358766
    },
    {
      "train_step": 1281,
      "sup_loss": 0.8385891620034747
    },
    {
      "train_step": 1282,
      "sup_loss": 0.8310482566774003
    },
    {
      "train_step": 1283,
      "sup_loss": 0.8018071341261733
    },
    {
      "train_step": 1284,
      "sup_loss": 0.801485012704084
    },
    {
      "train_step": 1285,
      "sup_loss": 0.7583057027515887
    },
    {
      "train_step": 1286,
      "sup_loss": 0.8041336205004129
    },
    {
      "train_step": 1287,
      "sup_loss": 0.633384440283735
    },
    {
      "train_step": 1288,
      "sup_loss": 0.6525445588370069
    },
    {
      "train_step": 1289,
      "sup_loss": 0.7911839240511079
    },
    {
      "train_step": 1290,
      "sup_loss": 1.1765650460765773
    },
    {
      "train_step": 1291,
      "sup_loss": 0.7793938464280323
    },
    {
      "train_step": 1292,
      "sup_loss": 0.6975251569328055
    },
    {
      "train_step": 1293,
      "sup_loss": 0.702943003215417
    },
    {
      "train_step": 1294,
      "sup_loss": 0.6769715658842212
    },
    {
      "train_step": 1295,
      "sup_loss": 0.6458803910775983
    },
    {
      "train_step": 1296,
      "sup_loss": 0.7491397905323222
    },
    {
      "train_step": 1297,
      "sup_loss": 0.761250874059183
    },
    {
      "train_step": 1298,
      "sup_loss": 0.7865574783490819
    },
    {
      "train_step": 1299,
      "sup_loss": 0.6523396762546579
    },
    {
      "train_step": 1300,
      "sup_loss": 0.6403067007938611
    },
    {
      "train_step": 1301,
      "sup_loss": 0.6435044067098394
    },
    {
      "train_step": 1302,
      "sup_loss": 0.6516034948157162
    },
    {
      "train_step": 1303,
      "sup_loss": 0.8415930076886585
    },
    {
      "train_step": 1304,
      "sup_loss": 0.6004116391866128
    },
    {
      "train_step": 1305,
      "sup_loss": 0.6232442821974454
    },
    {
      "train_step": 1306,
      "sup_loss": 0.8200488180318045
    },
    {
      "train_step": 1307,
      "sup_loss": 0.7783781560833448
    },
    {
      "train_step": 1308,
      "sup_loss": 0.5866098486674063
    },
    {
      "train_step": 1309,
      "sup_loss": 0.7431152477227329
    },
    {
      "train_step": 1310,
      "sup_loss": 0.6588564567738814
    },
    {
      "train_step": 1311,
      "sup_loss": 0.7845785359149872
    },
    {
      "train_step": 1312,
      "sup_loss": 0.7584363154881215
    },
    {
      "train_step": 1313,
      "sup_loss": 0.7864246142416692
    },
    {
      "train_step": 1314,
      "sup_loss": 0.8212554064066957
    },
    {
      "train_step": 1315,
      "sup_loss": 0.592157923834928
    },
    {
      "train_step": 1316,
      "sup_loss": 0.799179416181588
    },
    {
      "train_step": 1317,
      "sup_loss": 0.6609467363526172
    },
    {
      "train_step": 1318,
      "sup_loss": 0.6314047666899228
    },
    {
      "train_step": 1319,
      "sup_loss": 0.8115476584395984
    },
    {
      "train_step": 1320,
      "sup_loss": 0.7684972548850216
    },
    {
      "train_step": 1321,
      "sup_loss": 0.7208747104790718
    },
    {
      "train_step": 1322,
      "sup_loss": 0.5979651178037395
    },
    {
      "train_step": 1323,
      "sup_loss": 0.8012247603801944
    },
    {
      "train_step": 1324,
      "sup_loss": 0.8986457471960447
    },
    {
      "train_step": 1325,
      "sup_loss": 0.7010746422035319
    },
    {
      "train_step": 1326,
      "sup_loss": 0.7973490040792369
    },
    {
      "train_step": 1327,
      "sup_loss": 0.7159286620196489
    },
    {
      "train_step": 1328,
      "sup_loss": 0.7984909859615517
    },
    {
      "train_step": 1329,
      "sup_loss": 0.595887118761026
    },
    {
      "train_step": 1330,
      "sup_loss": 0.6637588229814128
    },
    {
      "train_step": 1331,
      "sup_loss": 0.6871636795501077
    },
    {
      "train_step": 1332,
      "sup_loss": 0.6834407286870403
    },
    {
      "train_step": 1333,
      "sup_loss": 0.7569091313912499
    },
    {
      "train_step": 1334,
      "sup_loss": 0.6570144865777655
    },
    {
      "train_step": 1335,
      "sup_loss": 0.6970395680589049
    },
    {
      "train_step": 1336,
      "sup_loss": 0.7691435563965985
    },
    {
      "train_step": 1337,
      "sup_loss": 0.4285027932416847
    },
    {
      "train_step": 1338,
      "sup_loss": 0.31733973851544606
    },
    {
      "train_step": 1339,
      "sup_loss": 0.6259392823361994
    },
    {
      "train_step": 1340,
      "sup_loss": 0.8277574339281871
    },
    {
      "train_step": 1341,
      "sup_loss": 0.8306097483261046
    },
    {
      "train_step": 1342,
      "sup_loss": 0.7203334992573435
    },
    {
      "train_step": 1343,
      "sup_loss": 0.571104501733532
    },
    {
      "train_step": 1344,
      "sup_loss": 0.7555536649026068
    },
    {
      "train_step": 1345,
      "sup_loss": 0.699096530478015
    },
    {
      "train_step": 1346,
      "sup_loss": 0.7891837909248476
    },
    {
      "train_step": 1347,
      "sup_loss": 0.6151302122670249
    },
    {
      "train_step": 1348,
      "sup_loss": 0.6952559150839946
    },
    {
      "train_step": 1349,
      "sup_loss": 0.6434297854255636
    },
    {
      "train_step": 1350,
      "sup_loss": 0.671047627869689
    },
    {
      "train_step": 1351,
      "sup_loss": 0.8374519570251931
    },
    {
      "train_step": 1352,
      "sup_loss": 0.7469510203462003
    },
    {
      "train_step": 1353,
      "sup_loss": 0.8673503114085236
    },
    {
      "train_step": 1354,
      "sup_loss": 0.6920398241095286
    },
    {
      "train_step": 1355,
      "sup_loss": 0.6855402139101012
    },
    {
      "train_step": 1356,
      "sup_loss": 0.8116682813901686
    },
    {
      "train_step": 1357,
      "sup_loss": 0.7343539366491313
    },
    {
      "train_step": 1358,
      "sup_loss": 0.7056152199695389
    },
    {
      "train_step": 1359,
      "sup_loss": 0.6625700804235395
    },
    {
      "train_step": 1360,
      "sup_loss": 0.836767873904592
    },
    {
      "train_step": 1361,
      "sup_loss": 0.7635197408060672
    },
    {
      "train_step": 1362,
      "sup_loss": 0.6132954211073298
    },
    {
      "train_step": 1363,
      "sup_loss": 0.8012019794128792
    },
    {
      "train_step": 1364,
      "sup_loss": 0.7803666110970421
    },
    {
      "train_step": 1365,
      "sup_loss": 0.6765573034726167
    },
    {
      "train_step": 1366,
      "sup_loss": 0.6048565993524297
    },
    {
      "train_step": 1367,
      "sup_loss": 0.7120923126540788
    },
    {
      "train_step": 1368,
      "sup_loss": 0.7842518628647677
    },
    {
      "train_step": 1369,
      "sup_loss": 0.735182824807737
    },
    {
      "train_step": 1370,
      "sup_loss": 0.7309514244641485
    },
    {
      "train_step": 1371,
      "sup_loss": 0.8618283607718391
    },
    {
      "train_step": 1372,
      "sup_loss": 0.6186963190475704
    },
    {
      "train_step": 1373,
      "sup_loss": 0.5482772305966415
    },
    {
      "train_step": 1374,
      "sup_loss": 0.6586780752126277
    },
    {
      "train_step": 1375,
      "sup_loss": 0.8555295670899513
    },
    {
      "train_step": 1376,
      "sup_loss": 0.5076810931107886
    },
    {
      "train_step": 1377,
      "sup_loss": 0.6029705789624179
    },
    {
      "train_step": 1378,
      "sup_loss": 0.8459235376838814
    },
    {
      "train_step": 1379,
      "sup_loss": 0.8186030700868584
    },
    {
      "train_step": 1380,
      "sup_loss": 0.7866930845002417
    },
    {
      "train_step": 1381,
      "sup_loss": 0.7960983382387243
    },
    {
      "train_step": 1382,
      "sup_loss": 0.7453552045098207
    },
    {
      "train_step": 1383,
      "sup_loss": 0.847393015751154
    },
    {
      "train_step": 1384,
      "sup_loss": 0.6888445724608976
    },
    {
      "train_step": 1385,
      "sup_loss": 0.846758233114379
    },
    {
      "train_step": 1386,
      "sup_loss": 0.6291287639977619
    },
    {
      "train_step": 1387,
      "sup_loss": 0.6536136873635464
    },
    {
      "train_step": 1388,
      "sup_loss": 0.7384019962639814
    },
    {
      "train_step": 1389,
      "sup_loss": 0.6301019155405672
    },
    {
      "train_step": 1390,
      "sup_loss": 0.8213217446920521
    },
    {
      "train_step": 1391,
      "sup_loss": 0.7881319191103967
    },
    {
      "train_step": 1392,
      "sup_loss": 0.6640167594314164
    },
    {
      "train_step": 1393,
      "sup_loss": 0.6759359840411172
    },
    {
      "train_step": 1394,
      "sup_loss": 0.7367012207683616
    },
    {
      "train_step": 1395,
      "sup_loss": 0.7392825151797362
    },
    {
      "train_step": 1396,
      "sup_loss": 0.8126981522239907
    },
    {
      "train_step": 1397,
      "sup_loss": 0.5699794594859937
    },
    {
      "train_step": 1398,
      "sup_loss": 0.8047954292797428
    },
    {
      "train_step": 1399,
      "sup_loss": 0.6663009307052217
    },
    {
      "train_step": 1400,
      "sup_loss": 0.6591085043330636
    },
    {
      "train_step": 1401,
      "sup_loss": 0.7087825262826929
    },
    {
      "train_step": 1402,
      "sup_loss": 0.7861085724476694
    },
    {
      "train_step": 1403,
      "sup_loss": 0.8462767728729703
    },
    {
      "train_step": 1404,
      "sup_loss": 0.8200914568205242
    },
    {
      "train_step": 1405,
      "sup_loss": 0.8246935797325311
    },
    {
      "train_step": 1406,
      "sup_loss": 0.7875134990143373
    },
    {
      "train_step": 1407,
      "sup_loss": 0.8416190285131573
    },
    {
      "train_step": 1408,
      "sup_loss": 0.5970123947410196
    },
    {
      "train_step": 1409,
      "sup_loss": 0.5345230470141416
    },
    {
      "train_step": 1410,
      "sup_loss": 0.6912435761662452
    },
    {
      "train_step": 1411,
      "sup_loss": 0.6345643949319821
    },
    {
      "train_step": 1412,
      "sup_loss": 0.8306524472192729
    },
    {
      "train_step": 1413,
      "sup_loss": 0.6735912618179499
    },
    {
      "train_step": 1414,
      "sup_loss": 0.4268589301813989
    },
    {
      "train_step": 1415,
      "sup_loss": 0.7891586589409232
    },
    {
      "train_step": 1416,
      "sup_loss": 0.6087249636534039
    },
    {
      "train_step": 1417,
      "sup_loss": 0.6036173404493201
    },
    {
      "train_step": 1418,
      "sup_loss": 0.758947622039153
    },
    {
      "train_step": 1419,
      "sup_loss": 0.6836864975942873
    },
    {
      "train_step": 1420,
      "sup_loss": 0.6816528612854518
    },
    {
      "train_step": 1421,
      "sup_loss": 0.6399883515262765
    },
    {
      "train_step": 1422,
      "sup_loss": 0.7693934299841882
    },
    {
      "train_step": 1423,
      "sup_loss": 0.7458505998230232
    },
    {
      "train_step": 1424,
      "sup_loss": 0.6476081265672295
    },
    {
      "train_step": 1425,
      "sup_loss": 0.7818015622686805
    },
    {
      "train_step": 1426,
      "sup_loss": 0.7999808559995388
    },
    {
      "train_step": 1427,
      "sup_loss": 0.613143720491177
    },
    {
      "train_step": 1428,
      "sup_loss": 0.7321120538923535
    },
    {
      "train_step": 1429,
      "sup_loss": 0.6987864871811718
    },
    {
      "train_step": 1430,
      "sup_loss": 0.4876638277310339
    },
    {
      "train_step": 1431,
      "sup_loss": 0.5269057514902014
    },
    {
      "train_step": 1432,
      "sup_loss": 0.8100722029096311
    },
    {
      "train_step": 1433,
      "sup_loss": 0.48009957831339006
    },
    {
      "train_step": 1434,
      "sup_loss": 0.70158103367977
    },
    {
      "train_step": 1435,
      "sup_loss": 0.725213062752493
    },
    {
      "train_step": 1436,
      "sup_loss": 0.8154438006403174
    },
    {
      "train_step": 1437,
      "sup_loss": 0.5754420580730415
    },
    {
      "train_step": 1438,
      "sup_loss": 0.5266717578509537
    },
    {
      "train_step": 1439,
      "sup_loss": 0.7230744238487217
    },
    {
      "train_step": 1440,
      "sup_loss": 0.822259885500258
    },
    {
      "train_step": 1441,
      "sup_loss": 0.7446581370681025
    },
    {
      "train_step": 1442,
      "sup_loss": 0.5367521558927361
    },
    {
      "train_step": 1443,
      "sup_loss": 0.7053462840256739
    },
    {
      "train_step": 1444,
      "sup_loss": 0.843260969383223
    },
    {
      "train_step": 1445,
      "sup_loss": 0.8378705937123168
    },
    {
      "train_step": 1446,
      "sup_loss": 0.673837704762011
    },
    {
      "train_step": 1447,
      "sup_loss": 0.6749894839861551
    },
    {
      "train_step": 1448,
      "sup_loss": 0.7735496768824516
    },
    {
      "train_step": 1449,
      "sup_loss": 0.8720311297173909
    },
    {
      "train_step": 1450,
      "sup_loss": 0.7110688105381745
    },
    {
      "train_step": 1451,
      "sup_loss": 0.9100607950787958
    },
    {
      "train_step": 1452,
      "sup_loss": 0.7272376620844678
    },
    {
      "train_step": 1453,
      "sup_loss": 0.5747289781391459
    },
    {
      "train_step": 1454,
      "sup_loss": 0.7684447221935106
    },
    {
      "train_step": 1455,
      "sup_loss": 0.7022846981945208
    },
    {
      "train_step": 1456,
      "sup_loss": 0.48276667907459286
    },
    {
      "train_step": 1457,
      "sup_loss": 0.8194171838097327
    },
    {
      "train_step": 1458,
      "sup_loss": 0.7029685690707974
    },
    {
      "train_step": 1459,
      "sup_loss": 0.6322753895250375
    },
    {
      "train_step": 1460,
      "sup_loss": 0.71304477525762
    },
    {
      "train_step": 1461,
      "sup_loss": 0.7014145813199637
    },
    {
      "train_step": 1462,
      "sup_loss": 0.6886779487817555
    },
    {
      "train_step": 1463,
      "sup_loss": 0.5728394375396236
    },
    {
      "train_step": 1464,
      "sup_loss": 0.7043553362317827
    },
    {
      "train_step": 1465,
      "sup_loss": 0.845608060190338
    },
    {
      "train_step": 1466,
      "sup_loss": 0.5202063891361909
    },
    {
      "train_step": 1467,
      "sup_loss": 0.8156964096103372
    },
    {
      "train_step": 1468,
      "sup_loss": 0.3503350201371572
    },
    {
      "train_step": 1469,
      "sup_loss": 0.719807951950616
    },
    {
      "train_step": 1470,
      "sup_loss": 0.7291213353208673
    },
    {
      "train_step": 1471,
      "sup_loss": 0.5556388316569353
    },
    {
      "train_step": 1472,
      "sup_loss": 0.7745190257604525
    },
    {
      "train_step": 1473,
      "sup_loss": 0.5458764189459732
    },
    {
      "train_step": 1474,
      "sup_loss": 0.635895269850143
    },
    {
      "train_step": 1475,
      "sup_loss": 0.7176398179874162
    },
    {
      "train_step": 1476,
      "sup_loss": 0.8517654601063767
    },
    {
      "train_step": 1477,
      "sup_loss": 0.6629642033173396
    },
    {
      "train_step": 1478,
      "sup_loss": 0.5157388951117257
    },
    {
      "train_step": 1479,
      "sup_loss": 0.5429009526562206
    },
    {
      "train_step": 1480,
      "sup_loss": 0.6059287254220491
    },
    {
      "train_step": 1481,
      "sup_loss": 0.45491171283973725
    },
    {
      "train_step": 1482,
      "sup_loss": 0.6917961826828098
    },
    {
      "train_step": 1483,
      "sup_loss": 0.790224887634136
    },
    {
      "train_step": 1484,
      "sup_loss": 0.5676985497326771
    },
    {
      "train_step": 1485,
      "sup_loss": 0.6591312310999927
    },
    {
      "train_step": 1486,
      "sup_loss": 0.5372800052533426
    },
    {
      "train_step": 1487,
      "sup_loss": 0.8759680539430711
    },
    {
      "train_step": 1488,
      "sup_loss": 0.5751866453089246
    },
    {
      "train_step": 1489,
      "sup_loss": 0.6635724127515297
    },
    {
      "train_step": 1490,
      "sup_loss": 0.631509925495188
    },
    {
      "train_step": 1491,
      "sup_loss": 0.8065889121923258
    },
    {
      "train_step": 1492,
      "sup_loss": 0.6217443944337907
    },
    {
      "train_step": 1493,
      "sup_loss": 0.5747775554266156
    },
    {
      "train_step": 1494,
      "sup_loss": 0.5295925194744941
    },
    {
      "train_step": 1495,
      "sup_loss": 0.697837247876374
    },
    {
      "train_step": 1496,
      "sup_loss": 0.8022901311765122
    },
    {
      "train_step": 1497,
      "sup_loss": 0.92418025019936
    },
    {
      "train_step": 1498,
      "sup_loss": 0.29743460607329064
    },
    {
      "train_step": 1499,
      "sup_loss": 0.6462872045489481
    },
    {
      "train_step": 1500,
      "sup_loss": 0.7258903077782696
    },
    {
      "train_step": 1501,
      "sup_loss": 0.8472891997220477
    },
    {
      "train_step": 1502,
      "sup_loss": 0.7959748206209216
    },
    {
      "train_step": 1503,
      "sup_loss": 0.8189575827211844
    },
    {
      "train_step": 1504,
      "sup_loss": 0.6604858105355429
    },
    {
      "train_step": 1505,
      "sup_loss": 0.58249349043108
    },
    {
      "train_step": 1506,
      "sup_loss": 0.6741773251768856
    },
    {
      "train_step": 1507,
      "sup_loss": 0.748547530731214
    },
    {
      "train_step": 1508,
      "sup_loss": 0.6399941577301931
    },
    {
      "train_step": 1509,
      "sup_loss": 0.6154403981041754
    },
    {
      "train_step": 1510,
      "sup_loss": 0.6789656252762102
    },
    {
      "train_step": 1511,
      "sup_loss": 0.7512798048294184
    },
    {
      "train_step": 1512,
      "sup_loss": 0.8252807979136585
    },
    {
      "train_step": 1513,
      "sup_loss": 0.7076750789014973
    },
    {
      "train_step": 1514,
      "sup_loss": 0.543540540501684
    },
    {
      "train_step": 1515,
      "sup_loss": 0.8447561522500572
    },
    {
      "train_step": 1516,
      "sup_loss": 0.5948983632822125
    },
    {
      "train_step": 1517,
      "sup_loss": 0.7256406865246972
    },
    {
      "train_step": 1518,
      "sup_loss": 0.6187675330503121
    },
    {
      "train_step": 1519,
      "sup_loss": 0.9033717760182643
    },
    {
      "train_step": 1520,
      "sup_loss": 0.512831188658684
    },
    {
      "train_step": 1521,
      "sup_loss": 0.690218363767091
    },
    {
      "train_step": 1522,
      "sup_loss": 0.6635322334682432
    },
    {
      "train_step": 1523,
      "sup_loss": 0.6527903442965746
    },
    {
      "train_step": 1524,
      "sup_loss": 0.5537043428069004
    },
    {
      "train_step": 1525,
      "sup_loss": 0.8416355062132426
    },
    {
      "train_step": 1526,
      "sup_loss": 0.846975161660968
    },
    {
      "train_step": 1527,
      "sup_loss": 0.8471229026463212
    },
    {
      "train_step": 1528,
      "sup_loss": 0.6811490774337517
    },
    {
      "train_step": 1529,
      "sup_loss": 0.6964834857078144
    },
    {
      "train_step": 1530,
      "sup_loss": 0.46293559245162375
    },
    {
      "train_step": 1531,
      "sup_loss": 0.657789576083587
    },
    {
      "train_step": 1532,
      "sup_loss": 0.5963458812523178
    },
    {
      "train_step": 1533,
      "sup_loss": 0.7296258241746104
    },
    {
      "train_step": 1534,
      "sup_loss": 0.7692304776963358
    },
    {
      "train_step": 1535,
      "sup_loss": 0.542249955283239
    },
    {
      "train_step": 1536,
      "sup_loss": 0.7684365106748757
    },
    {
      "train_step": 1537,
      "sup_loss": 0.7522417452460161
    },
    {
      "train_step": 1538,
      "sup_loss": 0.6225269948168781
    },
    {
      "train_step": 1539,
      "sup_loss": 0.7302100464324467
    },
    {
      "train_step": 1540,
      "sup_loss": 0.7156170714359634
    },
    {
      "train_step": 1541,
      "sup_loss": 0.8045315182032243
    },
    {
      "train_step": 1542,
      "sup_loss": 0.5529948189804157
    },
    {
      "train_step": 1543,
      "sup_loss": 0.7548880111791259
    },
    {
      "train_step": 1544,
      "sup_loss": 0.41577541234418447
    },
    {
      "train_step": 1545,
      "sup_loss": 0.7136476062167741
    },
    {
      "train_step": 1546,
      "sup_loss": 0.5645869973732123
    },
    {
      "train_step": 1547,
      "sup_loss": 0.6243440930168247
    },
    {
      "train_step": 1548,
      "sup_loss": 0.7649103064444293
    },
    {
      "train_step": 1549,
      "sup_loss": 0.6443881472780039
    },
    {
      "train_step": 1550,
      "sup_loss": 0.7970357638153025
    },
    {
      "train_step": 1551,
      "sup_loss": 0.7406162741341515
    },
    {
      "train_step": 1552,
      "sup_loss": 0.7163408833882969
    },
    {
      "train_step": 1553,
      "sup_loss": 0.7588906925141522
    },
    {
      "train_step": 1554,
      "sup_loss": 0.5872874981843976
    },
    {
      "train_step": 1555,
      "sup_loss": 0.7705343269140539
    },
    {
      "train_step": 1556,
      "sup_loss": 0.6396990994285984
    },
    {
      "train_step": 1557,
      "sup_loss": 0.7592749112670768
    },
    {
      "train_step": 1558,
      "sup_loss": 0.6520964893987021
    },
    {
      "train_step": 1559,
      "sup_loss": 0.6066275518771883
    },
    {
      "train_step": 1560,
      "sup_loss": 0.42459640681713223
    },
    {
      "train_step": 1561,
      "sup_loss": 0.7993141501079575
    },
    {
      "train_step": 1562,
      "sup_loss": 0.7838521204347755
    },
    {
      "train_step": 1563,
      "sup_loss": 0.6485168741062902
    },
    {
      "train_step": 1564,
      "sup_loss": 0.8122472031174895
    },
    {
      "train_step": 1565,
      "sup_loss": 0.6348329511269459
    },
    {
      "train_step": 1566,
      "sup_loss": 0.5766103734270273
    },
    {
      "train_step": 1567,
      "sup_loss": 0.8339228857614054
    },
    {
      "train_step": 1568,
      "sup_loss": 0.5596751797391692
    },
    {
      "train_step": 1569,
      "sup_loss": 0.6315352194958207
    },
    {
      "train_step": 1570,
      "sup_loss": 0.6256869830582658
    },
    {
      "train_step": 1571,
      "sup_loss": 0.6792754346086075
    },
    {
      "train_step": 1572,
      "sup_loss": 0.7874569957697871
    },
    {
      "train_step": 1573,
      "sup_loss": 0.8100496074132844
    },
    {
      "train_step": 1574,
      "sup_loss": 0.6379197370959685
    },
    {
      "train_step": 1575,
      "sup_loss": 0.705837738316137
    },
    {
      "train_step": 1576,
      "sup_loss": 0.6101465515808843
    },
    {
      "train_step": 1577,
      "sup_loss": 0.7072788455879301
    },
    {
      "train_step": 1578,
      "sup_loss": 0.7755697273465131
    },
    {
      "train_step": 1579,
      "sup_loss": 0.6448280357873281
    },
    {
      "train_step": 1580,
      "sup_loss": 0.728279228880036
    },
    {
      "train_step": 1581,
      "sup_loss": 0.7440292065571394
    },
    {
      "train_step": 1582,
      "sup_loss": 0.5655718854942596
    },
    {
      "train_step": 1583,
      "sup_loss": 0.5676119056547371
    },
    {
      "train_step": 1584,
      "sup_loss": 0.7007078115373431
    },
    {
      "train_step": 1585,
      "sup_loss": 0.5954142123383528
    },
    {
      "train_step": 1586,
      "sup_loss": 0.687137348645462
    },
    {
      "train_step": 1587,
      "sup_loss": 0.7085941400441413
    },
    {
      "train_step": 1588,
      "sup_loss": 0.8122944645472795
    },
    {
      "train_step": 1589,
      "sup_loss": 0.7780048864136141
    },
    {
      "train_step": 1590,
      "sup_loss": 0.7882340826833656
    },
    {
      "train_step": 1591,
      "sup_loss": 0.6144546817218635
    },
    {
      "train_step": 1592,
      "sup_loss": 0.8141054152941581
    },
    {
      "train_step": 1593,
      "sup_loss": 0.6376583127280575
    },
    {
      "train_step": 1594,
      "sup_loss": 0.7045629863533256
    },
    {
      "train_step": 1595,
      "sup_loss": 0.6760865657829876
    },
    {
      "train_step": 1596,
      "sup_loss": 0.9611532028140779
    },
    {
      "train_step": 1597,
      "sup_loss": 0.46422263280694676
    },
    {
      "train_step": 1598,
      "sup_loss": 0.7877471627515067
    },
    {
      "train_step": 1599,
      "sup_loss": 0.7099624529904225
    },
    {
      "train_step": 1600,
      "sup_loss": 0.663341028776071
    },
    {
      "train_step": 1601,
      "sup_loss": 0.7496058904888712
    },
    {
      "train_step": 1602,
      "sup_loss": 0.7968313779960429
    },
    {
      "train_step": 1603,
      "sup_loss": 0.7871162975011283
    },
    {
      "train_step": 1604,
      "sup_loss": 0.5625032578432733
    },
    {
      "train_step": 1605,
      "sup_loss": 0.6094326814601511
    },
    {
      "train_step": 1606,
      "sup_loss": 0.4473668556954422
    },
    {
      "train_step": 1607,
      "sup_loss": 0.674278373812852
    },
    {
      "train_step": 1608,
      "sup_loss": 0.8209311153225458
    },
    {
      "train_step": 1609,
      "sup_loss": 0.931505426029448
    },
    {
      "train_step": 1610,
      "sup_loss": 0.610263539170025
    },
    {
      "train_step": 1611,
      "sup_loss": 0.7248996823681743
    },
    {
      "train_step": 1612,
      "sup_loss": 0.7280135072812387
    },
    {
      "train_step": 1613,
      "sup_loss": 0.6128290863175354
    },
    {
      "train_step": 1614,
      "sup_loss": 0.7892628988158492
    },
    {
      "train_step": 1615,
      "sup_loss": 0.7796729085670825
    },
    {
      "train_step": 1616,
      "sup_loss": 0.6698540903409117
    },
    {
      "train_step": 1617,
      "sup_loss": 0.5698381613124105
    },
    {
      "train_step": 1618,
      "sup_loss": 0.533567875526655
    },
    {
      "train_step": 1619,
      "sup_loss": 0.6853400587804348
    },
    {
      "train_step": 1620,
      "sup_loss": 0.8142576804259021
    },
    {
      "train_step": 1621,
      "sup_loss": 0.8830676134402455
    },
    {
      "train_step": 1622,
      "sup_loss": 0.7427347513686503
    },
    {
      "train_step": 1623,
      "sup_loss": 0.8335505349731119
    },
    {
      "train_step": 1624,
      "sup_loss": 0.8883647900949865
    },
    {
      "train_step": 1625,
      "sup_loss": 0.5343059088842097
    },
    {
      "train_step": 1626,
      "sup_loss": 0.6787771554341099
    },
    {
      "train_step": 1627,
      "sup_loss": 0.6146190669051121
    },
    {
      "train_step": 1628,
      "sup_loss": 0.8128882833868428
    },
    {
      "train_step": 1629,
      "sup_loss": 0.7618422989987128
    },
    {
      "train_step": 1630,
      "sup_loss": 0.7670457343733702
    },
    {
      "train_step": 1631,
      "sup_loss": 0.6380078881163094
    },
    {
      "train_step": 1632,
      "sup_loss": 0.7210415879234678
    },
    {
      "train_step": 1633,
      "sup_loss": 0.5787398208607729
    },
    {
      "train_step": 1634,
      "sup_loss": 0.8293339613896117
    },
    {
      "train_step": 1635,
      "sup_loss": 0.8629941975596659
    },
    {
      "train_step": 1636,
      "sup_loss": 0.8823292959195629
    },
    {
      "train_step": 1637,
      "sup_loss": 0.6998246213163529
    },
    {
      "train_step": 1638,
      "sup_loss": 0.715892338466876
    },
    {
      "train_step": 1639,
      "sup_loss": 0.40390454725814356
    },
    {
      "train_step": 1640,
      "sup_loss": 0.8220730979898958
    },
    {
      "train_step": 1641,
      "sup_loss": 0.771501376633116
    },
    {
      "train_step": 1642,
      "sup_loss": 0.7096048666943998
    },
    {
      "train_step": 1643,
      "sup_loss": 0.7867988613389468
    },
    {
      "train_step": 1644,
      "sup_loss": 0.839499570124492
    },
    {
      "train_step": 1645,
      "sup_loss": 0.6747657471052739
    },
    {
      "train_step": 1646,
      "sup_loss": 0.6541448658019972
    },
    {
      "train_step": 1647,
      "sup_loss": 0.6455260852464483
    },
    {
      "train_step": 1648,
      "sup_loss": 0.6694196514769191
    },
    {
      "train_step": 1649,
      "sup_loss": 0.6265669414688692
    },
    {
      "train_step": 1650,
      "sup_loss": 0.6938505600765633
    },
    {
      "train_step": 1651,
      "sup_loss": 0.7241162080324905
    },
    {
      "train_step": 1652,
      "sup_loss": 0.7214382555053047
    },
    {
      "train_step": 1653,
      "sup_loss": 0.7859911684272646
    },
    {
      "train_step": 1654,
      "sup_loss": 0.5675906206540734
    },
    {
      "train_step": 1655,
      "sup_loss": 0.5470305335208139
    },
    {
      "train_step": 1656,
      "sup_loss": 0.7312523568961469
    },
    {
      "train_step": 1657,
      "sup_loss": 0.49163304752707804
    },
    {
      "train_step": 1658,
      "sup_loss": 0.5873004166073174
    },
    {
      "train_step": 1659,
      "sup_loss": 0.6751621303417371
    },
    {
      "train_step": 1660,
      "sup_loss": 0.7658319212572385
    },
    {
      "train_step": 1661,
      "sup_loss": 0.7689783403435816
    },
    {
      "train_step": 1662,
      "sup_loss": 0.5382345531039566
    },
    {
      "train_step": 1663,
      "sup_loss": 0.7266994584403142
    },
    {
      "train_step": 1664,
      "sup_loss": 0.7771130689049564
    },
    {
      "train_step": 1665,
      "sup_loss": 0.7288836171501659
    },
    {
      "train_step": 1666,
      "sup_loss": 0.6869591071959189
    },
    {
      "train_step": 1667,
      "sup_loss": 0.7361139321050698
    },
    {
      "train_step": 1668,
      "sup_loss": 0.6471361600225707
    },
    {
      "train_step": 1669,
      "sup_loss": 0.6975984484226566
    },
    {
      "train_step": 1670,
      "sup_loss": 0.5609446037207566
    },
    {
      "train_step": 1671,
      "sup_loss": 0.8018422199174638
    },
    {
      "train_step": 1672,
      "sup_loss": 0.8591862509229368
    },
    {
      "train_step": 1673,
      "sup_loss": 0.8060425088108335
    },
    {
      "train_step": 1674,
      "sup_loss": 0.7949353884367979
    },
    {
      "train_step": 1675,
      "sup_loss": 0.3844322773333152
    },
    {
      "train_step": 1676,
      "sup_loss": 0.8638293050844137
    },
    {
      "train_step": 1677,
      "sup_loss": 0.8451020924786267
    },
    {
      "train_step": 1678,
      "sup_loss": 1.0139885471468688
    },
    {
      "train_step": 1679,
      "sup_loss": 0.5232249531429974
    },
    {
      "train_step": 1680,
      "sup_loss": 0.7765901848483281
    },
    {
      "train_step": 1681,
      "sup_loss": 0.7351598740240365
    },
    {
      "train_step": 1682,
      "sup_loss": 0.6493073724854151
    },
    {
      "train_step": 1683,
      "sup_loss": 0.5159715736435759
    },
    {
      "train_step": 1684,
      "sup_loss": 0.6249919563612906
    },
    {
      "train_step": 1685,
      "sup_loss": 0.6610177447960544
    },
    {
      "train_step": 1686,
      "sup_loss": 0.6736575083123715
    },
    {
      "train_step": 1687,
      "sup_loss": 0.5724128646131186
    },
    {
      "train_step": 1688,
      "sup_loss": 0.5449939548145363
    },
    {
      "train_step": 1689,
      "sup_loss": 0.7008567695472153
    },
    {
      "train_step": 1690,
      "sup_loss": 0.5279861927652024
    },
    {
      "train_step": 1691,
      "sup_loss": 0.5689443247029456
    },
    {
      "train_step": 1692,
      "sup_loss": 0.6996708997647159
    },
    {
      "train_step": 1693,
      "sup_loss": 0.5947930890361989
    },
    {
      "train_step": 1694,
      "sup_loss": 0.549537036987515
    },
    {
      "train_step": 1695,
      "sup_loss": 0.7544938856534931
    },
    {
      "train_step": 1696,
      "sup_loss": 0.7208360101698195
    },
    {
      "train_step": 1697,
      "sup_loss": 0.649746632557938
    },
    {
      "train_step": 1698,
      "sup_loss": 0.603065629796539
    },
    {
      "train_step": 1699,
      "sup_loss": 0.7173337960806646
    },
    {
      "train_step": 1700,
      "sup_loss": 0.791430056767483
    },
    {
      "train_step": 1701,
      "sup_loss": 0.5821112912765247
    },
    {
      "train_step": 1702,
      "sup_loss": 0.6023225312031844
    },
    {
      "train_step": 1703,
      "sup_loss": 0.7691847966839007
    },
    {
      "train_step": 1704,
      "sup_loss": 0.5621999094893291
    },
    {
      "train_step": 1705,
      "sup_loss": 0.7127901319616876
    },
    {
      "train_step": 1706,
      "sup_loss": 0.6444653652351349
    },
    {
      "train_step": 1707,
      "sup_loss": 0.638325196975814
    },
    {
      "train_step": 1708,
      "sup_loss": 0.6999807505310691
    },
    {
      "train_step": 1709,
      "sup_loss": 0.8082815545363959
    },
    {
      "train_step": 1710,
      "sup_loss": 0.8141032491113573
    },
    {
      "train_step": 1711,
      "sup_loss": 0.5974037725137133
    },
    {
      "train_step": 1712,
      "sup_loss": 0.7732676152400154
    },
    {
      "train_step": 1713,
      "sup_loss": 0.7808371538970013
    },
    {
      "train_step": 1714,
      "sup_loss": 0.7687703147230912
    },
    {
      "train_step": 1715,
      "sup_loss": 0.6311524492922381
    },
    {
      "train_step": 1716,
      "sup_loss": 0.7634840117222136
    },
    {
      "train_step": 1717,
      "sup_loss": 0.7248813747225924
    },
    {
      "train_step": 1718,
      "sup_loss": 0.6431935035344984
    },
    {
      "train_step": 1719,
      "sup_loss": 0.6570835670171757
    },
    {
      "train_step": 1720,
      "sup_loss": 0.6486959132174618
    },
    {
      "train_step": 1721,
      "sup_loss": 0.49291513937665526
    },
    {
      "train_step": 1722,
      "sup_loss": 0.7218728584712192
    },
    {
      "train_step": 1723,
      "sup_loss": 0.6274236615934587
    },
    {
      "train_step": 1724,
      "sup_loss": 0.7258361189541304
    },
    {
      "train_step": 1725,
      "sup_loss": 0.5654448869276281
    },
    {
      "train_step": 1726,
      "sup_loss": 0.8295817142956796
    },
    {
      "train_step": 1727,
      "sup_loss": 0.6759235628170495
    },
    {
      "train_step": 1728,
      "sup_loss": 0.602930262437926
    },
    {
      "train_step": 1729,
      "sup_loss": 0.548627802500315
    },
    {
      "train_step": 1730,
      "sup_loss": 0.6242697957885904
    },
    {
      "train_step": 1731,
      "sup_loss": 0.6283500517885483
    },
    {
      "train_step": 1732,
      "sup_loss": 0.6909977987832607
    },
    {
      "train_step": 1733,
      "sup_loss": 0.5231676757612508
    },
    {
      "train_step": 1734,
      "sup_loss": 0.6721184766757791
    },
    {
      "train_step": 1735,
      "sup_loss": 0.8003570744202965
    },
    {
      "train_step": 1736,
      "sup_loss": 0.6469404223857103
    },
    {
      "train_step": 1737,
      "sup_loss": 0.6032783228757075
    },
    {
      "train_step": 1738,
      "sup_loss": 0.626338722531047
    },
    {
      "train_step": 1739,
      "sup_loss": 0.649330059437646
    },
    {
      "train_step": 1740,
      "sup_loss": 0.7871711324384321
    },
    {
      "train_step": 1741,
      "sup_loss": 0.4903521229144537
    },
    {
      "train_step": 1742,
      "sup_loss": 0.5551356324291714
    },
    {
      "train_step": 1743,
      "sup_loss": 0.7332540530518177
    },
    {
      "train_step": 1744,
      "sup_loss": 0.9191287494182476
    },
    {
      "train_step": 1745,
      "sup_loss": 0.5013904153028387
    },
    {
      "train_step": 1746,
      "sup_loss": 0.7977850541787876
    },
    {
      "train_step": 1747,
      "sup_loss": 0.6452331768756955
    },
    {
      "train_step": 1748,
      "sup_loss": 0.7213247330098234
    },
    {
      "train_step": 1749,
      "sup_loss": 0.599510093671125
    },
    {
      "train_step": 1750,
      "sup_loss": 0.6943192649509445
    },
    {
      "train_step": 1751,
      "sup_loss": 0.690050489844005
    },
    {
      "train_step": 1752,
      "sup_loss": 0.4661012507431633
    },
    {
      "train_step": 1753,
      "sup_loss": 0.7170598630966095
    },
    {
      "train_step": 1754,
      "sup_loss": 0.4319634871039157
    },
    {
      "train_step": 1755,
      "sup_loss": 0.6052914458393546
    },
    {
      "train_step": 1756,
      "sup_loss": 0.7218842758537376
    },
    {
      "train_step": 1757,
      "sup_loss": 0.6633047812939078
    },
    {
      "train_step": 1758,
      "sup_loss": 0.7680294479660481
    },
    {
      "train_step": 1759,
      "sup_loss": 0.7347618175993371
    },
    {
      "train_step": 1760,
      "sup_loss": 0.675702611414132
    },
    {
      "train_step": 1761,
      "sup_loss": 0.8867795765752309
    },
    {
      "train_step": 1762,
      "sup_loss": 0.5328543488257952
    },
    {
      "train_step": 1763,
      "sup_loss": 0.6344763301432611
    },
    {
      "train_step": 1764,
      "sup_loss": 0.6770169730613276
    },
    {
      "train_step": 1765,
      "sup_loss": 0.7584555522657394
    },
    {
      "train_step": 1766,
      "sup_loss": 0.5773373962397476
    },
    {
      "train_step": 1767,
      "sup_loss": 0.6533766586021146
    },
    {
      "train_step": 1768,
      "sup_loss": 0.7295710048646215
    },
    {
      "train_step": 1769,
      "sup_loss": 0.7119695326320957
    },
    {
      "train_step": 1770,
      "sup_loss": 0.8468679509223302
    },
    {
      "train_step": 1771,
      "sup_loss": 0.645318256019965
    },
    {
      "train_step": 1772,
      "sup_loss": 0.6462464055611933
    },
    {
      "train_step": 1773,
      "sup_loss": 0.6557628581101393
    },
    {
      "train_step": 1774,
      "sup_loss": 0.7624997640504618
    },
    {
      "train_step": 1775,
      "sup_loss": 0.6795745687167078
    },
    {
      "train_step": 1776,
      "sup_loss": 0.7344399789627244
    },
    {
      "train_step": 1777,
      "sup_loss": 0.6513453168591324
    },
    {
      "train_step": 1778,
      "sup_loss": 0.8445676629092864
    },
    {
      "train_step": 1779,
      "sup_loss": 0.4977826116937918
    },
    {
      "train_step": 1780,
      "sup_loss": 0.8045811650719101
    },
    {
      "train_step": 1781,
      "sup_loss": 0.6145001058041473
    },
    {
      "train_step": 1782,
      "sup_loss": 0.8318233187162851
    },
    {
      "train_step": 1783,
      "sup_loss": 0.6490819090527387
    },
    {
      "train_step": 1784,
      "sup_loss": 0.8288170932259655
    },
    {
      "train_step": 1785,
      "sup_loss": 0.8510506506502121
    },
    {
      "train_step": 1786,
      "sup_loss": 0.7628782882924073
    },
    {
      "train_step": 1787,
      "sup_loss": 0.3877397459283562
    },
    {
      "train_step": 1788,
      "sup_loss": 0.8095541324904377
    },
    {
      "train_step": 1789,
      "sup_loss": 0.7047711713351886
    },
    {
      "train_step": 1790,
      "sup_loss": 0.6568099670494165
    },
    {
      "train_step": 1791,
      "sup_loss": 0.688733667955875
    },
    {
      "train_step": 1792,
      "sup_loss": 0.8383662572843898
    },
    {
      "train_step": 1793,
      "sup_loss": 0.6929705540302099
    },
    {
      "train_step": 1794,
      "sup_loss": 0.5376586305703602
    },
    {
      "train_step": 1795,
      "sup_loss": 0.7650252380213747
    },
    {
      "train_step": 1796,
      "sup_loss": 0.5242206388075338
    },
    {
      "train_step": 1797,
      "sup_loss": 0.7675301106598981
    },
    {
      "train_step": 1798,
      "sup_loss": 0.5432783184459512
    },
    {
      "train_step": 1799,
      "sup_loss": 0.7573605680659606
    },
    {
      "train_step": 1800,
      "sup_loss": 0.7133950726227293
    },
    {
      "train_step": 1801,
      "sup_loss": 0.702096394993068
    },
    {
      "train_step": 1802,
      "sup_loss": 0.6459256942018121
    },
    {
      "train_step": 1803,
      "sup_loss": 0.6052584793937423
    },
    {
      "train_step": 1804,
      "sup_loss": 0.8187378375486996
    },
    {
      "train_step": 1805,
      "sup_loss": 0.6172585971452713
    },
    {
      "train_step": 1806,
      "sup_loss": 0.6642537860797336
    },
    {
      "train_step": 1807,
      "sup_loss": 0.5928250181561083
    },
    {
      "train_step": 1808,
      "sup_loss": 0.910035405128248
    },
    {
      "train_step": 1809,
      "sup_loss": 0.6397526853832537
    },
    {
      "train_step": 1810,
      "sup_loss": 0.7228380081951412
    },
    {
      "train_step": 1811,
      "sup_loss": 0.6758662963204612
    },
    {
      "train_step": 1812,
      "sup_loss": 0.7998733283720949
    },
    {
      "train_step": 1813,
      "sup_loss": 0.6878676915908344
    },
    {
      "train_step": 1814,
      "sup_loss": 0.653416653467206
    },
    {
      "train_step": 1815,
      "sup_loss": 0.8471695930218517
    },
    {
      "train_step": 1816,
      "sup_loss": 0.3223603698803116
    },
    {
      "train_step": 1817,
      "sup_loss": 0.6689069951389484
    },
    {
      "train_step": 1818,
      "sup_loss": 0.5855776389982754
    },
    {
      "train_step": 1819,
      "sup_loss": 0.6474755259539842
    },
    {
      "train_step": 1820,
      "sup_loss": 0.6015251559499163
    },
    {
      "train_step": 1821,
      "sup_loss": 0.5967242682318734
    },
    {
      "train_step": 1822,
      "sup_loss": 0.7252055419335832
    },
    {
      "train_step": 1823,
      "sup_loss": 0.6718929986494194
    },
    {
      "train_step": 1824,
      "sup_loss": 0.8267054549404019
    },
    {
      "train_step": 1825,
      "sup_loss": 0.7127200078977236
    },
    {
      "train_step": 1826,
      "sup_loss": 0.6841883710786517
    },
    {
      "train_step": 1827,
      "sup_loss": 0.8529167686151665
    },
    {
      "train_step": 1828,
      "sup_loss": 0.3994798968458251
    },
    {
      "train_step": 1829,
      "sup_loss": 0.7264718999244184
    },
    {
      "train_step": 1830,
      "sup_loss": 0.5035304153036049
    },
    {
      "train_step": 1831,
      "sup_loss": 0.6128659364145085
    },
    {
      "train_step": 1832,
      "sup_loss": 0.8475563929237564
    },
    {
      "train_step": 1833,
      "sup_loss": 0.635445570360699
    },
    {
      "train_step": 1834,
      "sup_loss": 0.7784602700428752
    },
    {
      "train_step": 1835,
      "sup_loss": 0.7651502340592369
    },
    {
      "train_step": 1836,
      "sup_loss": 0.49380897554026215
    },
    {
      "train_step": 1837,
      "sup_loss": 0.7986205165255362
    },
    {
      "train_step": 1838,
      "sup_loss": 0.652992439985807
    },
    {
      "train_step": 1839,
      "sup_loss": 0.7198338724181874
    },
    {
      "train_step": 1840,
      "sup_loss": 0.7229241354290139
    },
    {
      "train_step": 1841,
      "sup_loss": 0.7213054855274381
    },
    {
      "train_step": 1842,
      "sup_loss": 0.6907644931616166
    },
    {
      "train_step": 1843,
      "sup_loss": 0.5885950510218038
    },
    {
      "train_step": 1844,
      "sup_loss": 0.5406525833902545
    },
    {
      "train_step": 1845,
      "sup_loss": 0.8181371275893725
    },
    {
      "train_step": 1846,
      "sup_loss": 0.5882321874806752
    },
    {
      "train_step": 1847,
      "sup_loss": 0.7238909127763415
    },
    {
      "train_step": 1848,
      "sup_loss": 0.8927859710395557
    },
    {
      "train_step": 1849,
      "sup_loss": 0.5220727574607745
    },
    {
      "train_step": 1850,
      "sup_loss": 0.505908245811671
    },
    {
      "train_step": 1851,
      "sup_loss": 0.6324420830240711
    },
    {
      "train_step": 1852,
      "sup_loss": 0.46870570034174486
    },
    {
      "train_step": 1853,
      "sup_loss": 0.865305513185657
    },
    {
      "train_step": 1854,
      "sup_loss": 0.553169883348942
    },
    {
      "train_step": 1855,
      "sup_loss": 0.42506589075463147
    },
    {
      "train_step": 1856,
      "sup_loss": 0.8209132626288717
    },
    {
      "train_step": 1857,
      "sup_loss": 0.44777858239282947
    },
    {
      "train_step": 1858,
      "sup_loss": 0.6298991783453952
    },
    {
      "train_step": 1859,
      "sup_loss": 0.4875905816163413
    },
    {
      "train_step": 1860,
      "sup_loss": 0.5962533002324318
    },
    {
      "train_step": 1861,
      "sup_loss": 0.6066330743416506
    },
    {
      "train_step": 1862,
      "sup_loss": 0.6594605345000789
    },
    {
      "train_step": 1863,
      "sup_loss": 0.7566770363822153
    },
    {
      "train_step": 1864,
      "sup_loss": 0.7113497643053387
    },
    {
      "train_step": 1865,
      "sup_loss": 0.6466648585609345
    },
    {
      "train_step": 1866,
      "sup_loss": 0.7952278911665956
    },
    {
      "train_step": 1867,
      "sup_loss": 0.7103368948519034
    },
    {
      "train_step": 1868,
      "sup_loss": 0.7899164387989664
    },
    {
      "train_step": 1869,
      "sup_loss": 0.6458794317995272
    },
    {
      "train_step": 1870,
      "sup_loss": 0.3369310182117449
    },
    {
      "train_step": 1871,
      "sup_loss": 0.7093676369187147
    },
    {
      "train_step": 1872,
      "sup_loss": 0.7401893120393154
    },
    {
      "train_step": 1873,
      "sup_loss": 0.6817742929137776
    },
    {
      "train_step": 1874,
      "sup_loss": 0.7551543603357799
    },
    {
      "train_step": 1875,
      "sup_loss": 0.6509510091893121
    },
    {
      "train_step": 1876,
      "sup_loss": 0.5451041339070722
    },
    {
      "train_step": 1877,
      "sup_loss": 0.7223675197198254
    },
    {
      "train_step": 1878,
      "sup_loss": 0.8303509542550775
    },
    {
      "train_step": 1879,
      "sup_loss": 0.39080571944688336
    },
    {
      "train_step": 1880,
      "sup_loss": 0.5534001107348192
    },
    {
      "train_step": 1881,
      "sup_loss": 0.7543758317727096
    },
    {
      "train_step": 1882,
      "sup_loss": 0.6570729131590456
    },
    {
      "train_step": 1883,
      "sup_loss": 0.6439918406199001
    },
    {
      "train_step": 1884,
      "sup_loss": 0.6279401300706772
    },
    {
      "train_step": 1885,
      "sup_loss": 0.8196592277518799
    },
    {
      "train_step": 1886,
      "sup_loss": 0.6704592229852133
    },
    {
      "train_step": 1887,
      "sup_loss": 0.7135602203286388
    },
    {
      "train_step": 1888,
      "sup_loss": 0.6276687077769381
    },
    {
      "train_step": 1889,
      "sup_loss": 0.6398881634291113
    },
    {
      "train_step": 1890,
      "sup_loss": 0.6615656882029497
    },
    {
      "train_step": 1891,
      "sup_loss": 0.4694843403681069
    },
    {
      "train_step": 1892,
      "sup_loss": 0.7276252767451896
    },
    {
      "train_step": 1893,
      "sup_loss": 0.6379373857573343
    },
    {
      "train_step": 1894,
      "sup_loss": 0.7019342341658958
    },
    {
      "train_step": 1895,
      "sup_loss": 0.6950015799900815
    },
    {
      "train_step": 1896,
      "sup_loss": 0.7305106401154856
    },
    {
      "train_step": 1897,
      "sup_loss": 0.49670898009537917
    },
    {
      "train_step": 1898,
      "sup_loss": 0.6775761071404176
    },
    {
      "train_step": 1899,
      "sup_loss": 0.7284097010898264
    },
    {
      "train_step": 1900,
      "sup_loss": 0.6428121410059083
    },
    {
      "train_step": 1901,
      "sup_loss": 0.6175251898833977
    },
    {
      "train_step": 1902,
      "sup_loss": 0.6760272842738948
    },
    {
      "train_step": 1903,
      "sup_loss": 0.8391086363951881
    },
    {
      "train_step": 1904,
      "sup_loss": 0.7446324407196698
    },
    {
      "train_step": 1905,
      "sup_loss": 0.7519471330277832
    },
    {
      "train_step": 1906,
      "sup_loss": 0.6050877916265553
    },
    {
      "train_step": 1907,
      "sup_loss": 0.7350571028137372
    },
    {
      "train_step": 1908,
      "sup_loss": 0.6290512607354254
    },
    {
      "train_step": 1909,
      "sup_loss": 0.8044711298617914
    },
    {
      "train_step": 1910,
      "sup_loss": 0.6112881087798233
    },
    {
      "train_step": 1911,
      "sup_loss": 0.7330991000816758
    },
    {
      "train_step": 1912,
      "sup_loss": 0.7702095471155754
    },
    {
      "train_step": 1913,
      "sup_loss": 0.6374054873646089
    },
    {
      "train_step": 1914,
      "sup_loss": 0.7179887908438151
    },
    {
      "train_step": 1915,
      "sup_loss": 0.7079143872364878
    },
    {
      "train_step": 1916,
      "sup_loss": 0.6850512392370746
    },
    {
      "train_step": 1917,
      "sup_loss": 0.7735986330824576
    },
    {
      "train_step": 1918,
      "sup_loss": 0.6577310371098849
    },
    {
      "train_step": 1919,
      "sup_loss": 0.5990315558524646
    },
    {
      "train_step": 1920,
      "sup_loss": 0.5976944589957786
    },
    {
      "train_step": 1921,
      "sup_loss": 0.6256854342786616
    },
    {
      "train_step": 1922,
      "sup_loss": 0.7929766780256673
    },
    {
      "train_step": 1923,
      "sup_loss": 0.5760028276362241
    },
    {
      "train_step": 1924,
      "sup_loss": 0.5895298636172372
    },
    {
      "train_step": 1925,
      "sup_loss": 0.5744223505267547
    },
    {
      "train_step": 1926,
      "sup_loss": 0.816724343059783
    },
    {
      "train_step": 1927,
      "sup_loss": 0.4263071610160089
    },
    {
      "train_step": 1928,
      "sup_loss": 0.629606166295013
    },
    {
      "train_step": 1929,
      "sup_loss": 0.7659540093263758
    },
    {
      "train_step": 1930,
      "sup_loss": 0.6464166540580424
    },
    {
      "train_step": 1931,
      "sup_loss": 0.6624135491225935
    },
    {
      "train_step": 1932,
      "sup_loss": 0.6465229564233061
    },
    {
      "train_step": 1933,
      "sup_loss": 0.647176694035799
    },
    {
      "train_step": 1934,
      "sup_loss": 0.6168305421729443
    },
    {
      "train_step": 1935,
      "sup_loss": 0.5193093377000035
    },
    {
      "train_step": 1936,
      "sup_loss": 0.5713210823759132
    },
    {
      "train_step": 1937,
      "sup_loss": 0.6834929392124536
    },
    {
      "train_step": 1938,
      "sup_loss": 0.49873606435218265
    },
    {
      "train_step": 1939,
      "sup_loss": 0.5396491439159598
    },
    {
      "train_step": 1940,
      "sup_loss": 0.8120600818996762
    },
    {
      "train_step": 1941,
      "sup_loss": 0.7814807604129386
    },
    {
      "train_step": 1942,
      "sup_loss": 0.5109472015351496
    },
    {
      "train_step": 1943,
      "sup_loss": 0.671971358556215
    },
    {
      "train_step": 1944,
      "sup_loss": 0.5292845931667427
    },
    {
      "train_step": 1945,
      "sup_loss": 0.7465428399127667
    },
    {
      "train_step": 1946,
      "sup_loss": 0.728275191285353
    },
    {
      "train_step": 1947,
      "sup_loss": 0.7201368964957626
    },
    {
      "train_step": 1948,
      "sup_loss": 0.4555191842755646
    },
    {
      "train_step": 1949,
      "sup_loss": 0.6469717412948839
    },
    {
      "train_step": 1950,
      "sup_loss": 0.5540614580986857
    },
    {
      "train_step": 1951,
      "sup_loss": 0.8244999885472296
    },
    {
      "train_step": 1952,
      "sup_loss": 0.2207311833362826
    },
    {
      "train_step": 1953,
      "sup_loss": 0.7206306504517409
    },
    {
      "train_step": 1954,
      "sup_loss": 0.5467653980609095
    },
    {
      "train_step": 1955,
      "sup_loss": 0.4237698232252681
    },
    {
      "train_step": 1956,
      "sup_loss": 0.2439935019030397
    },
    {
      "train_step": 1957,
      "sup_loss": 0.3637568551684358
    },
    {
      "train_step": 1958,
      "sup_loss": 0.7887475644351465
    },
    {
      "train_step": 1959,
      "sup_loss": 0.6860884501852693
    },
    {
      "train_step": 1960,
      "sup_loss": 0.6486102428509772
    },
    {
      "train_step": 1961,
      "sup_loss": 0.7169915085307459
    },
    {
      "train_step": 1962,
      "sup_loss": 0.44264463202448345
    },
    {
      "train_step": 1963,
      "sup_loss": 0.5085069088425129
    },
    {
      "train_step": 1964,
      "sup_loss": 0.7315974956796062
    },
    {
      "train_step": 1965,
      "sup_loss": 0.31603769845488827
    },
    {
      "train_step": 1966,
      "sup_loss": 0.866228437791291
    },
    {
      "train_step": 1967,
      "sup_loss": 0.3694347121598842
    },
    {
      "train_step": 1968,
      "sup_loss": 0.6730586362537766
    },
    {
      "train_step": 1969,
      "sup_loss": 0.6980272389841037
    },
    {
      "train_step": 1970,
      "sup_loss": 0.6850179843343075
    },
    {
      "train_step": 1971,
      "sup_loss": 0.5979589363489907
    },
    {
      "train_step": 1972,
      "sup_loss": 0.6907192881968798
    },
    {
      "train_step": 1973,
      "sup_loss": 0.6212838100973562
    },
    {
      "train_step": 1974,
      "sup_loss": 0.7198243721297515
    },
    {
      "train_step": 1975,
      "sup_loss": 0.7618696940400499
    },
    {
      "train_step": 1976,
      "sup_loss": 0.652464378123909
    },
    {
      "train_step": 1977,
      "sup_loss": 0.4209308545868279
    },
    {
      "train_step": 1978,
      "sup_loss": 0.562283390693488
    },
    {
      "train_step": 1979,
      "sup_loss": 0.5491363794767927
    },
    {
      "train_step": 1980,
      "sup_loss": 0.4646305695737298
    },
    {
      "train_step": 1981,
      "sup_loss": 0.7957676800351073
    },
    {
      "train_step": 1982,
      "sup_loss": 0.598087193900471
    },
    {
      "train_step": 1983,
      "sup_loss": 0.6041159878783975
    },
    {
      "train_step": 1984,
      "sup_loss": 0.5636798850586908
    },
    {
      "train_step": 1985,
      "sup_loss": 0.8039079903612913
    },
    {
      "train_step": 1986,
      "sup_loss": 0.6077856152079765
    },
    {
      "train_step": 1987,
      "sup_loss": 0.7360135482890646
    },
    {
      "train_step": 1988,
      "sup_loss": 0.4105941247078459
    },
    {
      "train_step": 1989,
      "sup_loss": 0.5546439318405326
    },
    {
      "train_step": 1990,
      "sup_loss": 0.7535694818100007
    },
    {
      "train_step": 1991,
      "sup_loss": 0.7066101992610939
    },
    {
      "train_step": 1992,
      "sup_loss": 0.6414249068667832
    },
    {
      "train_step": 1993,
      "sup_loss": 0.6217912241465066
    },
    {
      "train_step": 1994,
      "sup_loss": 0.7294800432250304
    },
    {
      "train_step": 1995,
      "sup_loss": 0.7880161114389638
    },
    {
      "train_step": 1996,
      "sup_loss": 0.6378896771072432
    },
    {
      "train_step": 1997,
      "sup_loss": 0.5702360308732909
    },
    {
      "train_step": 1998,
      "sup_loss": 0.5301866902840632
    },
    {
      "train_step": 1999,
      "sup_loss": 0.7795636332280927
    },
    {
      "train_step": 2000,
      "sup_loss": 0.6837570074676987
    },
    {
      "train_step": 2001,
      "sup_loss": 0.5837753989545799
    },
    {
      "train_step": 2002,
      "sup_loss": 0.49214769074068604
    },
    {
      "train_step": 2003,
      "sup_loss": 0.7033374767384909
    },
    {
      "train_step": 2004,
      "sup_loss": 0.6019770542709395
    },
    {
      "train_step": 2005,
      "sup_loss": 0.5263777734225489
    },
    {
      "train_step": 2006,
      "sup_loss": 0.726169093968347
    },
    {
      "train_step": 2007,
      "sup_loss": 0.748391925621037
    },
    {
      "train_step": 2008,
      "sup_loss": 0.713896649760474
    },
    {
      "train_step": 2009,
      "sup_loss": 0.6028722803714671
    },
    {
      "train_step": 2010,
      "sup_loss": 0.5120019796088146
    },
    {
      "train_step": 2011,
      "sup_loss": 0.29964463957900295
    },
    {
      "train_step": 2012,
      "sup_loss": 0.46338989461490354
    },
    {
      "train_step": 2013,
      "sup_loss": 0.7349462017871523
    },
    {
      "train_step": 2014,
      "sup_loss": 0.4533617833186881
    },
    {
      "train_step": 2015,
      "sup_loss": 0.8074894954246815
    },
    {
      "train_step": 2016,
      "sup_loss": 0.60864202075954
    },
    {
      "train_step": 2017,
      "sup_loss": 0.6711185645147004
    },
    {
      "train_step": 2018,
      "sup_loss": 0.6281426035657796
    },
    {
      "train_step": 2019,
      "sup_loss": 0.49927374617316433
    },
    {
      "train_step": 2020,
      "sup_loss": 0.7266180632748876
    },
    {
      "train_step": 2021,
      "sup_loss": 0.5362124204153492
    },
    {
      "train_step": 2022,
      "sup_loss": 0.7416359332068175
    },
    {
      "train_step": 2023,
      "sup_loss": 0.49689451250231714
    },
    {
      "train_step": 2024,
      "sup_loss": 0.5900763506801936
    },
    {
      "train_step": 2025,
      "sup_loss": 0.7868701849012198
    },
    {
      "train_step": 2026,
      "sup_loss": 0.6811361093759809
    },
    {
      "train_step": 2027,
      "sup_loss": 0.7287428368238065
    },
    {
      "train_step": 2028,
      "sup_loss": 0.630699316854487
    },
    {
      "train_step": 2029,
      "sup_loss": 0.4049409218140814
    },
    {
      "train_step": 2030,
      "sup_loss": 0.5502212483968549
    },
    {
      "train_step": 2031,
      "sup_loss": 0.8730355180649135
    },
    {
      "train_step": 2032,
      "sup_loss": 0.5533536300953757
    },
    {
      "train_step": 2033,
      "sup_loss": 0.6702539374574807
    },
    {
      "train_step": 2034,
      "sup_loss": 0.6028015851962667
    },
    {
      "train_step": 2035,
      "sup_loss": 0.6858173154585374
    },
    {
      "train_step": 2036,
      "sup_loss": 0.7014236555466343
    },
    {
      "train_step": 2037,
      "sup_loss": 0.7131834776831649
    },
    {
      "train_step": 2038,
      "sup_loss": 0.5972172877860139
    },
    {
      "train_step": 2039,
      "sup_loss": 0.6135786610372158
    },
    {
      "train_step": 2040,
      "sup_loss": 0.7810538332586006
    },
    {
      "train_step": 2041,
      "sup_loss": 0.7015135953632403
    },
    {
      "train_step": 2042,
      "sup_loss": 0.5874710855250163
    },
    {
      "train_step": 2043,
      "sup_loss": 0.7618990683074631
    },
    {
      "train_step": 2044,
      "sup_loss": 0.6771354411577989
    },
    {
      "train_step": 2045,
      "sup_loss": 0.62538027686441
    },
    {
      "train_step": 2046,
      "sup_loss": 0.8945134030858818
    },
    {
      "train_step": 2047,
      "sup_loss": 0.636890125015888
    },
    {
      "train_step": 2048,
      "sup_loss": 0.583810482996057
    },
    {
      "train_step": 2049,
      "sup_loss": 0.6530917891846912
    },
    {
      "train_step": 2050,
      "sup_loss": 0.5187169995426287
    },
    {
      "train_step": 2051,
      "sup_loss": 0.7404965184906512
    },
    {
      "train_step": 2052,
      "sup_loss": 0.5847541997134494
    },
    {
      "train_step": 2053,
      "sup_loss": 0.649427150395617
    },
    {
      "train_step": 2054,
      "sup_loss": 0.6790430876467688
    },
    {
      "train_step": 2055,
      "sup_loss": 0.6616503302745171
    },
    {
      "train_step": 2056,
      "sup_loss": 0.658563689300522
    },
    {
      "train_step": 2057,
      "sup_loss": 0.6937053572064226
    },
    {
      "train_step": 2058,
      "sup_loss": 0.5740551839377188
    },
    {
      "train_step": 2059,
      "sup_loss": 0.6989759672399423
    },
    {
      "train_step": 2060,
      "sup_loss": 0.6052438543314473
    },
    {
      "train_step": 2061,
      "sup_loss": 0.7589539240492885
    },
    {
      "train_step": 2062,
      "sup_loss": 0.7484252888373163
    },
    {
      "train_step": 2063,
      "sup_loss": 0.7649902689679882
    },
    {
      "train_step": 2064,
      "sup_loss": 0.4969363111532219
    },
    {
      "train_step": 2065,
      "sup_loss": 0.6220458892394541
    },
    {
      "train_step": 2066,
      "sup_loss": 0.5449575892772344
    },
    {
      "train_step": 2067,
      "sup_loss": 0.6256083873375706
    },
    {
      "train_step": 2068,
      "sup_loss": 0.6275134344438626
    },
    {
      "train_step": 2069,
      "sup_loss": 0.6382826980506981
    },
    {
      "train_step": 2070,
      "sup_loss": 0.5095510876238424
    },
    {
      "train_step": 2071,
      "sup_loss": 0.7776965044458172
    },
    {
      "train_step": 2072,
      "sup_loss": 0.6032734573668642
    },
    {
      "train_step": 2073,
      "sup_loss": 0.6429839719981175
    },
    {
      "train_step": 2074,
      "sup_loss": 0.7945763708422628
    },
    {
      "train_step": 2075,
      "sup_loss": 0.8079269740920976
    },
    {
      "train_step": 2076,
      "sup_loss": 0.8103598034960658
    },
    {
      "train_step": 2077,
      "sup_loss": 0.6081018871672205
    },
    {
      "train_step": 2078,
      "sup_loss": 0.4858266047296809
    },
    {
      "train_step": 2079,
      "sup_loss": 0.5000852496879656
    },
    {
      "train_step": 2080,
      "sup_loss": 0.47771578691294203
    },
    {
      "train_step": 2081,
      "sup_loss": 0.6209650877403677
    },
    {
      "train_step": 2082,
      "sup_loss": 0.7590844307344543
    },
    {
      "train_step": 2083,
      "sup_loss": 0.45920362167986506
    },
    {
      "train_step": 2084,
      "sup_loss": 0.611969270419029
    },
    {
      "train_step": 2085,
      "sup_loss": 0.6074695820740761
    },
    {
      "train_step": 2086,
      "sup_loss": 0.7161651684909007
    },
    {
      "train_step": 2087,
      "sup_loss": 0.8260734167107817
    },
    {
      "train_step": 2088,
      "sup_loss": 0.7374750837019466
    },
    {
      "train_step": 2089,
      "sup_loss": 0.7296709985695069
    },
    {
      "train_step": 2090,
      "sup_loss": 0.6861991797023163
    },
    {
      "train_step": 2091,
      "sup_loss": 0.626607020877277
    },
    {
      "train_step": 2092,
      "sup_loss": 0.7823418797677315
    },
    {
      "train_step": 2093,
      "sup_loss": 0.6122921417559439
    },
    {
      "train_step": 2094,
      "sup_loss": 0.5812363114758068
    },
    {
      "train_step": 2095,
      "sup_loss": 0.8374719603971622
    },
    {
      "train_step": 2096,
      "sup_loss": 0.5232329485339281
    },
    {
      "train_step": 2097,
      "sup_loss": 0.8381056348437417
    },
    {
      "train_step": 2098,
      "sup_loss": 0.6614776305892908
    },
    {
      "train_step": 2099,
      "sup_loss": 0.486723267962456
    },
    {
      "train_step": 2100,
      "sup_loss": 0.5366979682253781
    },
    {
      "train_step": 2101,
      "sup_loss": 0.7136089541473303
    },
    {
      "train_step": 2102,
      "sup_loss": 0.5859234842605023
    },
    {
      "train_step": 2103,
      "sup_loss": 0.6935899808783076
    },
    {
      "train_step": 2104,
      "sup_loss": 0.48941926611876796
    },
    {
      "train_step": 2105,
      "sup_loss": 0.5964110169677389
    },
    {
      "train_step": 2106,
      "sup_loss": 0.44190291688880184
    },
    {
      "train_step": 2107,
      "sup_loss": 0.6714228340183188
    },
    {
      "train_step": 2108,
      "sup_loss": 0.6010190043921444
    },
    {
      "train_step": 2109,
      "sup_loss": 0.8728627797250288
    },
    {
      "train_step": 2110,
      "sup_loss": 0.80926260123395
    },
    {
      "train_step": 2111,
      "sup_loss": 0.4879566365566146
    },
    {
      "train_step": 2112,
      "sup_loss": 0.6681417563778693
    },
    {
      "train_step": 2113,
      "sup_loss": 0.5262839030390178
    },
    {
      "train_step": 2114,
      "sup_loss": 0.7034355340281735
    },
    {
      "train_step": 2115,
      "sup_loss": 0.5522798994881861
    },
    {
      "train_step": 2116,
      "sup_loss": 0.7731919387838684
    },
    {
      "train_step": 2117,
      "sup_loss": 0.6521830357389717
    },
    {
      "train_step": 2118,
      "sup_loss": 0.7313754350092899
    },
    {
      "train_step": 2119,
      "sup_loss": 0.5754111399372019
    },
    {
      "train_step": 2120,
      "sup_loss": 0.5811257826861671
    },
    {
      "train_step": 2121,
      "sup_loss": 0.6847875664500952
    },
    {
      "train_step": 2122,
      "sup_loss": 0.6429477684621675
    },
    {
      "train_step": 2123,
      "sup_loss": 0.622082420684527
    },
    {
      "train_step": 2124,
      "sup_loss": 0.6819238242383104
    },
    {
      "train_step": 2125,
      "sup_loss": 0.5823616771818906
    },
    {
      "train_step": 2126,
      "sup_loss": 0.6589754763665363
    },
    {
      "train_step": 2127,
      "sup_loss": 0.8375983782843457
    },
    {
      "train_step": 2128,
      "sup_loss": 0.6835404587178732
    },
    {
      "train_step": 2129,
      "sup_loss": 0.6057918165695589
    },
    {
      "train_step": 2130,
      "sup_loss": 0.473253974090812
    },
    {
      "train_step": 2131,
      "sup_loss": 0.7520124344882144
    },
    {
      "train_step": 2132,
      "sup_loss": 0.6683116596624826
    },
    {
      "train_step": 2133,
      "sup_loss": 0.7105075865626731
    },
    {
      "train_step": 2134,
      "sup_loss": 0.6486301922188861
    },
    {
      "train_step": 2135,
      "sup_loss": 0.7284098142917367
    },
    {
      "train_step": 2136,
      "sup_loss": 0.5433888965463435
    },
    {
      "train_step": 2137,
      "sup_loss": 0.6884302964601637
    },
    {
      "train_step": 2138,
      "sup_loss": 0.42339162371491806
    },
    {
      "train_step": 2139,
      "sup_loss": 0.40769167352276836
    },
    {
      "train_step": 2140,
      "sup_loss": 0.6277584597515878
    },
    {
      "train_step": 2141,
      "sup_loss": 0.6577821379371142
    },
    {
      "train_step": 2142,
      "sup_loss": 0.6283813469961047
    },
    {
      "train_step": 2143,
      "sup_loss": 0.4495823022752246
    },
    {
      "train_step": 2144,
      "sup_loss": 0.6674394865613261
    },
    {
      "train_step": 2145,
      "sup_loss": 0.6755062911954625
    },
    {
      "train_step": 2146,
      "sup_loss": 0.6249914092118032
    },
    {
      "train_step": 2147,
      "sup_loss": 0.6446673192936279
    },
    {
      "train_step": 2148,
      "sup_loss": 0.6108336237977199
    },
    {
      "train_step": 2149,
      "sup_loss": 0.5692069462247878
    },
    {
      "train_step": 2150,
      "sup_loss": 0.44441543562375524
    },
    {
      "train_step": 2151,
      "sup_loss": 0.6010349481042009
    },
    {
      "train_step": 2152,
      "sup_loss": 0.8011591782902706
    },
    {
      "train_step": 2153,
      "sup_loss": 0.5983538737499483
    },
    {
      "train_step": 2154,
      "sup_loss": 0.7737861960878111
    },
    {
      "train_step": 2155,
      "sup_loss": 0.44729665779353484
    },
    {
      "train_step": 2156,
      "sup_loss": 0.5028132297836426
    },
    {
      "train_step": 2157,
      "sup_loss": 0.7781963397703643
    },
    {
      "train_step": 2158,
      "sup_loss": 0.6637788301410775
    },
    {
      "train_step": 2159,
      "sup_loss": 0.5000628978877757
    },
    {
      "train_step": 2160,
      "sup_loss": 0.848987627002458
    },
    {
      "train_step": 2161,
      "sup_loss": 0.733611995892352
    },
    {
      "train_step": 2162,
      "sup_loss": 0.6564250844932717
    },
    {
      "train_step": 2163,
      "sup_loss": 0.6768009243053852
    },
    {
      "train_step": 2164,
      "sup_loss": 0.6988906533168813
    },
    {
      "train_step": 2165,
      "sup_loss": 0.7293232165355208
    },
    {
      "train_step": 2166,
      "sup_loss": 0.599922982885849
    },
    {
      "train_step": 2167,
      "sup_loss": 0.5464401970394142
    },
    {
      "train_step": 2168,
      "sup_loss": 0.5716887266862873
    },
    {
      "train_step": 2169,
      "sup_loss": 0.7051345038548749
    },
    {
      "train_step": 2170,
      "sup_loss": 0.5886243783618857
    },
    {
      "train_step": 2171,
      "sup_loss": 0.7454488023627447
    },
    {
      "train_step": 2172,
      "sup_loss": 0.7659411583154865
    },
    {
      "train_step": 2173,
      "sup_loss": 0.6142714737250736
    },
    {
      "train_step": 2174,
      "sup_loss": 0.8273337280444331
    },
    {
      "train_step": 2175,
      "sup_loss": 0.4464387161079335
    },
    {
      "train_step": 2176,
      "sup_loss": 0.7680778998490194
    },
    {
      "train_step": 2177,
      "sup_loss": 0.7855600538262923
    },
    {
      "train_step": 2178,
      "sup_loss": 0.7548398878130602
    },
    {
      "train_step": 2179,
      "sup_loss": 0.5188955959589365
    },
    {
      "train_step": 2180,
      "sup_loss": 0.5436644356275282
    },
    {
      "train_step": 2181,
      "sup_loss": 0.6272894285278686
    },
    {
      "train_step": 2182,
      "sup_loss": 0.6230457343958123
    },
    {
      "train_step": 2183,
      "sup_loss": 0.6472838039610112
    },
    {
      "train_step": 2184,
      "sup_loss": 0.6957252287535468
    },
    {
      "train_step": 2185,
      "sup_loss": 0.5538601165877809
    },
    {
      "train_step": 2186,
      "sup_loss": 0.6787460850489484
    },
    {
      "train_step": 2187,
      "sup_loss": 0.5162330805435038
    },
    {
      "train_step": 2188,
      "sup_loss": 0.471678388095892
    },
    {
      "train_step": 2189,
      "sup_loss": 0.7220301728620458
    },
    {
      "train_step": 2190,
      "sup_loss": 0.668392650089328
    },
    {
      "train_step": 2191,
      "sup_loss": 0.47918248924089957
    },
    {
      "train_step": 2192,
      "sup_loss": 0.7509453558225493
    },
    {
      "train_step": 2193,
      "sup_loss": 0.488563249717012
    },
    {
      "train_step": 2194,
      "sup_loss": 0.6298980693013052
    },
    {
      "train_step": 2195,
      "sup_loss": 0.6805641801610192
    },
    {
      "train_step": 2196,
      "sup_loss": 0.6329329102242067
    },
    {
      "train_step": 2197,
      "sup_loss": 0.5281999921067974
    },
    {
      "train_step": 2198,
      "sup_loss": 0.6126404329589437
    },
    {
      "train_step": 2199,
      "sup_loss": 0.6451667207426665
    },
    {
      "train_step": 2200,
      "sup_loss": 0.7298136224587858
    },
    {
      "train_step": 2201,
      "sup_loss": 0.5049527825300089
    },
    {
      "train_step": 2202,
      "sup_loss": 0.5517516661484904
    },
    {
      "train_step": 2203,
      "sup_loss": 0.41509073133783336
    },
    {
      "train_step": 2204,
      "sup_loss": 0.5740564843384004
    },
    {
      "train_step": 2205,
      "sup_loss": 0.5991291249967242
    },
    {
      "train_step": 2206,
      "sup_loss": 0.5967064450543444
    },
    {
      "train_step": 2207,
      "sup_loss": 0.4891488256115718
    },
    {
      "train_step": 2208,
      "sup_loss": 0.6303285720697682
    },
    {
      "train_step": 2209,
      "sup_loss": 0.5794905297737135
    },
    {
      "train_step": 2210,
      "sup_loss": 0.7754331220146309
    },
    {
      "train_step": 2211,
      "sup_loss": 0.7317909363071015
    },
    {
      "train_step": 2212,
      "sup_loss": 0.630359343743768
    },
    {
      "train_step": 2213,
      "sup_loss": 0.6536614173533701
    },
    {
      "train_step": 2214,
      "sup_loss": 0.8195534608937549
    },
    {
      "train_step": 2215,
      "sup_loss": 0.7264553847330213
    },
    {
      "train_step": 2216,
      "sup_loss": 0.670550861447448
    },
    {
      "train_step": 2217,
      "sup_loss": 0.5486146742942066
    },
    {
      "train_step": 2218,
      "sup_loss": 0.5584556272287797
    },
    {
      "train_step": 2219,
      "sup_loss": 0.6496156870699497
    },
    {
      "train_step": 2220,
      "sup_loss": 0.7012950065550864
    },
    {
      "train_step": 2221,
      "sup_loss": 0.6464874757195516
    },
    {
      "train_step": 2222,
      "sup_loss": 0.6251584388815619
    },
    {
      "train_step": 2223,
      "sup_loss": 0.4114354583034645
    },
    {
      "train_step": 2224,
      "sup_loss": 0.6892106558463645
    },
    {
      "train_step": 2225,
      "sup_loss": 0.8128481571530939
    },
    {
      "train_step": 2226,
      "sup_loss": 0.6684309994191436
    },
    {
      "train_step": 2227,
      "sup_loss": 0.4984764840876647
    },
    {
      "train_step": 2228,
      "sup_loss": 0.8375750201949712
    },
    {
      "train_step": 2229,
      "sup_loss": 0.5734862556539374
    },
    {
      "train_step": 2230,
      "sup_loss": 0.5037461617391176
    },
    {
      "train_step": 2231,
      "sup_loss": 0.4697630408918736
    },
    {
      "train_step": 2232,
      "sup_loss": 0.719712720138969
    },
    {
      "train_step": 2233,
      "sup_loss": 0.720767143168868
    },
    {
      "train_step": 2234,
      "sup_loss": 0.7651872405275717
    },
    {
      "train_step": 2235,
      "sup_loss": 0.7656594750576866
    },
    {
      "train_step": 2236,
      "sup_loss": 0.5271863130819954
    },
    {
      "train_step": 2237,
      "sup_loss": 0.5294921110716718
    },
    {
      "train_step": 2238,
      "sup_loss": 0.4786491047394166
    },
    {
      "train_step": 2239,
      "sup_loss": 0.7801537535830672
    },
    {
      "train_step": 2240,
      "sup_loss": 0.7211168169354641
    },
    {
      "train_step": 2241,
      "sup_loss": 0.5495626949385689
    },
    {
      "train_step": 2242,
      "sup_loss": 0.7619731944564849
    },
    {
      "train_step": 2243,
      "sup_loss": 0.6497391078702579
    },
    {
      "train_step": 2244,
      "sup_loss": 0.33705366944240295
    },
    {
      "train_step": 2245,
      "sup_loss": 0.5494453356869904
    },
    {
      "train_step": 2246,
      "sup_loss": 0.46082619594633034
    },
    {
      "train_step": 2247,
      "sup_loss": 0.673860547068826
    },
    {
      "train_step": 2248,
      "sup_loss": 0.5938045296966998
    },
    {
      "train_step": 2249,
      "sup_loss": 0.8075384463892363
    },
    {
      "train_step": 2250,
      "sup_loss": 0.7043683645506622
    },
    {
      "train_step": 2251,
      "sup_loss": 0.8035734069417261
    },
    {
      "train_step": 2252,
      "sup_loss": 0.48979208801997776
    },
    {
      "train_step": 2253,
      "sup_loss": 0.7218490078553322
    },
    {
      "train_step": 2254,
      "sup_loss": 0.5308926290833479
    },
    {
      "train_step": 2255,
      "sup_loss": 0.6176039870679942
    },
    {
      "train_step": 2256,
      "sup_loss": 0.7174282493094428
    },
    {
      "train_step": 2257,
      "sup_loss": 0.6032503346703227
    },
    {
      "train_step": 2258,
      "sup_loss": 0.8559272746080324
    },
    {
      "train_step": 2259,
      "sup_loss": 0.4020483316480508
    },
    {
      "train_step": 2260,
      "sup_loss": 0.6654836642090909
    },
    {
      "train_step": 2261,
      "sup_loss": 0.6478427116224852
    },
    {
      "train_step": 2262,
      "sup_loss": 0.4755642441028041
    },
    {
      "train_step": 2263,
      "sup_loss": 0.742718122366861
    },
    {
      "train_step": 2264,
      "sup_loss": 0.6497587568428232
    },
    {
      "train_step": 2265,
      "sup_loss": 0.5953875258569267
    },
    {
      "train_step": 2266,
      "sup_loss": 0.48471428952453327
    },
    {
      "train_step": 2267,
      "sup_loss": 0.4595633089778924
    },
    {
      "train_step": 2268,
      "sup_loss": 0.5396376445079947
    },
    {
      "train_step": 2269,
      "sup_loss": 0.7019461571772202
    },
    {
      "train_step": 2270,
      "sup_loss": 0.4773969049634117
    },
    {
      "train_step": 2271,
      "sup_loss": 0.655616900694326
    },
    {
      "train_step": 2272,
      "sup_loss": 0.7323702110951424
    },
    {
      "train_step": 2273,
      "sup_loss": 0.7652655565220869
    },
    {
      "train_step": 2274,
      "sup_loss": 0.6297684844541488
    },
    {
      "train_step": 2275,
      "sup_loss": 0.7031827052570379
    },
    {
      "train_step": 2276,
      "sup_loss": 0.661391291869488
    },
    {
      "train_step": 2277,
      "sup_loss": 0.7233949679957982
    },
    {
      "train_step": 2278,
      "sup_loss": 0.5903306766545521
    },
    {
      "train_step": 2279,
      "sup_loss": 0.7497441707721589
    },
    {
      "train_step": 2280,
      "sup_loss": 0.5876333354344935
    },
    {
      "train_step": 2281,
      "sup_loss": 0.5281964479181631
    },
    {
      "train_step": 2282,
      "sup_loss": 0.5534663952924624
    },
    {
      "train_step": 2283,
      "sup_loss": 0.659686994669808
    },
    {
      "train_step": 2284,
      "sup_loss": 0.617534313149019
    },
    {
      "train_step": 2285,
      "sup_loss": 0.48881793874511126
    },
    {
      "train_step": 2286,
      "sup_loss": 0.7737458463588541
    },
    {
      "train_step": 2287,
      "sup_loss": 0.6357826284002139
    },
    {
      "train_step": 2288,
      "sup_loss": 0.6510304954951949
    },
    {
      "train_step": 2289,
      "sup_loss": 0.7194528554742289
    },
    {
      "train_step": 2290,
      "sup_loss": 0.7499904452056958
    },
    {
      "train_step": 2291,
      "sup_loss": 0.5743423891394324
    },
    {
      "train_step": 2292,
      "sup_loss": 0.5952696531249779
    },
    {
      "train_step": 2293,
      "sup_loss": 0.717684864367082
    },
    {
      "train_step": 2294,
      "sup_loss": 0.6654401923239226
    },
    {
      "train_step": 2295,
      "sup_loss": 0.6226000801560194
    },
    {
      "train_step": 2296,
      "sup_loss": 0.5733014467894043
    },
    {
      "train_step": 2297,
      "sup_loss": 0.7137687000435065
    },
    {
      "train_step": 2298,
      "sup_loss": 0.556953345710011
    },
    {
      "train_step": 2299,
      "sup_loss": 0.5490414783345859
    },
    {
      "train_step": 2300,
      "sup_loss": 0.9226985122053459
    },
    {
      "train_step": 2301,
      "sup_loss": 0.5411111890939893
    },
    {
      "train_step": 2302,
      "sup_loss": 0.40517397933815563
    },
    {
      "train_step": 2303,
      "sup_loss": 0.6406924631377304
    },
    {
      "train_step": 2304,
      "sup_loss": 0.663336969623592
    },
    {
      "train_step": 2305,
      "sup_loss": 0.4031082313277506
    },
    {
      "train_step": 2306,
      "sup_loss": 0.6453407777446722
    },
    {
      "train_step": 2307,
      "sup_loss": 0.6738990300102038
    },
    {
      "train_step": 2308,
      "sup_loss": 0.7248930992320687
    },
    {
      "train_step": 2309,
      "sup_loss": 0.8930032623156142
    },
    {
      "train_step": 2310,
      "sup_loss": 0.5957613568117358
    },
    {
      "train_step": 2311,
      "sup_loss": 0.5713527373153003
    },
    {
      "train_step": 2312,
      "sup_loss": 0.4626086675336606
    },
    {
      "train_step": 2313,
      "sup_loss": 0.6840018917485551
    },
    {
      "train_step": 2314,
      "sup_loss": 0.6738907380694034
    },
    {
      "train_step": 2315,
      "sup_loss": 0.710465377374571
    },
    {
      "train_step": 2316,
      "sup_loss": 0.7213797693735348
    },
    {
      "train_step": 2317,
      "sup_loss": 0.5783917401786788
    },
    {
      "train_step": 2318,
      "sup_loss": 0.6968589688095326
    },
    {
      "train_step": 2319,
      "sup_loss": 0.7257737733223564
    },
    {
      "train_step": 2320,
      "sup_loss": 0.7266092991483157
    },
    {
      "train_step": 2321,
      "sup_loss": 0.7613263646178963
    },
    {
      "train_step": 2322,
      "sup_loss": 0.6373638378031278
    },
    {
      "train_step": 2323,
      "sup_loss": 0.7581346018118926
    },
    {
      "train_step": 2324,
      "sup_loss": 0.6894089959628618
    },
    {
      "train_step": 2325,
      "sup_loss": 0.4846468810204398
    },
    {
      "train_step": 2326,
      "sup_loss": 0.7432698777528696
    },
    {
      "train_step": 2327,
      "sup_loss": 0.7848204457450157
    },
    {
      "train_step": 2328,
      "sup_loss": 0.6445854404913774
    },
    {
      "train_step": 2329,
      "sup_loss": 0.5870179066066253
    },
    {
      "train_step": 2330,
      "sup_loss": 0.7318957006221226
    },
    {
      "train_step": 2331,
      "sup_loss": 0.92141284250419
    },
    {
      "train_step": 2332,
      "sup_loss": 0.7440965110813693
    },
    {
      "train_step": 2333,
      "sup_loss": 0.6453859512223921
    },
    {
      "train_step": 2334,
      "sup_loss": 0.6892577710804728
    },
    {
      "train_step": 2335,
      "sup_loss": 0.6146104442234692
    },
    {
      "train_step": 2336,
      "sup_loss": 0.8341675459285371
    },
    {
      "train_step": 2337,
      "sup_loss": 0.49791470995948595
    },
    {
      "train_step": 2338,
      "sup_loss": 0.48951569218617763
    },
    {
      "train_step": 2339,
      "sup_loss": 0.4707256355850002
    },
    {
      "train_step": 2340,
      "sup_loss": 0.6315926411781404
    },
    {
      "train_step": 2341,
      "sup_loss": 0.5629069042071964
    },
    {
      "train_step": 2342,
      "sup_loss": 0.7157455342277935
    },
    {
      "train_step": 2343,
      "sup_loss": 0.6790655078383951
    },
    {
      "train_step": 2344,
      "sup_loss": 0.7713476133848228
    },
    {
      "train_step": 2345,
      "sup_loss": 0.6087855556442734
    },
    {
      "train_step": 2346,
      "sup_loss": 0.7536446941597716
    },
    {
      "train_step": 2347,
      "sup_loss": 0.794780453068783
    },
    {
      "train_step": 2348,
      "sup_loss": 0.7238957444893339
    },
    {
      "train_step": 2349,
      "sup_loss": 0.5944731006482081
    },
    {
      "train_step": 2350,
      "sup_loss": 0.5806933461174268
    },
    {
      "train_step": 2351,
      "sup_loss": 0.5860830565406469
    },
    {
      "train_step": 2352,
      "sup_loss": 0.8188760287787483
    },
    {
      "train_step": 2353,
      "sup_loss": 0.6637541366065413
    },
    {
      "train_step": 2354,
      "sup_loss": 0.7525395691988713
    },
    {
      "train_step": 2355,
      "sup_loss": 0.7355922352199659
    },
    {
      "train_step": 2356,
      "sup_loss": 0.5677937394175107
    },
    {
      "train_step": 2357,
      "sup_loss": 0.44883134704420913
    },
    {
      "train_step": 2358,
      "sup_loss": 0.8144095925150138
    },
    {
      "train_step": 2359,
      "sup_loss": 0.42337953821969976
    },
    {
      "train_step": 2360,
      "sup_loss": 0.6132068788937624
    },
    {
      "train_step": 2361,
      "sup_loss": 0.6978026729520902
    },
    {
      "train_step": 2362,
      "sup_loss": 0.6110688222824099
    },
    {
      "train_step": 2363,
      "sup_loss": 0.5758425903773245
    },
    {
      "train_step": 2364,
      "sup_loss": 0.795732813825237
    },
    {
      "train_step": 2365,
      "sup_loss": 0.5100241866545733
    },
    {
      "train_step": 2366,
      "sup_loss": 0.7051723288221672
    },
    {
      "train_step": 2367,
      "sup_loss": 0.6146993497357295
    },
    {
      "train_step": 2368,
      "sup_loss": 0.5117361733181176
    },
    {
      "train_step": 2369,
      "sup_loss": 0.7292030537549954
    },
    {
      "train_step": 2370,
      "sup_loss": 0.5250887162855307
    },
    {
      "train_step": 2371,
      "sup_loss": 0.6615839015418661
    },
    {
      "train_step": 2372,
      "sup_loss": 0.5866066928122486
    },
    {
      "train_step": 2373,
      "sup_loss": 0.4390671509057742
    },
    {
      "train_step": 2374,
      "sup_loss": 0.6586021720449404
    },
    {
      "train_step": 2375,
      "sup_loss": 0.6049297981320036
    },
    {
      "train_step": 2376,
      "sup_loss": 0.831432530830162
    },
    {
      "train_step": 2377,
      "sup_loss": 0.770475415213635
    },
    {
      "train_step": 2378,
      "sup_loss": 0.4047652448088251
    },
    {
      "train_step": 2379,
      "sup_loss": 0.5273836721795232
    },
    {
      "train_step": 2380,
      "sup_loss": 0.5215936128184174
    },
    {
      "train_step": 2381,
      "sup_loss": 0.5577301819704976
    },
    {
      "train_step": 2382,
      "sup_loss": 0.6553788667495533
    },
    {
      "train_step": 2383,
      "sup_loss": 0.7336951818681412
    },
    {
      "train_step": 2384,
      "sup_loss": 0.508217857379247
    },
    {
      "train_step": 2385,
      "sup_loss": 0.7268344471225524
    },
    {
      "train_step": 2386,
      "sup_loss": 0.6618877746826213
    },
    {
      "train_step": 2387,
      "sup_loss": 0.6593806574650343
    },
    {
      "train_step": 2388,
      "sup_loss": 0.5829256729443473
    },
    {
      "train_step": 2389,
      "sup_loss": 0.6212788554094427
    },
    {
      "train_step": 2390,
      "sup_loss": 0.43626577892096213
    },
    {
      "train_step": 2391,
      "sup_loss": 0.5630679457799679
    },
    {
      "train_step": 2392,
      "sup_loss": 0.6846951036390629
    },
    {
      "train_step": 2393,
      "sup_loss": 0.7241102370864144
    },
    {
      "train_step": 2394,
      "sup_loss": 0.6184926224789065
    },
    {
      "train_step": 2395,
      "sup_loss": 0.7865071323280984
    },
    {
      "train_step": 2396,
      "sup_loss": 0.5526034807220584
    },
    {
      "train_step": 2397,
      "sup_loss": 0.7949812134186505
    },
    {
      "train_step": 2398,
      "sup_loss": 0.6609090489401973
    },
    {
      "train_step": 2399,
      "sup_loss": 0.5974987513281083
    },
    {
      "train_step": 2400,
      "sup_loss": 0.9009847693007281
    },
    {
      "train_step": 2401,
      "sup_loss": 0.6838965480528385
    },
    {
      "train_step": 2402,
      "sup_loss": 0.5475797147768088
    },
    {
      "train_step": 2403,
      "sup_loss": 0.7201352730620012
    },
    {
      "train_step": 2404,
      "sup_loss": 0.47956197819431223
    },
    {
      "train_step": 2405,
      "sup_loss": 0.7616776212352052
    },
    {
      "train_step": 2406,
      "sup_loss": 0.6471987495653218
    },
    {
      "train_step": 2407,
      "sup_loss": 0.768509465904676
    },
    {
      "train_step": 2408,
      "sup_loss": 0.5047326057540946
    },
    {
      "train_step": 2409,
      "sup_loss": 0.6576567815974371
    },
    {
      "train_step": 2410,
      "sup_loss": 0.6776277682763144
    },
    {
      "train_step": 2411,
      "sup_loss": 0.6824437336221603
    },
    {
      "train_step": 2412,
      "sup_loss": 0.6215981670810439
    },
    {
      "train_step": 2413,
      "sup_loss": 0.48553051220789273
    },
    {
      "train_step": 2414,
      "sup_loss": 0.5632191468752519
    },
    {
      "train_step": 2415,
      "sup_loss": 0.7610635997420048
    },
    {
      "train_step": 2416,
      "sup_loss": 0.7240161017329314
    },
    {
      "train_step": 2417,
      "sup_loss": 0.7347131961659921
    },
    {
      "train_step": 2418,
      "sup_loss": 0.6928425857269491
    },
    {
      "train_step": 2419,
      "sup_loss": 0.7651506573519625
    },
    {
      "train_step": 2420,
      "sup_loss": 0.8433661662320772
    },
    {
      "train_step": 2421,
      "sup_loss": 0.6230864821462461
    },
    {
      "train_step": 2422,
      "sup_loss": 0.5566853561982574
    },
    {
      "train_step": 2423,
      "sup_loss": 0.8046977313265202
    },
    {
      "train_step": 2424,
      "sup_loss": 0.6022245615306464
    },
    {
      "train_step": 2425,
      "sup_loss": 0.6933176104580829
    },
    {
      "train_step": 2426,
      "sup_loss": 0.4591096691445025
    },
    {
      "train_step": 2427,
      "sup_loss": 0.7490136967668306
    },
    {
      "train_step": 2428,
      "sup_loss": 0.4279710040663242
    },
    {
      "train_step": 2429,
      "sup_loss": 0.6142968384052642
    },
    {
      "train_step": 2430,
      "sup_loss": 0.5804306779350765
    },
    {
      "train_step": 2431,
      "sup_loss": 0.7666885014613324
    },
    {
      "train_step": 2432,
      "sup_loss": 0.7443509857489911
    },
    {
      "train_step": 2433,
      "sup_loss": 0.5199393889186763
    },
    {
      "train_step": 2434,
      "sup_loss": 0.6401969210763033
    },
    {
      "train_step": 2435,
      "sup_loss": 0.790472159144916
    },
    {
      "train_step": 2436,
      "sup_loss": 0.636445630792594
    },
    {
      "train_step": 2437,
      "sup_loss": 0.8758494367387024
    },
    {
      "train_step": 2438,
      "sup_loss": 0.8309221695683733
    },
    {
      "train_step": 2439,
      "sup_loss": 0.6454861640766933
    },
    {
      "train_step": 2440,
      "sup_loss": 0.6635109305863609
    },
    {
      "train_step": 2441,
      "sup_loss": 0.6868306424970699
    },
    {
      "train_step": 2442,
      "sup_loss": 0.7225871396612766
    },
    {
      "train_step": 2443,
      "sup_loss": 0.6839155226142882
    },
    {
      "train_step": 2444,
      "sup_loss": 0.779580751901034
    },
    {
      "train_step": 2445,
      "sup_loss": 0.4400365269917994
    },
    {
      "train_step": 2446,
      "sup_loss": 0.8012394975849
    },
    {
      "train_step": 2447,
      "sup_loss": 0.684353245462348
    },
    {
      "train_step": 2448,
      "sup_loss": 0.6376999644968703
    },
    {
      "train_step": 2449,
      "sup_loss": 0.662759341223482
    },
    {
      "train_step": 2450,
      "sup_loss": 0.6557894391737592
    },
    {
      "train_step": 2451,
      "sup_loss": 0.759808382596576
    },
    {
      "train_step": 2452,
      "sup_loss": 0.6847005197747761
    },
    {
      "train_step": 2453,
      "sup_loss": 0.5893471435112414
    },
    {
      "train_step": 2454,
      "sup_loss": 0.6532995940168665
    },
    {
      "train_step": 2455,
      "sup_loss": 0.5936043043905759
    },
    {
      "train_step": 2456,
      "sup_loss": 0.5849871384249534
    },
    {
      "train_step": 2457,
      "sup_loss": 0.5189708540842495
    },
    {
      "train_step": 2458,
      "sup_loss": 0.5261542204108907
    },
    {
      "train_step": 2459,
      "sup_loss": 0.39028691250029185
    },
    {
      "train_step": 2460,
      "sup_loss": 0.7002570046495942
    },
    {
      "train_step": 2461,
      "sup_loss": 0.5184357429425912
    },
    {
      "train_step": 2462,
      "sup_loss": 0.4090063488798857
    },
    {
      "train_step": 2463,
      "sup_loss": 0.5593518530614926
    },
    {
      "train_step": 2464,
      "sup_loss": 0.7024798601337171
    },
    {
      "train_step": 2465,
      "sup_loss": 0.6440408793272463
    },
    {
      "train_step": 2466,
      "sup_loss": 0.4818677070464231
    },
    {
      "train_step": 2467,
      "sup_loss": 0.61726421465142
    },
    {
      "train_step": 2468,
      "sup_loss": 0.4092376736126799
    },
    {
      "train_step": 2469,
      "sup_loss": 0.8165504364298324
    },
    {
      "train_step": 2470,
      "sup_loss": 0.6410193504153591
    },
    {
      "train_step": 2471,
      "sup_loss": 0.7979185666041402
    },
    {
      "train_step": 2472,
      "sup_loss": 0.5316718519653658
    },
    {
      "train_step": 2473,
      "sup_loss": 0.7066017606280485
    },
    {
      "train_step": 2474,
      "sup_loss": 0.7425868184138551
    },
    {
      "train_step": 2475,
      "sup_loss": 0.7981872023699417
    },
    {
      "train_step": 2476,
      "sup_loss": 0.6916476195255935
    },
    {
      "train_step": 2477,
      "sup_loss": 0.7117191268685377
    },
    {
      "train_step": 2478,
      "sup_loss": 0.8467479551721232
    },
    {
      "train_step": 2479,
      "sup_loss": 0.6645980600460921
    },
    {
      "train_step": 2480,
      "sup_loss": 0.7006649334843781
    },
    {
      "train_step": 2481,
      "sup_loss": 0.7325801680118901
    },
    {
      "train_step": 2482,
      "sup_loss": 0.6284073452777182
    },
    {
      "train_step": 2483,
      "sup_loss": 0.615355251037954
    },
    {
      "train_step": 2484,
      "sup_loss": 0.7563701658282388
    },
    {
      "train_step": 2485,
      "sup_loss": 0.6816634083791258
    },
    {
      "train_step": 2486,
      "sup_loss": 0.6784260991774694
    },
    {
      "train_step": 2487,
      "sup_loss": 0.8355700110000878
    },
    {
      "train_step": 2488,
      "sup_loss": 0.5878443806038063
    },
    {
      "train_step": 2489,
      "sup_loss": 0.700893751117447
    },
    {
      "train_step": 2490,
      "sup_loss": 0.5292692726852735
    },
    {
      "train_step": 2491,
      "sup_loss": 0.7904782365784147
    },
    {
      "train_step": 2492,
      "sup_loss": 0.5490373756015653
    },
    {
      "train_step": 2493,
      "sup_loss": 0.6766016733477981
    },
    {
      "train_step": 2494,
      "sup_loss": 0.706604078107249
    },
    {
      "train_step": 2495,
      "sup_loss": 0.8325703748337404
    },
    {
      "train_step": 2496,
      "sup_loss": 0.5500343102142882
    },
    {
      "train_step": 2497,
      "sup_loss": 0.48180535243083955
    },
    {
      "train_step": 2498,
      "sup_loss": 0.7434767537259536
    },
    {
      "train_step": 2499,
      "sup_loss": 0.6995437420074627
    },
    {
      "train_step": 2500,
      "sup_loss": 0.801025580966725
    },
    {
      "train_step": 2501,
      "sup_loss": 0.519235840353456
    },
    {
      "train_step": 2502,
      "sup_loss": 0.7237829658680097
    },
    {
      "train_step": 2503,
      "sup_loss": 0.6995557242924523
    },
    {
      "train_step": 2504,
      "sup_loss": 0.5757099607107299
    },
    {
      "train_step": 2505,
      "sup_loss": 0.6541268263009585
    },
    {
      "train_step": 2506,
      "sup_loss": 0.6663970613010127
    },
    {
      "train_step": 2507,
      "sup_loss": 0.684603228286169
    },
    {
      "train_step": 2508,
      "sup_loss": 0.5974639552088676
    },
    {
      "train_step": 2509,
      "sup_loss": 0.7271031083996343
    },
    {
      "train_step": 2510,
      "sup_loss": 0.6788266526260812
    },
    {
      "train_step": 2511,
      "sup_loss": 0.7707080466774838
    },
    {
      "train_step": 2512,
      "sup_loss": 0.6988665878724047
    },
    {
      "train_step": 2513,
      "sup_loss": 0.6612562954707593
    },
    {
      "train_step": 2514,
      "sup_loss": 0.699767055873058
    },
    {
      "train_step": 2515,
      "sup_loss": 0.6654687792176366
    },
    {
      "train_step": 2516,
      "sup_loss": 0.5661737950978797
    },
    {
      "train_step": 2517,
      "sup_loss": 0.7185829657722186
    },
    {
      "train_step": 2518,
      "sup_loss": 0.6988081466522009
    },
    {
      "train_step": 2519,
      "sup_loss": 0.6974998778886683
    },
    {
      "train_step": 2520,
      "sup_loss": 0.8112388777475079
    },
    {
      "train_step": 2521,
      "sup_loss": 0.6629238073569661
    },
    {
      "train_step": 2522,
      "sup_loss": 0.7618205666732206
    },
    {
      "train_step": 2523,
      "sup_loss": 0.7356470949872104
    },
    {
      "train_step": 2524,
      "sup_loss": 0.655782017031838
    },
    {
      "train_step": 2525,
      "sup_loss": 0.7512362978236486
    },
    {
      "train_step": 2526,
      "sup_loss": 0.5134073695568621
    },
    {
      "train_step": 2527,
      "sup_loss": 0.6856985460860545
    },
    {
      "train_step": 2528,
      "sup_loss": 0.6040797517693034
    },
    {
      "train_step": 2529,
      "sup_loss": 0.7961004077865328
    },
    {
      "train_step": 2530,
      "sup_loss": 0.6292208876855798
    },
    {
      "train_step": 2531,
      "sup_loss": 0.8131603441666773
    },
    {
      "train_step": 2532,
      "sup_loss": 0.6089740027402436
    },
    {
      "train_step": 2533,
      "sup_loss": 0.6856744122686826
    },
    {
      "train_step": 2534,
      "sup_loss": 0.5542063341913598
    },
    {
      "train_step": 2535,
      "sup_loss": 0.7366223899880341
    },
    {
      "train_step": 2536,
      "sup_loss": 0.7336317153319145
    },
    {
      "train_step": 2537,
      "sup_loss": 0.6878685452365885
    },
    {
      "train_step": 2538,
      "sup_loss": 0.6707926055161613
    },
    {
      "train_step": 2539,
      "sup_loss": 0.5471323197662759
    },
    {
      "train_step": 2540,
      "sup_loss": 0.7305956555252384
    },
    {
      "train_step": 2541,
      "sup_loss": 0.6774860341226767
    },
    {
      "train_step": 2542,
      "sup_loss": 0.5690290038305567
    },
    {
      "train_step": 2543,
      "sup_loss": 0.8206477478139019
    },
    {
      "train_step": 2544,
      "sup_loss": 0.5377461036281068
    },
    {
      "train_step": 2545,
      "sup_loss": 0.6860276411375207
    },
    {
      "train_step": 2546,
      "sup_loss": 0.4997985483338385
    },
    {
      "train_step": 2547,
      "sup_loss": 0.20485278662964698
    },
    {
      "train_step": 2548,
      "sup_loss": 0.41941607591239266
    },
    {
      "train_step": 2549,
      "sup_loss": 0.5875147387057553
    },
    {
      "train_step": 2550,
      "sup_loss": 0.7381397402883337
    },
    {
      "train_step": 2551,
      "sup_loss": 0.49190339088994467
    },
    {
      "train_step": 2552,
      "sup_loss": 0.7390630180891228
    },
    {
      "train_step": 2553,
      "sup_loss": 0.8225184601822267
    },
    {
      "train_step": 2554,
      "sup_loss": 0.4512563338708354
    },
    {
      "train_step": 2555,
      "sup_loss": 0.7153238028344968
    },
    {
      "train_step": 2556,
      "sup_loss": 0.45680809162940894
    },
    {
      "train_step": 2557,
      "sup_loss": 0.5696045194582435
    },
    {
      "train_step": 2558,
      "sup_loss": 0.6628125672334396
    },
    {
      "train_step": 2559,
      "sup_loss": 0.6250105547863498
    },
    {
      "train_step": 2560,
      "sup_loss": 0.37250967341817637
    },
    {
      "train_step": 2561,
      "sup_loss": 0.5412618875550301
    },
    {
      "train_step": 2562,
      "sup_loss": 0.5968945283107012
    },
    {
      "train_step": 2563,
      "sup_loss": 0.5914355591258268
    },
    {
      "train_step": 2564,
      "sup_loss": 0.7272671102254745
    },
    {
      "train_step": 2565,
      "sup_loss": 0.5422119538544287
    },
    {
      "train_step": 2566,
      "sup_loss": 0.6078240175438568
    },
    {
      "train_step": 2567,
      "sup_loss": 0.8419049580663382
    },
    {
      "train_step": 2568,
      "sup_loss": 0.6850260075012765
    },
    {
      "train_step": 2569,
      "sup_loss": 0.6812312883841776
    },
    {
      "train_step": 2570,
      "sup_loss": 0.6293671822646043
    },
    {
      "train_step": 2571,
      "sup_loss": 0.7283212857322371
    },
    {
      "train_step": 2572,
      "sup_loss": 0.7701858083610756
    },
    {
      "train_step": 2573,
      "sup_loss": 0.7238394852596364
    },
    {
      "train_step": 2574,
      "sup_loss": 0.8781075960337736
    },
    {
      "train_step": 2575,
      "sup_loss": 0.5138874016492347
    },
    {
      "train_step": 2576,
      "sup_loss": 0.585706688670204
    },
    {
      "train_step": 2577,
      "sup_loss": 0.4081487333949653
    },
    {
      "train_step": 2578,
      "sup_loss": 0.6895311379317095
    },
    {
      "train_step": 2579,
      "sup_loss": 0.41851058715222017
    },
    {
      "train_step": 2580,
      "sup_loss": 0.6235057737454502
    },
    {
      "train_step": 2581,
      "sup_loss": 0.6700485240232366
    },
    {
      "train_step": 2582,
      "sup_loss": 0.48926915882354693
    },
    {
      "train_step": 2583,
      "sup_loss": 0.8039920371324947
    },
    {
      "train_step": 2584,
      "sup_loss": 0.8374009763298373
    },
    {
      "train_step": 2585,
      "sup_loss": 0.3380051433218945
    },
    {
      "train_step": 2586,
      "sup_loss": 0.5446051695404889
    },
    {
      "train_step": 2587,
      "sup_loss": 0.8816509313888954
    },
    {
      "train_step": 2588,
      "sup_loss": 0.6761955763080421
    },
    {
      "train_step": 2589,
      "sup_loss": 0.5633777421878966
    },
    {
      "train_step": 2590,
      "sup_loss": 0.4685840045148756
    },
    {
      "train_step": 2591,
      "sup_loss": 0.48690124262550666
    },
    {
      "train_step": 2592,
      "sup_loss": 0.6619463242664649
    },
    {
      "train_step": 2593,
      "sup_loss": 0.5360223669850911
    },
    {
      "train_step": 2594,
      "sup_loss": 0.6053782600672237
    },
    {
      "train_step": 2595,
      "sup_loss": 0.7256911232974038
    },
    {
      "train_step": 2596,
      "sup_loss": 0.6107377285758506
    },
    {
      "train_step": 2597,
      "sup_loss": 0.7804217086086306
    },
    {
      "train_step": 2598,
      "sup_loss": 0.6189707366806452
    },
    {
      "train_step": 2599,
      "sup_loss": 0.7064727872500958
    },
    {
      "train_step": 2600,
      "sup_loss": 0.5942190893519929
    },
    {
      "train_step": 2601,
      "sup_loss": 0.5865696289395389
    },
    {
      "train_step": 2602,
      "sup_loss": 0.39060138128407595
    },
    {
      "train_step": 2603,
      "sup_loss": 0.42767557403711187
    },
    {
      "train_step": 2604,
      "sup_loss": 0.6168655937629935
    },
    {
      "train_step": 2605,
      "sup_loss": 0.5740426866105757
    },
    {
      "train_step": 2606,
      "sup_loss": 0.8376808616984622
    },
    {
      "train_step": 2607,
      "sup_loss": 0.5532797627419932
    },
    {
      "train_step": 2608,
      "sup_loss": 0.6045749374692518
    },
    {
      "train_step": 2609,
      "sup_loss": 0.5583916910213781
    },
    {
      "train_step": 2610,
      "sup_loss": 0.586412843447781
    },
    {
      "train_step": 2611,
      "sup_loss": 0.47863424161609386
    },
    {
      "train_step": 2612,
      "sup_loss": 0.4983000643257756
    },
    {
      "train_step": 2613,
      "sup_loss": 0.672593738946031
    },
    {
      "train_step": 2614,
      "sup_loss": 0.5213347895089084
    },
    {
      "train_step": 2615,
      "sup_loss": 0.7003876248574997
    },
    {
      "train_step": 2616,
      "sup_loss": 0.5054145357666343
    },
    {
      "train_step": 2617,
      "sup_loss": 0.3890095143894237
    },
    {
      "train_step": 2618,
      "sup_loss": 0.43223513046116224
    },
    {
      "train_step": 2619,
      "sup_loss": 0.7825517956356132
    },
    {
      "train_step": 2620,
      "sup_loss": 0.6157993005415064
    },
    {
      "train_step": 2621,
      "sup_loss": 0.7001642147431134
    },
    {
      "train_step": 2622,
      "sup_loss": 0.6805668027961915
    },
    {
      "train_step": 2623,
      "sup_loss": 0.6455310479410747
    },
    {
      "train_step": 2624,
      "sup_loss": 0.6259714610078908
    },
    {
      "train_step": 2625,
      "sup_loss": 0.579443044547895
    },
    {
      "train_step": 2626,
      "sup_loss": 0.5819080752496006
    },
    {
      "train_step": 2627,
      "sup_loss": 0.634834356170805
    },
    {
      "train_step": 2628,
      "sup_loss": 0.6806586125504399
    },
    {
      "train_step": 2629,
      "sup_loss": 0.603972365789849
    },
    {
      "train_step": 2630,
      "sup_loss": 0.5527903509364155
    },
    {
      "train_step": 2631,
      "sup_loss": 0.7329466546336911
    },
    {
      "train_step": 2632,
      "sup_loss": 0.8101208099758678
    },
    {
      "train_step": 2633,
      "sup_loss": 0.5619569993696859
    },
    {
      "train_step": 2634,
      "sup_loss": 0.4837654349105138
    },
    {
      "train_step": 2635,
      "sup_loss": 0.5712710288122758
    },
    {
      "train_step": 2636,
      "sup_loss": 0.764566393762132
    },
    {
      "train_step": 2637,
      "sup_loss": 0.5686907860507902
    },
    {
      "train_step": 2638,
      "sup_loss": 0.6313433362635256
    },
    {
      "train_step": 2639,
      "sup_loss": 0.6327463094044896
    },
    {
      "train_step": 2640,
      "sup_loss": 0.692287582687186
    },
    {
      "train_step": 2641,
      "sup_loss": 0.5932867035308305
    },
    {
      "train_step": 2642,
      "sup_loss": 0.4598525095781799
    },
    {
      "train_step": 2643,
      "sup_loss": 0.381307145020757
    },
    {
      "train_step": 2644,
      "sup_loss": 0.8174690578344452
    },
    {
      "train_step": 2645,
      "sup_loss": 0.6696823017232391
    },
    {
      "train_step": 2646,
      "sup_loss": 0.45833261344534915
    },
    {
      "train_step": 2647,
      "sup_loss": 0.8616358216430018
    },
    {
      "train_step": 2648,
      "sup_loss": 0.5457676549113366
    },
    {
      "train_step": 2649,
      "sup_loss": 0.6310096402063166
    },
    {
      "train_step": 2650,
      "sup_loss": 0.718204277578035
    },
    {
      "train_step": 2651,
      "sup_loss": 0.6710031364544491
    },
    {
      "train_step": 2652,
      "sup_loss": 0.6727253724830198
    },
    {
      "train_step": 2653,
      "sup_loss": 0.7280125238581164
    },
    {
      "train_step": 2654,
      "sup_loss": 0.39421531990338826
    },
    {
      "train_step": 2655,
      "sup_loss": 0.7550262097126685
    },
    {
      "train_step": 2656,
      "sup_loss": 0.5725097388444551
    },
    {
      "train_step": 2657,
      "sup_loss": 0.785631217452178
    },
    {
      "train_step": 2658,
      "sup_loss": 0.5894489021594289
    },
    {
      "train_step": 2659,
      "sup_loss": 0.5905095877507942
    },
    {
      "train_step": 2660,
      "sup_loss": 0.8384143536226273
    },
    {
      "train_step": 2661,
      "sup_loss": 0.6373644132996762
    },
    {
      "train_step": 2662,
      "sup_loss": 0.5930756107823214
    },
    {
      "train_step": 2663,
      "sup_loss": 0.8656779041376204
    },
    {
      "train_step": 2664,
      "sup_loss": 0.4024241754729902
    },
    {
      "train_step": 2665,
      "sup_loss": 0.42088983708720984
    },
    {
      "train_step": 2666,
      "sup_loss": 0.641596970471117
    },
    {
      "train_step": 2667,
      "sup_loss": 0.8077579249828871
    },
    {
      "train_step": 2668,
      "sup_loss": 0.3494576753827884
    },
    {
      "train_step": 2669,
      "sup_loss": 0.7123001167001669
    },
    {
      "train_step": 2670,
      "sup_loss": 0.5411514408634359
    },
    {
      "train_step": 2671,
      "sup_loss": 0.6421546507614433
    },
    {
      "train_step": 2672,
      "sup_loss": 0.8755816605926602
    },
    {
      "train_step": 2673,
      "sup_loss": 0.44655694304168086
    },
    {
      "train_step": 2674,
      "sup_loss": 0.49436693227739614
    },
    {
      "train_step": 2675,
      "sup_loss": 0.5733955861687839
    },
    {
      "train_step": 2676,
      "sup_loss": 0.7770467572786419
    },
    {
      "train_step": 2677,
      "sup_loss": 0.7542356410752632
    },
    {
      "train_step": 2678,
      "sup_loss": 0.6317303660669229
    },
    {
      "train_step": 2679,
      "sup_loss": 0.4858005601875223
    },
    {
      "train_step": 2680,
      "sup_loss": 0.6862267195830576
    },
    {
      "train_step": 2681,
      "sup_loss": 0.5657771868246398
    },
    {
      "train_step": 2682,
      "sup_loss": 0.498573071460449
    },
    {
      "train_step": 2683,
      "sup_loss": 0.6052040415669444
    },
    {
      "train_step": 2684,
      "sup_loss": 0.7325864824594924
    },
    {
      "train_step": 2685,
      "sup_loss": 0.6312836564803701
    },
    {
      "train_step": 2686,
      "sup_loss": 0.69525358924437
    },
    {
      "train_step": 2687,
      "sup_loss": 0.3993082008369663
    },
    {
      "train_step": 2688,
      "sup_loss": 0.730618172442563
    },
    {
      "train_step": 2689,
      "sup_loss": 0.5369864697652266
    },
    {
      "train_step": 2690,
      "sup_loss": 0.47291343263927876
    },
    {
      "train_step": 2691,
      "sup_loss": 0.78057455718831
    },
    {
      "train_step": 2692,
      "sup_loss": 0.41827722567666165
    },
    {
      "train_step": 2693,
      "sup_loss": 0.6123769741269518
    },
    {
      "train_step": 2694,
      "sup_loss": 0.5273755385732064
    },
    {
      "train_step": 2695,
      "sup_loss": 0.6434781724275325
    },
    {
      "train_step": 2696,
      "sup_loss": 0.5964791093469736
    },
    {
      "train_step": 2697,
      "sup_loss": 0.7126182671311821
    },
    {
      "train_step": 2698,
      "sup_loss": 0.6776987472507168
    },
    {
      "train_step": 2699,
      "sup_loss": 0.6222820534514928
    },
    {
      "train_step": 2700,
      "sup_loss": 0.7997088586734793
    },
    {
      "train_step": 2701,
      "sup_loss": 0.7201222080850446
    },
    {
      "train_step": 2702,
      "sup_loss": 0.6905915637758797
    },
    {
      "train_step": 2703,
      "sup_loss": 0.6956108529004722
    },
    {
      "train_step": 2704,
      "sup_loss": 0.6089909197761475
    },
    {
      "train_step": 2705,
      "sup_loss": 0.6339454359120364
    },
    {
      "train_step": 2706,
      "sup_loss": 0.5424879788431375
    },
    {
      "train_step": 2707,
      "sup_loss": 0.6005078814814568
    },
    {
      "train_step": 2708,
      "sup_loss": 0.6866909401467415
    },
    {
      "train_step": 2709,
      "sup_loss": 0.6423995743835357
    },
    {
      "train_step": 2710,
      "sup_loss": 0.7132932764221228
    },
    {
      "train_step": 2711,
      "sup_loss": 0.6558028751790355
    },
    {
      "train_step": 2712,
      "sup_loss": 0.8767963669590989
    },
    {
      "train_step": 2713,
      "sup_loss": 0.6610443831766568
    },
    {
      "train_step": 2714,
      "sup_loss": 0.6406615936713861
    },
    {
      "train_step": 2715,
      "sup_loss": 0.6058065641585657
    },
    {
      "train_step": 2716,
      "sup_loss": 0.49877878582483914
    },
    {
      "train_step": 2717,
      "sup_loss": 0.8282028026151939
    },
    {
      "train_step": 2718,
      "sup_loss": 0.7629971743797526
    },
    {
      "train_step": 2719,
      "sup_loss": 0.676554368991735
    },
    {
      "train_step": 2720,
      "sup_loss": 0.651807200214594
    },
    {
      "train_step": 2721,
      "sup_loss": 0.6779280234969703
    },
    {
      "train_step": 2722,
      "sup_loss": 0.6812861591876935
    },
    {
      "train_step": 2723,
      "sup_loss": 0.7074836183864365
    },
    {
      "train_step": 2724,
      "sup_loss": 0.5699174162946455
    },
    {
      "train_step": 2725,
      "sup_loss": 0.5461171111255277
    },
    {
      "train_step": 2726,
      "sup_loss": 0.8624567042611339
    },
    {
      "train_step": 2727,
      "sup_loss": 0.5899869385579583
    },
    {
      "train_step": 2728,
      "sup_loss": 0.596618684558505
    },
    {
      "train_step": 2729,
      "sup_loss": 0.4609781614703586
    },
    {
      "train_step": 2730,
      "sup_loss": 0.6919860511202218
    },
    {
      "train_step": 2731,
      "sup_loss": 0.5806853986064024
    },
    {
      "train_step": 2732,
      "sup_loss": 0.5189135971711076
    },
    {
      "train_step": 2733,
      "sup_loss": 0.4083401139664534
    },
    {
      "train_step": 2734,
      "sup_loss": 0.4554851089338642
    },
    {
      "train_step": 2735,
      "sup_loss": 0.8507286657362605
    },
    {
      "train_step": 2736,
      "sup_loss": 0.5074976384347996
    },
    {
      "train_step": 2737,
      "sup_loss": 0.5048867339769343
    },
    {
      "train_step": 2738,
      "sup_loss": 0.6185633773851651
    },
    {
      "train_step": 2739,
      "sup_loss": 0.6574709768004708
    },
    {
      "train_step": 2740,
      "sup_loss": 0.557544772940616
    },
    {
      "train_step": 2741,
      "sup_loss": 0.7387658723893504
    },
    {
      "train_step": 2742,
      "sup_loss": 0.6069565923148724
    },
    {
      "train_step": 2743,
      "sup_loss": 0.7987450472198051
    },
    {
      "train_step": 2744,
      "sup_loss": 0.6118405340149344
    },
    {
      "train_step": 2745,
      "sup_loss": 0.7068022376651736
    },
    {
      "train_step": 2746,
      "sup_loss": 0.5081419448699322
    },
    {
      "train_step": 2747,
      "sup_loss": 0.8363900287127971
    },
    {
      "train_step": 2748,
      "sup_loss": 0.5999824364822188
    },
    {
      "train_step": 2749,
      "sup_loss": 0.6957439788663653
    },
    {
      "train_step": 2750,
      "sup_loss": 0.5491842034210144
    },
    {
      "train_step": 2751,
      "sup_loss": 0.490837812981976
    },
    {
      "train_step": 2752,
      "sup_loss": 0.4901712748559511
    },
    {
      "train_step": 2753,
      "sup_loss": 0.7869582661463385
    },
    {
      "train_step": 2754,
      "sup_loss": 0.8367934720590384
    },
    {
      "train_step": 2755,
      "sup_loss": 0.6074623487888291
    },
    {
      "train_step": 2756,
      "sup_loss": 0.4717045517394145
    },
    {
      "train_step": 2757,
      "sup_loss": 0.3841296831018469
    },
    {
      "train_step": 2758,
      "sup_loss": 0.6609443681690342
    },
    {
      "train_step": 2759,
      "sup_loss": 0.6836817263377981
    },
    {
      "train_step": 2760,
      "sup_loss": 0.75054382979879
    },
    {
      "train_step": 2761,
      "sup_loss": 0.6414576060680208
    },
    {
      "train_step": 2762,
      "sup_loss": 0.4357954567266458
    },
    {
      "train_step": 2763,
      "sup_loss": 0.6238204630904868
    },
    {
      "train_step": 2764,
      "sup_loss": 0.4038883510595918
    },
    {
      "train_step": 2765,
      "sup_loss": 0.6924555201682341
    },
    {
      "train_step": 2766,
      "sup_loss": 0.3959209131983403
    },
    {
      "train_step": 2767,
      "sup_loss": 0.49969881193321386
    },
    {
      "train_step": 2768,
      "sup_loss": 0.5482782831805183
    },
    {
      "train_step": 2769,
      "sup_loss": 0.6214230558704823
    },
    {
      "train_step": 2770,
      "sup_loss": 0.45014518844148205
    },
    {
      "train_step": 2771,
      "sup_loss": 0.5015939021163249
    },
    {
      "train_step": 2772,
      "sup_loss": 0.7143693564036819
    },
    {
      "train_step": 2773,
      "sup_loss": 0.6823097861296584
    },
    {
      "train_step": 2774,
      "sup_loss": 0.6948922951348052
    },
    {
      "train_step": 2775,
      "sup_loss": 0.8174539285213053
    },
    {
      "train_step": 2776,
      "sup_loss": 0.5826253646154804
    },
    {
      "train_step": 2777,
      "sup_loss": 0.5558826181660501
    },
    {
      "train_step": 2778,
      "sup_loss": 0.5718627862352463
    },
    {
      "train_step": 2779,
      "sup_loss": 0.7201447932648424
    },
    {
      "train_step": 2780,
      "sup_loss": 0.739456320181997
    },
    {
      "train_step": 2781,
      "sup_loss": 0.6540896253583619
    },
    {
      "train_step": 2782,
      "sup_loss": 0.5977961405671237
    },
    {
      "train_step": 2783,
      "sup_loss": 0.7776675197007317
    },
    {
      "train_step": 2784,
      "sup_loss": 0.6563777430125021
    },
    {
      "train_step": 2785,
      "sup_loss": 0.6371862787203253
    },
    {
      "train_step": 2786,
      "sup_loss": 0.7939938823180503
    },
    {
      "train_step": 2787,
      "sup_loss": 0.5086772687805319
    },
    {
      "train_step": 2788,
      "sup_loss": 0.7451954766332307
    },
    {
      "train_step": 2789,
      "sup_loss": 0.6126036809405263
    },
    {
      "train_step": 2790,
      "sup_loss": 0.7476412738957355
    },
    {
      "train_step": 2791,
      "sup_loss": 0.7913320779431087
    },
    {
      "train_step": 2792,
      "sup_loss": 0.7876599210335321
    },
    {
      "train_step": 2793,
      "sup_loss": 0.640598162025736
    },
    {
      "train_step": 2794,
      "sup_loss": 0.5435059028250132
    },
    {
      "train_step": 2795,
      "sup_loss": 0.7102738001781124
    },
    {
      "train_step": 2796,
      "sup_loss": 0.5569828009624725
    },
    {
      "train_step": 2797,
      "sup_loss": 0.72110302595323
    },
    {
      "train_step": 2798,
      "sup_loss": 0.765897035232323
    },
    {
      "train_step": 2799,
      "sup_loss": 0.5621634922382863
    },
    {
      "train_step": 2800,
      "sup_loss": 0.6488391268878511
    },
    {
      "train_step": 2801,
      "sup_loss": 0.6306105122479286
    },
    {
      "train_step": 2802,
      "sup_loss": 0.5573703523547169
    },
    {
      "train_step": 2803,
      "sup_loss": 0.5040683195969694
    },
    {
      "train_step": 2804,
      "sup_loss": 0.5030393925612767
    },
    {
      "train_step": 2805,
      "sup_loss": 0.6595885220746925
    },
    {
      "train_step": 2806,
      "sup_loss": 0.5641178199467234
    },
    {
      "train_step": 2807,
      "sup_loss": 0.667032303467545
    },
    {
      "train_step": 2808,
      "sup_loss": 0.5209639133405759
    },
    {
      "train_step": 2809,
      "sup_loss": 0.687288608594781
    },
    {
      "train_step": 2810,
      "sup_loss": 0.7173768437514954
    },
    {
      "train_step": 2811,
      "sup_loss": 0.6097772380727914
    },
    {
      "train_step": 2812,
      "sup_loss": 0.5975565876822733
    },
    {
      "train_step": 2813,
      "sup_loss": 0.5856635518981482
    },
    {
      "train_step": 2814,
      "sup_loss": 0.5632686637572167
    },
    {
      "train_step": 2815,
      "sup_loss": 0.6864488828793962
    },
    {
      "train_step": 2816,
      "sup_loss": 0.6052249879786962
    },
    {
      "train_step": 2817,
      "sup_loss": 0.6057394272749398
    },
    {
      "train_step": 2818,
      "sup_loss": 0.3913923574534715
    },
    {
      "train_step": 2819,
      "sup_loss": 0.7814213799232202
    },
    {
      "train_step": 2820,
      "sup_loss": 0.5518556036287561
    },
    {
      "train_step": 2821,
      "sup_loss": 0.7173635894309425
    },
    {
      "train_step": 2822,
      "sup_loss": 0.7753972342043527
    },
    {
      "train_step": 2823,
      "sup_loss": 0.5158776453222695
    },
    {
      "train_step": 2824,
      "sup_loss": 0.7848957658785802
    },
    {
      "train_step": 2825,
      "sup_loss": 0.7029134304710825
    },
    {
      "train_step": 2826,
      "sup_loss": 0.7192814699726752
    },
    {
      "train_step": 2827,
      "sup_loss": 0.6220991375601095
    },
    {
      "train_step": 2828,
      "sup_loss": 0.8619003710357591
    },
    {
      "train_step": 2829,
      "sup_loss": 0.49573575895273486
    },
    {
      "train_step": 2830,
      "sup_loss": 0.635068265477363
    },
    {
      "train_step": 2831,
      "sup_loss": 0.7057516708561661
    },
    {
      "train_step": 2832,
      "sup_loss": 0.6187779630786094
    },
    {
      "train_step": 2833,
      "sup_loss": 0.5993557144377997
    },
    {
      "train_step": 2834,
      "sup_loss": 0.6147249681585779
    },
    {
      "train_step": 2835,
      "sup_loss": 0.504396289172601
    },
    {
      "train_step": 2836,
      "sup_loss": 0.38595511709463143
    },
    {
      "train_step": 2837,
      "sup_loss": 0.5507635141581961
    },
    {
      "train_step": 2838,
      "sup_loss": 0.43975353175943327
    },
    {
      "train_step": 2839,
      "sup_loss": 0.2730176919898672
    },
    {
      "train_step": 2840,
      "sup_loss": 0.6838321288091201
    },
    {
      "train_step": 2841,
      "sup_loss": 0.6810558775353983
    },
    {
      "train_step": 2842,
      "sup_loss": 0.7421328679835143
    },
    {
      "train_step": 2843,
      "sup_loss": 0.5954585049921814
    },
    {
      "train_step": 2844,
      "sup_loss": 0.4548584631137965
    },
    {
      "train_step": 2845,
      "sup_loss": 0.5507162964978719
    },
    {
      "train_step": 2846,
      "sup_loss": 0.5132541971519831
    },
    {
      "train_step": 2847,
      "sup_loss": 0.4552624439176847
    },
    {
      "train_step": 2848,
      "sup_loss": 0.5542547579108211
    },
    {
      "train_step": 2849,
      "sup_loss": 0.3806227538938741
    },
    {
      "train_step": 2850,
      "sup_loss": 0.6365288306103246
    },
    {
      "train_step": 2851,
      "sup_loss": 0.6760688383056424
    },
    {
      "train_step": 2852,
      "sup_loss": 0.5857953894432779
    },
    {
      "train_step": 2853,
      "sup_loss": 0.6459428349632275
    },
    {
      "train_step": 2854,
      "sup_loss": 0.3433893726238128
    },
    {
      "train_step": 2855,
      "sup_loss": 0.5543991941606946
    },
    {
      "train_step": 2856,
      "sup_loss": 0.6275675400544792
    },
    {
      "train_step": 2857,
      "sup_loss": 0.5430117855833032
    },
    {
      "train_step": 2858,
      "sup_loss": 0.6350588118799216
    },
    {
      "train_step": 2859,
      "sup_loss": 0.6093275669891097
    },
    {
      "train_step": 2860,
      "sup_loss": 0.6924425268198986
    },
    {
      "train_step": 2861,
      "sup_loss": 0.671818873696561
    },
    {
      "train_step": 2862,
      "sup_loss": 0.3799588569594726
    },
    {
      "train_step": 2863,
      "sup_loss": 0.7297672616348717
    },
    {
      "train_step": 2864,
      "sup_loss": 0.6886344719410785
    },
    {
      "train_step": 2865,
      "sup_loss": 0.6808574240001795
    },
    {
      "train_step": 2866,
      "sup_loss": 0.6509877692806745
    },
    {
      "train_step": 2867,
      "sup_loss": 0.6574957092429818
    },
    {
      "train_step": 2868,
      "sup_loss": 0.6918297574103154
    },
    {
      "train_step": 2869,
      "sup_loss": 0.7345261792758285
    },
    {
      "train_step": 2870,
      "sup_loss": 0.4156840176101265
    },
    {
      "train_step": 2871,
      "sup_loss": 0.7933544137179075
    },
    {
      "train_step": 2872,
      "sup_loss": 0.5977189812140915
    },
    {
      "train_step": 2873,
      "sup_loss": 0.5409640609766884
    },
    {
      "train_step": 2874,
      "sup_loss": 0.7173600517265196
    },
    {
      "train_step": 2875,
      "sup_loss": 0.6580195273793471
    },
    {
      "train_step": 2876,
      "sup_loss": 0.7812935512204625
    },
    {
      "train_step": 2877,
      "sup_loss": 0.7841087001211628
    },
    {
      "train_step": 2878,
      "sup_loss": 0.7622587605638922
    },
    {
      "train_step": 2879,
      "sup_loss": 0.5348690251579457
    },
    {
      "train_step": 2880,
      "sup_loss": 0.5484957341022557
    },
    {
      "train_step": 2881,
      "sup_loss": 0.7638800148494118
    },
    {
      "train_step": 2882,
      "sup_loss": 0.5981402656997724
    },
    {
      "train_step": 2883,
      "sup_loss": 0.6338286432609483
    },
    {
      "train_step": 2884,
      "sup_loss": 0.7750228850903225
    },
    {
      "train_step": 2885,
      "sup_loss": 0.7301875794118431
    },
    {
      "train_step": 2886,
      "sup_loss": 0.27737167881821956
    },
    {
      "train_step": 2887,
      "sup_loss": 0.6877620793247222
    },
    {
      "train_step": 2888,
      "sup_loss": 0.7240874163208907
    },
    {
      "train_step": 2889,
      "sup_loss": 0.7445131685769877
    },
    {
      "train_step": 2890,
      "sup_loss": 0.6264697929522274
    },
    {
      "train_step": 2891,
      "sup_loss": 0.7566308773380359
    },
    {
      "train_step": 2892,
      "sup_loss": 0.46891877813723737
    },
    {
      "train_step": 2893,
      "sup_loss": 0.5718443654528164
    },
    {
      "train_step": 2894,
      "sup_loss": 0.600783178528298
    },
    {
      "train_step": 2895,
      "sup_loss": 0.7375281722044362
    },
    {
      "train_step": 2896,
      "sup_loss": 0.6087611368457239
    },
    {
      "train_step": 2897,
      "sup_loss": 0.6116501336323735
    },
    {
      "train_step": 2898,
      "sup_loss": 0.5066667078889576
    },
    {
      "train_step": 2899,
      "sup_loss": 0.6962692896313037
    },
    {
      "train_step": 2900,
      "sup_loss": 0.49205476699274625
    },
    {
      "train_step": 2901,
      "sup_loss": 0.5994698231084216
    },
    {
      "train_step": 2902,
      "sup_loss": 0.5629060245592872
    },
    {
      "train_step": 2903,
      "sup_loss": 0.5780319175404786
    },
    {
      "train_step": 2904,
      "sup_loss": 0.6755578633270878
    },
    {
      "train_step": 2905,
      "sup_loss": 0.3612802872789294
    },
    {
      "train_step": 2906,
      "sup_loss": 0.803715898648101
    },
    {
      "train_step": 2907,
      "sup_loss": 0.5107133158475732
    },
    {
      "train_step": 2908,
      "sup_loss": 0.749108426334225
    },
    {
      "train_step": 2909,
      "sup_loss": 0.3956896473492195
    },
    {
      "train_step": 2910,
      "sup_loss": 0.695362829841899
    },
    {
      "train_step": 2911,
      "sup_loss": 0.6982849941466519
    },
    {
      "train_step": 2912,
      "sup_loss": 0.5098311583690487
    },
    {
      "train_step": 2913,
      "sup_loss": 0.654317261351641
    },
    {
      "train_step": 2914,
      "sup_loss": 0.6293759986102613
    },
    {
      "train_step": 2915,
      "sup_loss": 0.6441960707285783
    },
    {
      "train_step": 2916,
      "sup_loss": 0.657610336299281
    },
    {
      "train_step": 2917,
      "sup_loss": 0.4757694375102138
    },
    {
      "train_step": 2918,
      "sup_loss": 0.5644743516069909
    },
    {
      "train_step": 2919,
      "sup_loss": 0.5761975044604248
    },
    {
      "train_step": 2920,
      "sup_loss": 0.7337628142953156
    },
    {
      "train_step": 2921,
      "sup_loss": 0.41902867672449856
    },
    {
      "train_step": 2922,
      "sup_loss": 0.8127406995088755
    },
    {
      "train_step": 2923,
      "sup_loss": 0.7051666982592886
    },
    {
      "train_step": 2924,
      "sup_loss": 0.6676444516979791
    },
    {
      "train_step": 2925,
      "sup_loss": 0.6035389798370852
    },
    {
      "train_step": 2926,
      "sup_loss": 0.7282693543623936
    },
    {
      "train_step": 2927,
      "sup_loss": 0.7373391556775531
    },
    {
      "train_step": 2928,
      "sup_loss": 0.7069425626084336
    },
    {
      "train_step": 2929,
      "sup_loss": 0.7605109594773601
    },
    {
      "train_step": 2930,
      "sup_loss": 0.6974349236051058
    },
    {
      "train_step": 2931,
      "sup_loss": 0.6505095947865186
    },
    {
      "train_step": 2932,
      "sup_loss": 0.7006404475352346
    },
    {
      "train_step": 2933,
      "sup_loss": 0.6915045689509497
    },
    {
      "train_step": 2934,
      "sup_loss": 0.6206816640789553
    },
    {
      "train_step": 2935,
      "sup_loss": 0.5667063370502416
    },
    {
      "train_step": 2936,
      "sup_loss": 0.5576753015407848
    },
    {
      "train_step": 2937,
      "sup_loss": 0.3781045142027743
    },
    {
      "train_step": 2938,
      "sup_loss": 0.5222917771916064
    },
    {
      "train_step": 2939,
      "sup_loss": 0.7755812596291656
    },
    {
      "train_step": 2940,
      "sup_loss": 0.6020917244791606
    },
    {
      "train_step": 2941,
      "sup_loss": 0.7399468076462724
    },
    {
      "train_step": 2942,
      "sup_loss": 0.5539477652795616
    },
    {
      "train_step": 2943,
      "sup_loss": 0.5607896598175691
    },
    {
      "train_step": 2944,
      "sup_loss": 0.6946442207937275
    },
    {
      "train_step": 2945,
      "sup_loss": 0.6812786338643948
    },
    {
      "train_step": 2946,
      "sup_loss": 0.6715806508807273
    },
    {
      "train_step": 2947,
      "sup_loss": 0.5411611571816036
    },
    {
      "train_step": 2948,
      "sup_loss": 0.4742283979949725
    },
    {
      "train_step": 2949,
      "sup_loss": 0.47292915016732523
    },
    {
      "train_step": 2950,
      "sup_loss": 0.6126454797586454
    },
    {
      "train_step": 2951,
      "sup_loss": 0.6107436814704408
    },
    {
      "train_step": 2952,
      "sup_loss": 0.7085602913326325
    },
    {
      "train_step": 2953,
      "sup_loss": 0.4853910346294621
    },
    {
      "train_step": 2954,
      "sup_loss": 0.6456541864417672
    },
    {
      "train_step": 2955,
      "sup_loss": 0.5489901052910386
    },
    {
      "train_step": 2956,
      "sup_loss": 0.7079574186002969
    },
    {
      "train_step": 2957,
      "sup_loss": 0.6210150025467492
    },
    {
      "train_step": 2958,
      "sup_loss": 0.7597773676749894
    },
    {
      "train_step": 2959,
      "sup_loss": 0.691367628920086
    },
    {
      "train_step": 2960,
      "sup_loss": 0.5156590910336187
    },
    {
      "train_step": 2961,
      "sup_loss": 0.4582396977088489
    },
    {
      "train_step": 2962,
      "sup_loss": 0.5848642308603712
    },
    {
      "train_step": 2963,
      "sup_loss": 0.477034973153697
    },
    {
      "train_step": 2964,
      "sup_loss": 0.6549551357286715
    },
    {
      "train_step": 2965,
      "sup_loss": 0.6734424580721021
    },
    {
      "train_step": 2966,
      "sup_loss": 0.4848211373204251
    },
    {
      "train_step": 2967,
      "sup_loss": 0.7230582905335872
    },
    {
      "train_step": 2968,
      "sup_loss": 0.7155753459833003
    },
    {
      "train_step": 2969,
      "sup_loss": 0.4142100940177391
    },
    {
      "train_step": 2970,
      "sup_loss": 0.6255842359966927
    },
    {
      "train_step": 2971,
      "sup_loss": 0.6419407696213256
    },
    {
      "train_step": 2972,
      "sup_loss": 0.941042256214389
    },
    {
      "train_step": 2973,
      "sup_loss": 0.48121872784747
    },
    {
      "train_step": 2974,
      "sup_loss": 0.5068756657160909
    },
    {
      "train_step": 2975,
      "sup_loss": 0.5267141218545487
    },
    {
      "train_step": 2976,
      "sup_loss": 0.7332835693332184
    },
    {
      "train_step": 2977,
      "sup_loss": 0.7646800754296696
    },
    {
      "train_step": 2978,
      "sup_loss": 0.6694756009917974
    },
    {
      "train_step": 2979,
      "sup_loss": 0.7436935108538955
    },
    {
      "train_step": 2980,
      "sup_loss": 0.5072891131592489
    },
    {
      "train_step": 2981,
      "sup_loss": 0.7063628419477789
    },
    {
      "train_step": 2982,
      "sup_loss": 0.5815321543749109
    },
    {
      "train_step": 2983,
      "sup_loss": 0.4442405375802193
    },
    {
      "train_step": 2984,
      "sup_loss": 0.6013839106096269
    },
    {
      "train_step": 2985,
      "sup_loss": 0.6032257418276171
    },
    {
      "train_step": 2986,
      "sup_loss": 0.33562495687930727
    },
    {
      "train_step": 2987,
      "sup_loss": 0.6458752103975773
    },
    {
      "train_step": 2988,
      "sup_loss": 0.5856936609067726
    },
    {
      "train_step": 2989,
      "sup_loss": 0.5293516315670552
    },
    {
      "train_step": 2990,
      "sup_loss": 0.6973426114763813
    },
    {
      "train_step": 2991,
      "sup_loss": 0.6480787249315015
    },
    {
      "train_step": 2992,
      "sup_loss": 0.6745920079971566
    },
    {
      "train_step": 2993,
      "sup_loss": 0.743026074375915
    },
    {
      "train_step": 2994,
      "sup_loss": 0.8152848720619215
    },
    {
      "train_step": 2995,
      "sup_loss": 0.5469977208214962
    },
    {
      "train_step": 2996,
      "sup_loss": 0.44852041289704725
    },
    {
      "train_step": 2997,
      "sup_loss": 0.42560279852021393
    },
    {
      "train_step": 2998,
      "sup_loss": 0.4958011194800701
    },
    {
      "train_step": 2999,
      "sup_loss": 0.5360726828996288
    },
    {
      "train_step": 3000,
      "sup_loss": 0.5245064204114723
    },
    {
      "train_step": 3001,
      "sup_loss": 0.6717461158729658
    },
    {
      "train_step": 3002,
      "sup_loss": 0.7893386040427706
    },
    {
      "train_step": 3003,
      "sup_loss": 0.7781685323947847
    },
    {
      "train_step": 3004,
      "sup_loss": 0.5466681555750865
    },
    {
      "train_step": 3005,
      "sup_loss": 0.7493482480093433
    },
    {
      "train_step": 3006,
      "sup_loss": 0.4424061929722504
    },
    {
      "train_step": 3007,
      "sup_loss": 0.5448420425842901
    },
    {
      "train_step": 3008,
      "sup_loss": 0.4817198732832974
    },
    {
      "train_step": 3009,
      "sup_loss": 0.712838729116317
    },
    {
      "train_step": 3010,
      "sup_loss": 0.5262222693567729
    },
    {
      "train_step": 3011,
      "sup_loss": 0.6989664385135558
    },
    {
      "train_step": 3012,
      "sup_loss": 0.7359990447159689
    },
    {
      "train_step": 3013,
      "sup_loss": 0.8267380426497346
    },
    {
      "train_step": 3014,
      "sup_loss": 0.55925724589589
    },
    {
      "train_step": 3015,
      "sup_loss": 0.5631652199298951
    },
    {
      "train_step": 3016,
      "sup_loss": 0.878314021045181
    },
    {
      "train_step": 3017,
      "sup_loss": 0.675593026821625
    },
    {
      "train_step": 3018,
      "sup_loss": 0.699573833744091
    },
    {
      "train_step": 3019,
      "sup_loss": 0.8510423325640072
    },
    {
      "train_step": 3020,
      "sup_loss": 0.5649820795301014
    },
    {
      "train_step": 3021,
      "sup_loss": 0.5260356253071609
    },
    {
      "train_step": 3022,
      "sup_loss": 0.6647257868379055
    },
    {
      "train_step": 3023,
      "sup_loss": 0.4466692398637335
    },
    {
      "train_step": 3024,
      "sup_loss": 0.6843248418605363
    },
    {
      "train_step": 3025,
      "sup_loss": 0.5373350364055377
    },
    {
      "train_step": 3026,
      "sup_loss": 0.4847052873086444
    },
    {
      "train_step": 3027,
      "sup_loss": 0.7892663272212408
    },
    {
      "train_step": 3028,
      "sup_loss": 0.6404425960882523
    },
    {
      "train_step": 3029,
      "sup_loss": 0.6646802538652677
    },
    {
      "train_step": 3030,
      "sup_loss": 0.7016456231650994
    },
    {
      "train_step": 3031,
      "sup_loss": 0.6777353126955625
    },
    {
      "train_step": 3032,
      "sup_loss": 0.6976326692702224
    },
    {
      "train_step": 3033,
      "sup_loss": 0.5810308585062265
    },
    {
      "train_step": 3034,
      "sup_loss": 0.6713924395670583
    },
    {
      "train_step": 3035,
      "sup_loss": 0.5926244155551841
    },
    {
      "train_step": 3036,
      "sup_loss": 0.7999437278501901
    },
    {
      "train_step": 3037,
      "sup_loss": 0.7007159442202886
    },
    {
      "train_step": 3038,
      "sup_loss": 0.4797386104652052
    },
    {
      "train_step": 3039,
      "sup_loss": 0.6341002937424853
    },
    {
      "train_step": 3040,
      "sup_loss": 0.6531217791779622
    },
    {
      "train_step": 3041,
      "sup_loss": 0.6315761575053613
    },
    {
      "train_step": 3042,
      "sup_loss": 0.6874401759304245
    },
    {
      "train_step": 3043,
      "sup_loss": 0.6665830676268927
    },
    {
      "train_step": 3044,
      "sup_loss": 0.6209633309845516
    },
    {
      "train_step": 3045,
      "sup_loss": 0.6602869868336647
    },
    {
      "train_step": 3046,
      "sup_loss": 0.7895736545437252
    },
    {
      "train_step": 3047,
      "sup_loss": 0.7888422576151256
    },
    {
      "train_step": 3048,
      "sup_loss": 0.5952128650922808
    },
    {
      "train_step": 3049,
      "sup_loss": 0.4907629906463747
    },
    {
      "train_step": 3050,
      "sup_loss": 0.7704554540416283
    },
    {
      "train_step": 3051,
      "sup_loss": 0.6743980602642793
    },
    {
      "train_step": 3052,
      "sup_loss": 0.3812170527293469
    },
    {
      "train_step": 3053,
      "sup_loss": 0.7116784642419627
    },
    {
      "train_step": 3054,
      "sup_loss": 0.5968597097867323
    },
    {
      "train_step": 3055,
      "sup_loss": 0.642469346959921
    },
    {
      "train_step": 3056,
      "sup_loss": 0.40526336489455583
    },
    {
      "train_step": 3057,
      "sup_loss": 0.49531458406102613
    },
    {
      "train_step": 3058,
      "sup_loss": 0.5477644363963313
    },
    {
      "train_step": 3059,
      "sup_loss": 0.5881873789490886
    },
    {
      "train_step": 3060,
      "sup_loss": 0.47353960748179086
    },
    {
      "train_step": 3061,
      "sup_loss": 0.5629652476686408
    },
    {
      "train_step": 3062,
      "sup_loss": 0.8126211122595494
    },
    {
      "train_step": 3063,
      "sup_loss": 0.6628932317681208
    },
    {
      "train_step": 3064,
      "sup_loss": 0.626022654007941
    },
    {
      "train_step": 3065,
      "sup_loss": 0.489843681279977
    },
    {
      "train_step": 3066,
      "sup_loss": 0.5738531843934414
    },
    {
      "train_step": 3067,
      "sup_loss": 0.692735056767601
    },
    {
      "train_step": 3068,
      "sup_loss": 0.3052742759187433
    },
    {
      "train_step": 3069,
      "sup_loss": 0.7634165653826873
    },
    {
      "train_step": 3070,
      "sup_loss": 0.4932094517349407
    },
    {
      "train_step": 3071,
      "sup_loss": 0.5379660143784799
    },
    {
      "train_step": 3072,
      "sup_loss": 0.6864582800795531
    },
    {
      "train_step": 3073,
      "sup_loss": 0.5788218987433982
    },
    {
      "train_step": 3074,
      "sup_loss": 0.6166225190154879
    },
    {
      "train_step": 3075,
      "sup_loss": 0.5477297116026479
    },
    {
      "train_step": 3076,
      "sup_loss": 0.6155025374190407
    },
    {
      "train_step": 3077,
      "sup_loss": 0.45213994158596615
    },
    {
      "train_step": 3078,
      "sup_loss": 0.561097473236433
    },
    {
      "train_step": 3079,
      "sup_loss": 0.8964781374433519
    },
    {
      "train_step": 3080,
      "sup_loss": 0.5547852161637319
    },
    {
      "train_step": 3081,
      "sup_loss": 0.5608889977742081
    },
    {
      "train_step": 3082,
      "sup_loss": 0.5172733190130786
    },
    {
      "train_step": 3083,
      "sup_loss": 0.533562522864813
    },
    {
      "train_step": 3084,
      "sup_loss": 0.6256556371308852
    },
    {
      "train_step": 3085,
      "sup_loss": 0.48282560063885765
    },
    {
      "train_step": 3086,
      "sup_loss": 0.644869761654769
    },
    {
      "train_step": 3087,
      "sup_loss": 0.5227385391260073
    },
    {
      "train_step": 3088,
      "sup_loss": 0.8312590761057602
    },
    {
      "train_step": 3089,
      "sup_loss": 0.47397751247424413
    },
    {
      "train_step": 3090,
      "sup_loss": 0.573262928123299
    },
    {
      "train_step": 3091,
      "sup_loss": 0.5874562775629306
    },
    {
      "train_step": 3092,
      "sup_loss": 0.8017943398750623
    },
    {
      "train_step": 3093,
      "sup_loss": 0.6471368230069146
    },
    {
      "train_step": 3094,
      "sup_loss": 0.5510394220769038
    },
    {
      "train_step": 3095,
      "sup_loss": 0.4283758007734669
    },
    {
      "train_step": 3096,
      "sup_loss": 0.6997918624043243
    },
    {
      "train_step": 3097,
      "sup_loss": 0.5192561007766457
    },
    {
      "train_step": 3098,
      "sup_loss": 0.5760693899441259
    },
    {
      "train_step": 3099,
      "sup_loss": 0.6958179377794866
    },
    {
      "train_step": 3100,
      "sup_loss": 0.6777412511705535
    },
    {
      "train_step": 3101,
      "sup_loss": 0.6284941232763921
    },
    {
      "train_step": 3102,
      "sup_loss": 0.6365460766970453
    },
    {
      "train_step": 3103,
      "sup_loss": 0.29547314945742403
    },
    {
      "train_step": 3104,
      "sup_loss": 0.5032668046155894
    },
    {
      "train_step": 3105,
      "sup_loss": 0.5007243775561233
    },
    {
      "train_step": 3106,
      "sup_loss": 0.4921864980886064
    },
    {
      "train_step": 3107,
      "sup_loss": 0.5425469836280042
    },
    {
      "train_step": 3108,
      "sup_loss": 0.8210533929661795
    },
    {
      "train_step": 3109,
      "sup_loss": 0.7103966798247772
    },
    {
      "train_step": 3110,
      "sup_loss": 0.8002575920557138
    },
    {
      "train_step": 3111,
      "sup_loss": 0.6276309863664888
    },
    {
      "train_step": 3112,
      "sup_loss": 0.4346419776219274
    },
    {
      "train_step": 3113,
      "sup_loss": 0.6268657200575464
    },
    {
      "train_step": 3114,
      "sup_loss": 0.5637823687266759
    },
    {
      "train_step": 3115,
      "sup_loss": 0.5117820166286877
    },
    {
      "train_step": 3116,
      "sup_loss": 0.520311972642056
    },
    {
      "train_step": 3117,
      "sup_loss": 0.6788168260391994
    },
    {
      "train_step": 3118,
      "sup_loss": 0.5793350380950515
    },
    {
      "train_step": 3119,
      "sup_loss": 0.6543423354998074
    },
    {
      "train_step": 3120,
      "sup_loss": 0.6595271267654164
    },
    {
      "train_step": 3121,
      "sup_loss": 0.709166115680464
    },
    {
      "train_step": 3122,
      "sup_loss": 0.4864314530530964
    },
    {
      "train_step": 3123,
      "sup_loss": 0.41660670061920363
    },
    {
      "train_step": 3124,
      "sup_loss": 0.5784037415716945
    },
    {
      "train_step": 3125,
      "sup_loss": 0.575584802478043
    },
    {
      "train_step": 3126,
      "sup_loss": 0.5860785014828205
    },
    {
      "train_step": 3127,
      "sup_loss": 0.5010292609638536
    },
    {
      "train_step": 3128,
      "sup_loss": 0.6699753311170628
    },
    {
      "train_step": 3129,
      "sup_loss": 0.5374584567745901
    },
    {
      "train_step": 3130,
      "sup_loss": 0.5983129954410282
    },
    {
      "train_step": 3131,
      "sup_loss": 0.7263885425931138
    },
    {
      "train_step": 3132,
      "sup_loss": 0.5738429493728927
    },
    {
      "train_step": 3133,
      "sup_loss": 0.7093098872619745
    },
    {
      "train_step": 3134,
      "sup_loss": 0.5684566117533953
    },
    {
      "train_step": 3135,
      "sup_loss": 0.5644758919837672
    },
    {
      "train_step": 3136,
      "sup_loss": 0.6813099031109411
    },
    {
      "train_step": 3137,
      "sup_loss": 0.8212591942836432
    },
    {
      "train_step": 3138,
      "sup_loss": 0.6322244052586925
    },
    {
      "train_step": 3139,
      "sup_loss": 0.6133523214518716
    },
    {
      "train_step": 3140,
      "sup_loss": 0.6379591255257016
    },
    {
      "train_step": 3141,
      "sup_loss": 0.6186469843354502
    },
    {
      "train_step": 3142,
      "sup_loss": 0.6894985941719058
    },
    {
      "train_step": 3143,
      "sup_loss": 0.6210580344879906
    },
    {
      "train_step": 3144,
      "sup_loss": 0.5526836573252962
    },
    {
      "train_step": 3145,
      "sup_loss": 0.6394513081353863
    },
    {
      "train_step": 3146,
      "sup_loss": 0.6765779032780664
    },
    {
      "train_step": 3147,
      "sup_loss": 0.7789103726370469
    },
    {
      "train_step": 3148,
      "sup_loss": 0.636170230700052
    },
    {
      "train_step": 3149,
      "sup_loss": 0.5450380333291821
    },
    {
      "train_step": 3150,
      "sup_loss": 0.7072725171437455
    },
    {
      "train_step": 3151,
      "sup_loss": 0.6184731705212877
    },
    {
      "train_step": 3152,
      "sup_loss": 0.6200606309289198
    },
    {
      "train_step": 3153,
      "sup_loss": 0.867903339443572
    },
    {
      "train_step": 3154,
      "sup_loss": 0.7785087971476325
    },
    {
      "train_step": 3155,
      "sup_loss": 0.7000295331361178
    },
    {
      "train_step": 3156,
      "sup_loss": 0.6911932982671986
    },
    {
      "train_step": 3157,
      "sup_loss": 0.7192492568094877
    },
    {
      "train_step": 3158,
      "sup_loss": 0.4816062631880613
    },
    {
      "train_step": 3159,
      "sup_loss": 0.4463428758993147
    },
    {
      "train_step": 3160,
      "sup_loss": 0.4879657172303068
    },
    {
      "train_step": 3161,
      "sup_loss": 0.6828938251096537
    },
    {
      "train_step": 3162,
      "sup_loss": 0.4314086587380551
    },
    {
      "train_step": 3163,
      "sup_loss": 0.42676686149961984
    },
    {
      "train_step": 3164,
      "sup_loss": 0.4696056757186698
    },
    {
      "train_step": 3165,
      "sup_loss": 0.3790111284686785
    },
    {
      "train_step": 3166,
      "sup_loss": 0.7730602584055404
    },
    {
      "train_step": 3167,
      "sup_loss": 0.4834510454995639
    },
    {
      "train_step": 3168,
      "sup_loss": 0.8168250279684577
    },
    {
      "train_step": 3169,
      "sup_loss": 0.6366204267019477
    },
    {
      "train_step": 3170,
      "sup_loss": 0.7210891453708439
    },
    {
      "train_step": 3171,
      "sup_loss": 0.6477232637901429
    },
    {
      "train_step": 3172,
      "sup_loss": 0.762579185827245
    },
    {
      "train_step": 3173,
      "sup_loss": 0.6204590132011447
    },
    {
      "train_step": 3174,
      "sup_loss": 0.5573903046677966
    },
    {
      "train_step": 3175,
      "sup_loss": 0.504283020360063
    },
    {
      "train_step": 3176,
      "sup_loss": 0.5997152810077301
    },
    {
      "train_step": 3177,
      "sup_loss": 0.5916533259744942
    },
    {
      "train_step": 3178,
      "sup_loss": 0.6848624493905171
    },
    {
      "train_step": 3179,
      "sup_loss": 0.564727213177901
    },
    {
      "train_step": 3180,
      "sup_loss": 0.5812064055099649
    },
    {
      "train_step": 3181,
      "sup_loss": 0.45557897540761677
    },
    {
      "train_step": 3182,
      "sup_loss": 0.6411751597107964
    },
    {
      "train_step": 3183,
      "sup_loss": 0.5371913439457834
    },
    {
      "train_step": 3184,
      "sup_loss": 0.6462472340906602
    },
    {
      "train_step": 3185,
      "sup_loss": 0.6562436719866694
    },
    {
      "train_step": 3186,
      "sup_loss": 0.6775848815825873
    },
    {
      "train_step": 3187,
      "sup_loss": 0.6953477611904162
    },
    {
      "train_step": 3188,
      "sup_loss": 0.8245417518652511
    },
    {
      "train_step": 3189,
      "sup_loss": 0.576012879779056
    },
    {
      "train_step": 3190,
      "sup_loss": 0.6979457494834336
    },
    {
      "train_step": 3191,
      "sup_loss": 0.618605495965871
    },
    {
      "train_step": 3192,
      "sup_loss": 0.7612783627531816
    },
    {
      "train_step": 3193,
      "sup_loss": 0.6920323972716027
    },
    {
      "train_step": 3194,
      "sup_loss": 0.577152842983642
    },
    {
      "train_step": 3195,
      "sup_loss": 0.7918153861782666
    },
    {
      "train_step": 3196,
      "sup_loss": 0.4359563433992467
    },
    {
      "train_step": 3197,
      "sup_loss": 0.6269016549632839
    },
    {
      "train_step": 3198,
      "sup_loss": 0.5335734731793067
    },
    {
      "train_step": 3199,
      "sup_loss": 0.774368137655997
    },
    {
      "train_step": 3200,
      "sup_loss": 0.6948362837808156
    },
    {
      "train_step": 3201,
      "sup_loss": 0.6748752371255049
    },
    {
      "train_step": 3202,
      "sup_loss": 0.7279659969873665
    },
    {
      "train_step": 3203,
      "sup_loss": 0.5217600282934588
    },
    {
      "train_step": 3204,
      "sup_loss": 0.5465056270651644
    },
    {
      "train_step": 3205,
      "sup_loss": 0.509745846833859
    },
    {
      "train_step": 3206,
      "sup_loss": 0.7497292566840705
    },
    {
      "train_step": 3207,
      "sup_loss": 0.6336061575631818
    },
    {
      "train_step": 3208,
      "sup_loss": 0.6669453247938568
    },
    {
      "train_step": 3209,
      "sup_loss": 0.5811841764595836
    },
    {
      "train_step": 3210,
      "sup_loss": 0.6352172223383111
    },
    {
      "train_step": 3211,
      "sup_loss": 0.7331086804572824
    },
    {
      "train_step": 3212,
      "sup_loss": 0.7424552978726134
    },
    {
      "train_step": 3213,
      "sup_loss": 0.4499511254387031
    },
    {
      "train_step": 3214,
      "sup_loss": 0.44836421620399713
    },
    {
      "train_step": 3215,
      "sup_loss": 0.46554350532815947
    },
    {
      "train_step": 3216,
      "sup_loss": 0.4249639409209188
    },
    {
      "train_step": 3217,
      "sup_loss": 0.6119564973148733
    },
    {
      "train_step": 3218,
      "sup_loss": 0.5897971067861804
    },
    {
      "train_step": 3219,
      "sup_loss": 0.7651430421326962
    },
    {
      "train_step": 3220,
      "sup_loss": 0.45845165673656885
    },
    {
      "train_step": 3221,
      "sup_loss": 0.49582099260303625
    },
    {
      "train_step": 3222,
      "sup_loss": 0.38343409847898846
    },
    {
      "train_step": 3223,
      "sup_loss": 0.6772602423456947
    },
    {
      "train_step": 3224,
      "sup_loss": 0.7030836502470031
    },
    {
      "train_step": 3225,
      "sup_loss": 0.5367319071524078
    },
    {
      "train_step": 3226,
      "sup_loss": 0.635191501937994
    },
    {
      "train_step": 3227,
      "sup_loss": 0.5871054083120532
    },
    {
      "train_step": 3228,
      "sup_loss": 0.4959021313893627
    },
    {
      "train_step": 3229,
      "sup_loss": 0.7309515181820555
    },
    {
      "train_step": 3230,
      "sup_loss": 0.6402309989871777
    },
    {
      "train_step": 3231,
      "sup_loss": 0.5628795841781947
    },
    {
      "train_step": 3232,
      "sup_loss": 0.7373193942474883
    },
    {
      "train_step": 3233,
      "sup_loss": 0.7449568661789439
    },
    {
      "train_step": 3234,
      "sup_loss": 0.4658171022567191
    },
    {
      "train_step": 3235,
      "sup_loss": 0.6017105819088165
    },
    {
      "train_step": 3236,
      "sup_loss": 0.3476560856948647
    },
    {
      "train_step": 3237,
      "sup_loss": 0.7005888140373151
    },
    {
      "train_step": 3238,
      "sup_loss": 0.7016194662318036
    },
    {
      "train_step": 3239,
      "sup_loss": 0.6403637878378244
    },
    {
      "train_step": 3240,
      "sup_loss": 0.32786286138374654
    },
    {
      "train_step": 3241,
      "sup_loss": 0.3854474714685803
    },
    {
      "train_step": 3242,
      "sup_loss": 0.5165408169049246
    },
    {
      "train_step": 3243,
      "sup_loss": 0.6078928278131964
    },
    {
      "train_step": 3244,
      "sup_loss": 0.5182644363178627
    },
    {
      "train_step": 3245,
      "sup_loss": 0.7055445461974765
    },
    {
      "train_step": 3246,
      "sup_loss": 0.6315810935651878
    },
    {
      "train_step": 3247,
      "sup_loss": 0.3930220292261324
    },
    {
      "train_step": 3248,
      "sup_loss": 0.6030358135254172
    },
    {
      "train_step": 3249,
      "sup_loss": 0.5407839668247525
    },
    {
      "train_step": 3250,
      "sup_loss": 0.5419014233626582
    },
    {
      "train_step": 3251,
      "sup_loss": 0.5371883878347211
    },
    {
      "train_step": 3252,
      "sup_loss": 0.749662768144562
    },
    {
      "train_step": 3253,
      "sup_loss": 0.47732063593279755
    },
    {
      "train_step": 3254,
      "sup_loss": 0.5062056682651209
    },
    {
      "train_step": 3255,
      "sup_loss": 0.7070715074951
    },
    {
      "train_step": 3256,
      "sup_loss": 0.6833353167542919
    },
    {
      "train_step": 3257,
      "sup_loss": 1.0069465926695083
    },
    {
      "train_step": 3258,
      "sup_loss": 0.6463472475022732
    },
    {
      "train_step": 3259,
      "sup_loss": 0.6915612436346552
    },
    {
      "train_step": 3260,
      "sup_loss": 0.8516407395881371
    },
    {
      "train_step": 3261,
      "sup_loss": 0.5282834066281087
    },
    {
      "train_step": 3262,
      "sup_loss": 0.5775614271445175
    },
    {
      "train_step": 3263,
      "sup_loss": 0.8490829711433856
    },
    {
      "train_step": 3264,
      "sup_loss": 0.8129126943911222
    },
    {
      "train_step": 3265,
      "sup_loss": 0.5342042084090983
    },
    {
      "train_step": 3266,
      "sup_loss": 0.5679968207646674
    },
    {
      "train_step": 3267,
      "sup_loss": 0.6007981225944254
    },
    {
      "train_step": 3268,
      "sup_loss": 0.7576844956366366
    },
    {
      "train_step": 3269,
      "sup_loss": 0.776854511684483
    },
    {
      "train_step": 3270,
      "sup_loss": 0.4565622402379726
    },
    {
      "train_step": 3271,
      "sup_loss": 0.5148120675377275
    },
    {
      "train_step": 3272,
      "sup_loss": 0.5063731848593944
    },
    {
      "train_step": 3273,
      "sup_loss": 0.7278782066118528
    },
    {
      "train_step": 3274,
      "sup_loss": 0.7579116392322606
    },
    {
      "train_step": 3275,
      "sup_loss": 0.7584788251775164
    },
    {
      "train_step": 3276,
      "sup_loss": 0.6355886342464983
    },
    {
      "train_step": 3277,
      "sup_loss": 0.5604099982492912
    },
    {
      "train_step": 3278,
      "sup_loss": 0.4884310663941355
    },
    {
      "train_step": 3279,
      "sup_loss": 0.6351551222970985
    },
    {
      "train_step": 3280,
      "sup_loss": 0.6308423499592636
    },
    {
      "train_step": 3281,
      "sup_loss": 0.3784119309365346
    },
    {
      "train_step": 3282,
      "sup_loss": 0.6537144613577057
    },
    {
      "train_step": 3283,
      "sup_loss": 0.5865997132350849
    },
    {
      "train_step": 3284,
      "sup_loss": 0.6717921991021513
    },
    {
      "train_step": 3285,
      "sup_loss": 0.592736562789755
    },
    {
      "train_step": 3286,
      "sup_loss": 0.6349833660407636
    },
    {
      "train_step": 3287,
      "sup_loss": 0.6455933681558342
    },
    {
      "train_step": 3288,
      "sup_loss": 0.541190800036149
    },
    {
      "train_step": 3289,
      "sup_loss": 0.45728608952617733
    },
    {
      "train_step": 3290,
      "sup_loss": 0.5393523215511403
    },
    {
      "train_step": 3291,
      "sup_loss": 0.5615811152855033
    },
    {
      "train_step": 3292,
      "sup_loss": 0.6132764274187371
    },
    {
      "train_step": 3293,
      "sup_loss": 0.4913486303873209
    },
    {
      "train_step": 3294,
      "sup_loss": 0.449378745794474
    },
    {
      "train_step": 3295,
      "sup_loss": 0.745554084983018
    },
    {
      "train_step": 3296,
      "sup_loss": 0.7186593190928602
    },
    {
      "train_step": 3297,
      "sup_loss": 0.48774479414493216
    },
    {
      "train_step": 3298,
      "sup_loss": 0.5699791924418554
    },
    {
      "train_step": 3299,
      "sup_loss": 0.6202500296785025
    },
    {
      "train_step": 3300,
      "sup_loss": 0.49212829760101046
    },
    {
      "train_step": 3301,
      "sup_loss": 0.5634935780030936
    },
    {
      "train_step": 3302,
      "sup_loss": 0.49470166821527956
    },
    {
      "train_step": 3303,
      "sup_loss": 0.7840085724600236
    },
    {
      "train_step": 3304,
      "sup_loss": 0.6738038934973907
    },
    {
      "train_step": 3305,
      "sup_loss": 0.6013315040970542
    },
    {
      "train_step": 3306,
      "sup_loss": 0.6489368137112499
    },
    {
      "train_step": 3307,
      "sup_loss": 0.5372414420822882
    },
    {
      "train_step": 3308,
      "sup_loss": 0.5177968027872084
    },
    {
      "train_step": 3309,
      "sup_loss": 0.6617498878312114
    },
    {
      "train_step": 3310,
      "sup_loss": 0.664511425471261
    },
    {
      "train_step": 3311,
      "sup_loss": 0.4585871174025502
    },
    {
      "train_step": 3312,
      "sup_loss": 0.7628904652940338
    },
    {
      "train_step": 3313,
      "sup_loss": 0.6225433688394816
    },
    {
      "train_step": 3314,
      "sup_loss": 0.4996420826191227
    },
    {
      "train_step": 3315,
      "sup_loss": 0.31684674974075205
    },
    {
      "train_step": 3316,
      "sup_loss": 0.6258448689766125
    },
    {
      "train_step": 3317,
      "sup_loss": 0.6150218230850525
    },
    {
      "train_step": 3318,
      "sup_loss": 0.8048699496304572
    },
    {
      "train_step": 3319,
      "sup_loss": 0.6155070367205574
    },
    {
      "train_step": 3320,
      "sup_loss": 0.564997304503405
    },
    {
      "train_step": 3321,
      "sup_loss": 0.6158673038344867
    },
    {
      "train_step": 3322,
      "sup_loss": 0.5664666235796576
    },
    {
      "train_step": 3323,
      "sup_loss": 0.7024258013974894
    },
    {
      "train_step": 3324,
      "sup_loss": 0.8345110231640271
    },
    {
      "train_step": 3325,
      "sup_loss": 0.8327188262535086
    },
    {
      "train_step": 3326,
      "sup_loss": 0.7035385348353963
    },
    {
      "train_step": 3327,
      "sup_loss": 0.4867284032231736
    },
    {
      "train_step": 3328,
      "sup_loss": 0.5840454105114584
    },
    {
      "train_step": 3329,
      "sup_loss": 0.44770911305589606
    },
    {
      "train_step": 3330,
      "sup_loss": 0.5660466805678092
    },
    {
      "train_step": 3331,
      "sup_loss": 0.7813064272091881
    },
    {
      "train_step": 3332,
      "sup_loss": 0.5434451966648863
    },
    {
      "train_step": 3333,
      "sup_loss": 0.5932352631349904
    },
    {
      "train_step": 3334,
      "sup_loss": 0.6093126611346416
    },
    {
      "train_step": 3335,
      "sup_loss": 0.5700889796192631
    },
    {
      "train_step": 3336,
      "sup_loss": 0.4484834770197939
    },
    {
      "train_step": 3337,
      "sup_loss": 0.7323386810871201
    },
    {
      "train_step": 3338,
      "sup_loss": 0.5564203149574681
    },
    {
      "train_step": 3339,
      "sup_loss": 0.5664151071745865
    },
    {
      "train_step": 3340,
      "sup_loss": 0.5931047619857117
    },
    {
      "train_step": 3341,
      "sup_loss": 0.6816239495281377
    },
    {
      "train_step": 3342,
      "sup_loss": 0.44010205935524094
    },
    {
      "train_step": 3343,
      "sup_loss": 0.3457450011890105
    },
    {
      "train_step": 3344,
      "sup_loss": 0.6171838121770892
    },
    {
      "train_step": 3345,
      "sup_loss": 0.6744526912030563
    },
    {
      "train_step": 3346,
      "sup_loss": 0.5460658791912151
    },
    {
      "train_step": 3347,
      "sup_loss": 0.8038849901379881
    },
    {
      "train_step": 3348,
      "sup_loss": 0.6534803450005473
    },
    {
      "train_step": 3349,
      "sup_loss": 0.4855130677429278
    },
    {
      "train_step": 3350,
      "sup_loss": 0.6132582339042933
    },
    {
      "train_step": 3351,
      "sup_loss": 0.6866351317171966
    },
    {
      "train_step": 3352,
      "sup_loss": 0.5525798509409307
    },
    {
      "train_step": 3353,
      "sup_loss": 0.6354047902914586
    },
    {
      "train_step": 3354,
      "sup_loss": 0.5697524697559117
    },
    {
      "train_step": 3355,
      "sup_loss": 0.6387852268515605
    },
    {
      "train_step": 3356,
      "sup_loss": 0.6100418927363624
    },
    {
      "train_step": 3357,
      "sup_loss": 0.6292648371591817
    },
    {
      "train_step": 3358,
      "sup_loss": 0.67281963625399
    },
    {
      "train_step": 3359,
      "sup_loss": 0.6027618005021984
    },
    {
      "train_step": 3360,
      "sup_loss": 0.6599864839301695
    },
    {
      "train_step": 3361,
      "sup_loss": 0.4405377992748965
    },
    {
      "train_step": 3362,
      "sup_loss": 0.43600194190251446
    },
    {
      "train_step": 3363,
      "sup_loss": 0.7629396472485733
    },
    {
      "train_step": 3364,
      "sup_loss": 0.5921161666534036
    },
    {
      "train_step": 3365,
      "sup_loss": 0.5117786810037048
    },
    {
      "train_step": 3366,
      "sup_loss": 0.758353442207825
    },
    {
      "train_step": 3367,
      "sup_loss": 0.5035563852806962
    },
    {
      "train_step": 3368,
      "sup_loss": 0.6937913035066633
    },
    {
      "train_step": 3369,
      "sup_loss": 0.5460390216844675
    },
    {
      "train_step": 3370,
      "sup_loss": 0.5977425458551913
    },
    {
      "train_step": 3371,
      "sup_loss": 0.523702747253543
    },
    {
      "train_step": 3372,
      "sup_loss": 0.6266842102576787
    },
    {
      "train_step": 3373,
      "sup_loss": 0.701536341631448
    },
    {
      "train_step": 3374,
      "sup_loss": 0.5777394451114869
    },
    {
      "train_step": 3375,
      "sup_loss": 0.6793687177076648
    },
    {
      "train_step": 3376,
      "sup_loss": 0.40525491347131803
    },
    {
      "train_step": 3377,
      "sup_loss": 0.761298903695592
    },
    {
      "train_step": 3378,
      "sup_loss": 0.7084747786001983
    },
    {
      "train_step": 3379,
      "sup_loss": 0.6128590922302312
    },
    {
      "train_step": 3380,
      "sup_loss": 0.6388786463989305
    },
    {
      "train_step": 3381,
      "sup_loss": 0.8348926416048238
    },
    {
      "train_step": 3382,
      "sup_loss": 0.6545391396865268
    },
    {
      "train_step": 3383,
      "sup_loss": 0.6305288331993688
    },
    {
      "train_step": 3384,
      "sup_loss": 0.5636803335549334
    },
    {
      "train_step": 3385,
      "sup_loss": 0.6668719601149466
    },
    {
      "train_step": 3386,
      "sup_loss": 0.637964165922107
    },
    {
      "train_step": 3387,
      "sup_loss": 0.6855133909629535
    },
    {
      "train_step": 3388,
      "sup_loss": 0.3667218913611307
    },
    {
      "train_step": 3389,
      "sup_loss": 0.4632098366104181
    },
    {
      "train_step": 3390,
      "sup_loss": 0.7463815340616943
    },
    {
      "train_step": 3391,
      "sup_loss": 0.6207930460255136
    },
    {
      "train_step": 3392,
      "sup_loss": 0.4920468769517478
    },
    {
      "train_step": 3393,
      "sup_loss": 0.6748562115705719
    },
    {
      "train_step": 3394,
      "sup_loss": 0.6360490242911698
    },
    {
      "train_step": 3395,
      "sup_loss": 0.5267796079314734
    },
    {
      "train_step": 3396,
      "sup_loss": 0.5044426789492004
    },
    {
      "train_step": 3397,
      "sup_loss": 0.5789276732800862
    },
    {
      "train_step": 3398,
      "sup_loss": 0.4794105542630134
    },
    {
      "train_step": 3399,
      "sup_loss": 0.5889076104877529
    },
    {
      "train_step": 3400,
      "sup_loss": 0.6883486733512605
    },
    {
      "train_step": 3401,
      "sup_loss": 0.7589517206655874
    },
    {
      "train_step": 3402,
      "sup_loss": 0.6137291265365062
    },
    {
      "train_step": 3403,
      "sup_loss": 0.6175063547149401
    },
    {
      "train_step": 3404,
      "sup_loss": 0.7471059558671541
    },
    {
      "train_step": 3405,
      "sup_loss": 0.6092085708936847
    },
    {
      "train_step": 3406,
      "sup_loss": 0.6888752719110379
    },
    {
      "train_step": 3407,
      "sup_loss": 0.6417482604433158
    },
    {
      "train_step": 3408,
      "sup_loss": 0.7852439594803959
    },
    {
      "train_step": 3409,
      "sup_loss": 0.5290642010464542
    },
    {
      "train_step": 3410,
      "sup_loss": 0.6134634135301239
    },
    {
      "train_step": 3411,
      "sup_loss": 0.532236226537222
    },
    {
      "train_step": 3412,
      "sup_loss": 0.5607659048073949
    },
    {
      "train_step": 3413,
      "sup_loss": 0.4456810555906694
    },
    {
      "train_step": 3414,
      "sup_loss": 0.5181152065399204
    },
    {
      "train_step": 3415,
      "sup_loss": 0.4819675565832953
    },
    {
      "train_step": 3416,
      "sup_loss": 0.40700838720466315
    },
    {
      "train_step": 3417,
      "sup_loss": 0.661726804635353
    },
    {
      "train_step": 3418,
      "sup_loss": 0.7248799072251904
    },
    {
      "train_step": 3419,
      "sup_loss": 0.4832594701432672
    },
    {
      "train_step": 3420,
      "sup_loss": 0.5213463530664216
    },
    {
      "train_step": 3421,
      "sup_loss": 0.35205741091408266
    },
    {
      "train_step": 3422,
      "sup_loss": 0.655852896783356
    },
    {
      "train_step": 3423,
      "sup_loss": 0.606504732900251
    },
    {
      "train_step": 3424,
      "sup_loss": 0.8703665276781329
    },
    {
      "train_step": 3425,
      "sup_loss": 0.5386565900227178
    },
    {
      "train_step": 3426,
      "sup_loss": 0.534814009034631
    },
    {
      "train_step": 3427,
      "sup_loss": 0.2760902032088622
    },
    {
      "train_step": 3428,
      "sup_loss": 0.639283401074837
    },
    {
      "train_step": 3429,
      "sup_loss": 0.5855304222143957
    },
    {
      "train_step": 3430,
      "sup_loss": 0.49834548453543
    },
    {
      "train_step": 3431,
      "sup_loss": 0.6173672956028732
    },
    {
      "train_step": 3432,
      "sup_loss": 0.5433761163775054
    },
    {
      "train_step": 3433,
      "sup_loss": 0.5158632252626689
    },
    {
      "train_step": 3434,
      "sup_loss": 0.6165835263214118
    },
    {
      "train_step": 3435,
      "sup_loss": 0.6708558861075606
    },
    {
      "train_step": 3436,
      "sup_loss": 0.7997970228952356
    },
    {
      "train_step": 3437,
      "sup_loss": 0.6958334448643297
    },
    {
      "train_step": 3438,
      "sup_loss": 0.578250569656287
    },
    {
      "train_step": 3439,
      "sup_loss": 0.6534209144443507
    },
    {
      "train_step": 3440,
      "sup_loss": 0.6514286745587178
    },
    {
      "train_step": 3441,
      "sup_loss": 0.40944980222876853
    },
    {
      "train_step": 3442,
      "sup_loss": 0.7490955584211401
    },
    {
      "train_step": 3443,
      "sup_loss": 0.7805764263232832
    },
    {
      "train_step": 3444,
      "sup_loss": 0.592119743151958
    },
    {
      "train_step": 3445,
      "sup_loss": 0.5949532521497741
    },
    {
      "train_step": 3446,
      "sup_loss": 0.4007729285687743
    },
    {
      "train_step": 3447,
      "sup_loss": 0.6737736494346558
    },
    {
      "train_step": 3448,
      "sup_loss": 0.5682891265949275
    },
    {
      "train_step": 3449,
      "sup_loss": 0.8095568385526077
    },
    {
      "train_step": 3450,
      "sup_loss": 0.6232321240095288
    },
    {
      "train_step": 3451,
      "sup_loss": 0.27578876866976926
    },
    {
      "train_step": 3452,
      "sup_loss": 0.4408610692136732
    },
    {
      "train_step": 3453,
      "sup_loss": 0.5671400048808806
    },
    {
      "train_step": 3454,
      "sup_loss": 0.39322959639461014
    },
    {
      "train_step": 3455,
      "sup_loss": 0.568027971306468
    },
    {
      "train_step": 3456,
      "sup_loss": 0.5655220553768007
    },
    {
      "train_step": 3457,
      "sup_loss": 0.4923211811061311
    },
    {
      "train_step": 3458,
      "sup_loss": 0.534375518448501
    },
    {
      "train_step": 3459,
      "sup_loss": 0.7514788060924731
    },
    {
      "train_step": 3460,
      "sup_loss": 0.5132673015850944
    },
    {
      "train_step": 3461,
      "sup_loss": 0.5166567083751359
    },
    {
      "train_step": 3462,
      "sup_loss": 0.5644079288642285
    },
    {
      "train_step": 3463,
      "sup_loss": 0.45096130045663757
    },
    {
      "train_step": 3464,
      "sup_loss": 0.5240596180954202
    },
    {
      "train_step": 3465,
      "sup_loss": 0.6181844910522347
    },
    {
      "train_step": 3466,
      "sup_loss": 0.5192067933899035
    },
    {
      "train_step": 3467,
      "sup_loss": 0.49785469259285814
    },
    {
      "train_step": 3468,
      "sup_loss": 0.8280501271377195
    },
    {
      "train_step": 3469,
      "sup_loss": 0.7647148828963598
    },
    {
      "train_step": 3470,
      "sup_loss": 0.6284990670262596
    },
    {
      "train_step": 3471,
      "sup_loss": 0.44742205358213616
    },
    {
      "train_step": 3472,
      "sup_loss": 0.5438384834905916
    },
    {
      "train_step": 3473,
      "sup_loss": 0.6911767465303124
    },
    {
      "train_step": 3474,
      "sup_loss": 0.5015415692837426
    },
    {
      "train_step": 3475,
      "sup_loss": 0.6475063945494305
    },
    {
      "train_step": 3476,
      "sup_loss": 0.7609699922897541
    },
    {
      "train_step": 3477,
      "sup_loss": 0.4627808573023963
    },
    {
      "train_step": 3478,
      "sup_loss": 0.6507231295436735
    },
    {
      "train_step": 3479,
      "sup_loss": 0.5908602403234413
    },
    {
      "train_step": 3480,
      "sup_loss": 0.6458782583975897
    },
    {
      "train_step": 3481,
      "sup_loss": 0.47621484066975
    },
    {
      "train_step": 3482,
      "sup_loss": 0.5063099960304732
    },
    {
      "train_step": 3483,
      "sup_loss": 0.6082319389165796
    },
    {
      "train_step": 3484,
      "sup_loss": 0.4349598143936599
    },
    {
      "train_step": 3485,
      "sup_loss": 0.5420559999695087
    },
    {
      "train_step": 3486,
      "sup_loss": 0.8210192099360067
    },
    {
      "train_step": 3487,
      "sup_loss": 0.6783131427846384
    },
    {
      "train_step": 3488,
      "sup_loss": 0.6102362308699005
    },
    {
      "train_step": 3489,
      "sup_loss": 0.5914659027069024
    },
    {
      "train_step": 3490,
      "sup_loss": 0.6256912992433452
    },
    {
      "train_step": 3491,
      "sup_loss": 0.5099810991732903
    },
    {
      "train_step": 3492,
      "sup_loss": 0.8595592103941537
    },
    {
      "train_step": 3493,
      "sup_loss": 0.2961098130110537
    },
    {
      "train_step": 3494,
      "sup_loss": 0.7944942180309335
    },
    {
      "train_step": 3495,
      "sup_loss": 0.3326158995708939
    },
    {
      "train_step": 3496,
      "sup_loss": 0.7056592689274508
    },
    {
      "train_step": 3497,
      "sup_loss": 0.7137312918475547
    },
    {
      "train_step": 3498,
      "sup_loss": 0.5820189676242601
    },
    {
      "train_step": 3499,
      "sup_loss": 0.7436382027858706
    },
    {
      "train_step": 3500,
      "sup_loss": 0.5457223757166383
    },
    {
      "train_step": 3501,
      "sup_loss": 0.6146957742480975
    },
    {
      "train_step": 3502,
      "sup_loss": 0.4427769773582299
    },
    {
      "train_step": 3503,
      "sup_loss": 0.4776723689060486
    },
    {
      "train_step": 3504,
      "sup_loss": 0.4730240867995796
    },
    {
      "train_step": 3505,
      "sup_loss": 0.6992513199267157
    },
    {
      "train_step": 3506,
      "sup_loss": 0.6017645961938174
    },
    {
      "train_step": 3507,
      "sup_loss": 0.5464690730786766
    },
    {
      "train_step": 3508,
      "sup_loss": 0.443859719764897
    },
    {
      "train_step": 3509,
      "sup_loss": 0.6731332634608536
    },
    {
      "train_step": 3510,
      "sup_loss": 0.776095174807807
    },
    {
      "train_step": 3511,
      "sup_loss": 0.5781969298761769
    },
    {
      "train_step": 3512,
      "sup_loss": 0.5296507033457737
    },
    {
      "train_step": 3513,
      "sup_loss": 0.6693610840517175
    },
    {
      "train_step": 3514,
      "sup_loss": 0.6938589555574783
    },
    {
      "train_step": 3515,
      "sup_loss": 0.47842611858341705
    },
    {
      "train_step": 3516,
      "sup_loss": 0.7427860290228077
    },
    {
      "train_step": 3517,
      "sup_loss": 0.7505451598615307
    },
    {
      "train_step": 3518,
      "sup_loss": 0.7169953875806416
    },
    {
      "train_step": 3519,
      "sup_loss": 0.7458963885927684
    },
    {
      "train_step": 3520,
      "sup_loss": 0.6674529796873365
    },
    {
      "train_step": 3521,
      "sup_loss": 0.3807245005838214
    },
    {
      "train_step": 3522,
      "sup_loss": 0.44161617929485114
    },
    {
      "train_step": 3523,
      "sup_loss": 0.46783315757767996
    },
    {
      "train_step": 3524,
      "sup_loss": 0.5746780013637349
    },
    {
      "train_step": 3525,
      "sup_loss": 0.6355138410778397
    },
    {
      "train_step": 3526,
      "sup_loss": 0.6334289744930446
    },
    {
      "train_step": 3527,
      "sup_loss": 0.6230701897444293
    },
    {
      "train_step": 3528,
      "sup_loss": 0.49813382278532337
    },
    {
      "train_step": 3529,
      "sup_loss": 0.5186129336226434
    },
    {
      "train_step": 3530,
      "sup_loss": 0.7165384036083001
    },
    {
      "train_step": 3531,
      "sup_loss": 0.4703815002010291
    },
    {
      "train_step": 3532,
      "sup_loss": 0.6609715665999025
    },
    {
      "train_step": 3533,
      "sup_loss": 0.687408145766096
    },
    {
      "train_step": 3534,
      "sup_loss": 0.5541458034838169
    },
    {
      "train_step": 3535,
      "sup_loss": 0.40350395482432383
    },
    {
      "train_step": 3536,
      "sup_loss": 0.6013398420623548
    },
    {
      "train_step": 3537,
      "sup_loss": 0.6100229244252764
    },
    {
      "train_step": 3538,
      "sup_loss": 0.6932476999629373
    },
    {
      "train_step": 3539,
      "sup_loss": 0.6081800437747383
    },
    {
      "train_step": 3540,
      "sup_loss": 0.7258177097273483
    },
    {
      "train_step": 3541,
      "sup_loss": 0.622016401824972
    },
    {
      "train_step": 3542,
      "sup_loss": 0.5713495020855214
    },
    {
      "train_step": 3543,
      "sup_loss": 0.6337148371748397
    },
    {
      "train_step": 3544,
      "sup_loss": 0.5820740598162091
    },
    {
      "train_step": 3545,
      "sup_loss": 0.49497304599893555
    },
    {
      "train_step": 3546,
      "sup_loss": 0.6895355057445126
    },
    {
      "train_step": 3547,
      "sup_loss": 0.38420197571637427
    },
    {
      "train_step": 3548,
      "sup_loss": 0.7294685473311898
    },
    {
      "train_step": 3549,
      "sup_loss": 0.6580267225211289
    },
    {
      "train_step": 3550,
      "sup_loss": 0.6707090130910571
    },
    {
      "train_step": 3551,
      "sup_loss": 0.491807499324543
    },
    {
      "train_step": 3552,
      "sup_loss": 0.7202942210554554
    },
    {
      "train_step": 3553,
      "sup_loss": 0.6005011930957612
    },
    {
      "train_step": 3554,
      "sup_loss": 0.7513377618610413
    },
    {
      "train_step": 3555,
      "sup_loss": 0.5652347705387138
    },
    {
      "train_step": 3556,
      "sup_loss": 0.4037873637212312
    },
    {
      "train_step": 3557,
      "sup_loss": 0.6010211613929323
    },
    {
      "train_step": 3558,
      "sup_loss": 0.45512041552235566
    },
    {
      "train_step": 3559,
      "sup_loss": 0.44180216399647154
    },
    {
      "train_step": 3560,
      "sup_loss": 0.6996999002767536
    },
    {
      "train_step": 3561,
      "sup_loss": 0.6393735523483642
    },
    {
      "train_step": 3562,
      "sup_loss": 0.633816498320982
    },
    {
      "train_step": 3563,
      "sup_loss": 0.6601557657044124
    },
    {
      "train_step": 3564,
      "sup_loss": 0.6119061934279788
    },
    {
      "train_step": 3565,
      "sup_loss": 0.8366857386048646
    },
    {
      "train_step": 3566,
      "sup_loss": 0.616585940587484
    },
    {
      "train_step": 3567,
      "sup_loss": 0.6656196812393107
    },
    {
      "train_step": 3568,
      "sup_loss": 0.42568071959750803
    },
    {
      "train_step": 3569,
      "sup_loss": 0.7784521354714449
    },
    {
      "train_step": 3570,
      "sup_loss": 0.5971892306104903
    },
    {
      "train_step": 3571,
      "sup_loss": 0.703818216939804
    },
    {
      "train_step": 3572,
      "sup_loss": 0.7761950967152561
    },
    {
      "train_step": 3573,
      "sup_loss": 0.6636178209596236
    },
    {
      "train_step": 3574,
      "sup_loss": 0.31564898575315226
    },
    {
      "train_step": 3575,
      "sup_loss": 0.5958027362936645
    },
    {
      "train_step": 3576,
      "sup_loss": 0.4927470166747237
    },
    {
      "train_step": 3577,
      "sup_loss": 0.54161661527816
    },
    {
      "train_step": 3578,
      "sup_loss": 0.7083842077930879
    },
    {
      "train_step": 3579,
      "sup_loss": 0.4843561970803618
    },
    {
      "train_step": 3580,
      "sup_loss": 0.7499605027220768
    },
    {
      "train_step": 3581,
      "sup_loss": 0.6226812784759133
    },
    {
      "train_step": 3582,
      "sup_loss": 0.7427840843550054
    },
    {
      "train_step": 3583,
      "sup_loss": 0.6253127671254564
    },
    {
      "train_step": 3584,
      "sup_loss": 0.6871490934684956
    },
    {
      "train_step": 3585,
      "sup_loss": 0.5630211876943826
    },
    {
      "train_step": 3586,
      "sup_loss": 0.5877808659571915
    },
    {
      "train_step": 3587,
      "sup_loss": 0.4695719545135091
    },
    {
      "train_step": 3588,
      "sup_loss": 0.6321271307153375
    },
    {
      "train_step": 3589,
      "sup_loss": 0.4831870655681803
    },
    {
      "train_step": 3590,
      "sup_loss": 0.6169814444368695
    },
    {
      "train_step": 3591,
      "sup_loss": 0.4257843661088584
    },
    {
      "train_step": 3592,
      "sup_loss": 0.5872915415798863
    },
    {
      "train_step": 3593,
      "sup_loss": 0.6189922277186598
    },
    {
      "train_step": 3594,
      "sup_loss": 0.5636734391501725
    },
    {
      "train_step": 3595,
      "sup_loss": 0.5908694046432943
    },
    {
      "train_step": 3596,
      "sup_loss": 0.6955443085067654
    },
    {
      "train_step": 3597,
      "sup_loss": 0.5552585297799715
    },
    {
      "train_step": 3598,
      "sup_loss": 0.7171930947916365
    },
    {
      "train_step": 3599,
      "sup_loss": 0.6506874109312877
    },
    {
      "train_step": 3600,
      "sup_loss": 0.6439096084232176
    },
    {
      "train_step": 3601,
      "sup_loss": 0.7472227063041569
    },
    {
      "train_step": 3602,
      "sup_loss": 0.5429538480869661
    },
    {
      "train_step": 3603,
      "sup_loss": 0.5573120311969119
    },
    {
      "train_step": 3604,
      "sup_loss": 0.5465168453588672
    },
    {
      "train_step": 3605,
      "sup_loss": 0.5372071907591734
    },
    {
      "train_step": 3606,
      "sup_loss": 0.7221877455704798
    },
    {
      "train_step": 3607,
      "sup_loss": 0.6754783031114812
    },
    {
      "train_step": 3608,
      "sup_loss": 0.49834479953220184
    },
    {
      "train_step": 3609,
      "sup_loss": 0.5519007074748101
    },
    {
      "train_step": 3610,
      "sup_loss": 0.6677514069981584
    },
    {
      "train_step": 3611,
      "sup_loss": 0.4352757459684072
    },
    {
      "train_step": 3612,
      "sup_loss": 0.4154899906973379
    },
    {
      "train_step": 3613,
      "sup_loss": 0.6611655591789029
    },
    {
      "train_step": 3614,
      "sup_loss": 0.8063588795864614
    },
    {
      "train_step": 3615,
      "sup_loss": 0.4371630013644452
    },
    {
      "train_step": 3616,
      "sup_loss": 0.5147234952056176
    },
    {
      "train_step": 3617,
      "sup_loss": 0.7011497396287827
    },
    {
      "train_step": 3618,
      "sup_loss": 0.5262302285637088
    },
    {
      "train_step": 3619,
      "sup_loss": 0.6650786039617178
    },
    {
      "train_step": 3620,
      "sup_loss": 0.6013431026385846
    },
    {
      "train_step": 3621,
      "sup_loss": 0.4498736621079624
    },
    {
      "train_step": 3622,
      "sup_loss": 0.6936386714973216
    },
    {
      "train_step": 3623,
      "sup_loss": 0.5883534575249019
    },
    {
      "train_step": 3624,
      "sup_loss": 0.5477580991265188
    },
    {
      "train_step": 3625,
      "sup_loss": 0.3964615710360635
    },
    {
      "train_step": 3626,
      "sup_loss": 0.6403035798429569
    },
    {
      "train_step": 3627,
      "sup_loss": 0.5368652945923831
    },
    {
      "train_step": 3628,
      "sup_loss": 0.46770579203828727
    },
    {
      "train_step": 3629,
      "sup_loss": 0.6177189307728567
    },
    {
      "train_step": 3630,
      "sup_loss": 0.5529717980173731
    },
    {
      "train_step": 3631,
      "sup_loss": 0.6402798407072228
    },
    {
      "train_step": 3632,
      "sup_loss": 0.49258565651454633
    },
    {
      "train_step": 3633,
      "sup_loss": 0.6631763141130685
    },
    {
      "train_step": 3634,
      "sup_loss": 0.6047526553062511
    },
    {
      "train_step": 3635,
      "sup_loss": 0.6072702031825529
    },
    {
      "train_step": 3636,
      "sup_loss": 0.481399216278381
    },
    {
      "train_step": 3637,
      "sup_loss": 0.5155415161583233
    },
    {
      "train_step": 3638,
      "sup_loss": 0.7794020238934244
    },
    {
      "train_step": 3639,
      "sup_loss": 0.672776397537262
    },
    {
      "train_step": 3640,
      "sup_loss": 0.4191314253633879
    },
    {
      "train_step": 3641,
      "sup_loss": 0.4018932554661698
    },
    {
      "train_step": 3642,
      "sup_loss": 0.7860978508727362
    },
    {
      "train_step": 3643,
      "sup_loss": 0.6775914321683354
    },
    {
      "train_step": 3644,
      "sup_loss": 0.7479496287213527
    },
    {
      "train_step": 3645,
      "sup_loss": 0.6698690237413104
    },
    {
      "train_step": 3646,
      "sup_loss": 0.4775662090214496
    },
    {
      "train_step": 3647,
      "sup_loss": 0.3325523163343854
    },
    {
      "train_step": 3648,
      "sup_loss": 0.49362367654273637
    },
    {
      "train_step": 3649,
      "sup_loss": 0.8389926255131039
    },
    {
      "train_step": 3650,
      "sup_loss": 0.729372105713866
    },
    {
      "train_step": 3651,
      "sup_loss": 0.656187465010277
    },
    {
      "train_step": 3652,
      "sup_loss": 0.5466537050062458
    },
    {
      "train_step": 3653,
      "sup_loss": 0.4704439493644983
    },
    {
      "train_step": 3654,
      "sup_loss": 0.76799957045104
    },
    {
      "train_step": 3655,
      "sup_loss": 0.7435613869251074
    },
    {
      "train_step": 3656,
      "sup_loss": 0.760473552064346
    },
    {
      "train_step": 3657,
      "sup_loss": 0.569247294217715
    },
    {
      "train_step": 3658,
      "sup_loss": 0.5701480173421792
    },
    {
      "train_step": 3659,
      "sup_loss": 0.6962431497851725
    },
    {
      "train_step": 3660,
      "sup_loss": 0.4127369930157091
    },
    {
      "train_step": 3661,
      "sup_loss": 0.29551926669752165
    },
    {
      "train_step": 3662,
      "sup_loss": 0.7040491578691779
    },
    {
      "train_step": 3663,
      "sup_loss": 0.454323769340568
    },
    {
      "train_step": 3664,
      "sup_loss": 0.6275299392131422
    },
    {
      "train_step": 3665,
      "sup_loss": 0.42697200211395164
    },
    {
      "train_step": 3666,
      "sup_loss": 0.7318253533286483
    },
    {
      "train_step": 3667,
      "sup_loss": 0.5387394560997231
    },
    {
      "train_step": 3668,
      "sup_loss": 0.7386122397879417
    },
    {
      "train_step": 3669,
      "sup_loss": 0.6932332836127532
    },
    {
      "train_step": 3670,
      "sup_loss": 0.5740833759260116
    },
    {
      "train_step": 3671,
      "sup_loss": 0.6899965190090389
    },
    {
      "train_step": 3672,
      "sup_loss": 0.6908940826969807
    },
    {
      "train_step": 3673,
      "sup_loss": 0.6435125726573246
    },
    {
      "train_step": 3674,
      "sup_loss": 0.38629521450761045
    },
    {
      "train_step": 3675,
      "sup_loss": 0.7078211936284493
    },
    {
      "train_step": 3676,
      "sup_loss": 0.8443329827755464
    },
    {
      "train_step": 3677,
      "sup_loss": 0.5890092594953531
    },
    {
      "train_step": 3678,
      "sup_loss": 0.6075474376840052
    },
    {
      "train_step": 3679,
      "sup_loss": 0.33585728086532946
    },
    {
      "train_step": 3680,
      "sup_loss": 0.45840424487125875
    },
    {
      "train_step": 3681,
      "sup_loss": 0.5327752220595933
    },
    {
      "train_step": 3682,
      "sup_loss": 0.6366765299983095
    },
    {
      "train_step": 3683,
      "sup_loss": 0.6187281588119832
    },
    {
      "train_step": 3684,
      "sup_loss": 0.7691748566864687
    },
    {
      "train_step": 3685,
      "sup_loss": 0.6843821591436259
    },
    {
      "train_step": 3686,
      "sup_loss": 0.4465035050064255
    },
    {
      "train_step": 3687,
      "sup_loss": 0.6154063562382618
    },
    {
      "train_step": 3688,
      "sup_loss": 0.4615791085895476
    },
    {
      "train_step": 3689,
      "sup_loss": 0.5235231820278478
    },
    {
      "train_step": 3690,
      "sup_loss": 0.71114155813699
    },
    {
      "train_step": 3691,
      "sup_loss": 0.8155614131750768
    },
    {
      "train_step": 3692,
      "sup_loss": 0.6046402765044326
    },
    {
      "train_step": 3693,
      "sup_loss": 0.5128828332559644
    },
    {
      "train_step": 3694,
      "sup_loss": 0.8129952925218124
    },
    {
      "train_step": 3695,
      "sup_loss": 0.6695778991408338
    },
    {
      "train_step": 3696,
      "sup_loss": 0.639731980847829
    },
    {
      "train_step": 3697,
      "sup_loss": 0.659942559644933
    },
    {
      "train_step": 3698,
      "sup_loss": 0.580412579353923
    },
    {
      "train_step": 3699,
      "sup_loss": 0.3718089948341999
    },
    {
      "train_step": 3700,
      "sup_loss": 0.6004778413024356
    },
    {
      "train_step": 3701,
      "sup_loss": 1.0063518720999032
    },
    {
      "train_step": 3702,
      "sup_loss": 0.5556211769477267
    },
    {
      "train_step": 3703,
      "sup_loss": 0.6117892758824213
    },
    {
      "train_step": 3704,
      "sup_loss": 0.6626478864596103
    },
    {
      "train_step": 3705,
      "sup_loss": 0.6072063719468878
    },
    {
      "train_step": 3706,
      "sup_loss": 0.42800145688289315
    },
    {
      "train_step": 3707,
      "sup_loss": 0.7295252302546757
    },
    {
      "train_step": 3708,
      "sup_loss": 0.438146215697159
    },
    {
      "train_step": 3709,
      "sup_loss": 0.7036522357288357
    },
    {
      "train_step": 3710,
      "sup_loss": 0.486140317511328
    },
    {
      "train_step": 3711,
      "sup_loss": 0.5529658103302415
    },
    {
      "train_step": 3712,
      "sup_loss": 0.505631518200154
    },
    {
      "train_step": 3713,
      "sup_loss": 0.7654323157910853
    },
    {
      "train_step": 3714,
      "sup_loss": 0.6376068064393043
    },
    {
      "train_step": 3715,
      "sup_loss": 0.574045167328231
    },
    {
      "train_step": 3716,
      "sup_loss": 0.37387438142041995
    },
    {
      "train_step": 3717,
      "sup_loss": 0.742126317355418
    },
    {
      "train_step": 3718,
      "sup_loss": 0.6155714578286474
    },
    {
      "train_step": 3719,
      "sup_loss": 0.7272911879474184
    },
    {
      "train_step": 3720,
      "sup_loss": 0.7225946207176964
    },
    {
      "train_step": 3721,
      "sup_loss": 0.6724705831042435
    },
    {
      "train_step": 3722,
      "sup_loss": 0.568556042673932
    },
    {
      "train_step": 3723,
      "sup_loss": 0.7574725342887562
    },
    {
      "train_step": 3724,
      "sup_loss": 0.604345523012247
    },
    {
      "train_step": 3725,
      "sup_loss": 0.4513985342209829
    },
    {
      "train_step": 3726,
      "sup_loss": 0.6030211330425503
    },
    {
      "train_step": 3727,
      "sup_loss": 0.5466400203372063
    },
    {
      "train_step": 3728,
      "sup_loss": 0.7032884092855989
    },
    {
      "train_step": 3729,
      "sup_loss": 0.7826931011712834
    },
    {
      "train_step": 3730,
      "sup_loss": 0.5159125536329415
    },
    {
      "train_step": 3731,
      "sup_loss": 0.3700275226736699
    },
    {
      "train_step": 3732,
      "sup_loss": 0.3109413359433741
    },
    {
      "train_step": 3733,
      "sup_loss": 0.5829862731722696
    },
    {
      "train_step": 3734,
      "sup_loss": 0.5323169099575199
    },
    {
      "train_step": 3735,
      "sup_loss": 0.3613696598134698
    },
    {
      "train_step": 3736,
      "sup_loss": 0.4593084241751371
    },
    {
      "train_step": 3737,
      "sup_loss": 0.5746368213010503
    },
    {
      "train_step": 3738,
      "sup_loss": 0.6923570133231892
    },
    {
      "train_step": 3739,
      "sup_loss": 0.47706049859997873
    },
    {
      "train_step": 3740,
      "sup_loss": 0.49236045116274835
    },
    {
      "train_step": 3741,
      "sup_loss": 0.6541168862954921
    },
    {
      "train_step": 3742,
      "sup_loss": 0.6441172421365541
    },
    {
      "train_step": 3743,
      "sup_loss": 0.6031907153387573
    },
    {
      "train_step": 3744,
      "sup_loss": 0.5352206655803184
    },
    {
      "train_step": 3745,
      "sup_loss": 0.3560399816831334
    },
    {
      "train_step": 3746,
      "sup_loss": 0.5091767606551394
    },
    {
      "train_step": 3747,
      "sup_loss": 0.6633909967911307
    },
    {
      "train_step": 3748,
      "sup_loss": 0.670945712104142
    },
    {
      "train_step": 3749,
      "sup_loss": 0.6803040984603262
    },
    {
      "train_step": 3750,
      "sup_loss": 0.6774616314860116
    },
    {
      "train_step": 3751,
      "sup_loss": 0.6431351184926191
    },
    {
      "train_step": 3752,
      "sup_loss": 0.7175060542493468
    },
    {
      "train_step": 3753,
      "sup_loss": 0.7248762044149129
    },
    {
      "train_step": 3754,
      "sup_loss": 0.6320202821490074
    },
    {
      "train_step": 3755,
      "sup_loss": 0.7881242565377594
    },
    {
      "train_step": 3756,
      "sup_loss": 0.6678295929936544
    },
    {
      "train_step": 3757,
      "sup_loss": 0.48901358901542025
    },
    {
      "train_step": 3758,
      "sup_loss": 0.6478064162613122
    },
    {
      "train_step": 3759,
      "sup_loss": 0.6433060887783273
    },
    {
      "train_step": 3760,
      "sup_loss": 0.47805849870026357
    },
    {
      "train_step": 3761,
      "sup_loss": 0.8195782868201885
    },
    {
      "train_step": 3762,
      "sup_loss": 0.6209156779775589
    },
    {
      "train_step": 3763,
      "sup_loss": 0.6106916904885502
    },
    {
      "train_step": 3764,
      "sup_loss": 0.7319021413153386
    },
    {
      "train_step": 3765,
      "sup_loss": 0.5443952023487666
    },
    {
      "train_step": 3766,
      "sup_loss": 0.5442078502080083
    },
    {
      "train_step": 3767,
      "sup_loss": 0.7408887802940383
    },
    {
      "train_step": 3768,
      "sup_loss": 0.7460572404763022
    },
    {
      "train_step": 3769,
      "sup_loss": 0.30879070837074063
    },
    {
      "train_step": 3770,
      "sup_loss": 0.793270922457067
    },
    {
      "train_step": 3771,
      "sup_loss": 0.47725180812820067
    },
    {
      "train_step": 3772,
      "sup_loss": 0.45690172420048997
    },
    {
      "train_step": 3773,
      "sup_loss": 0.4899270103265574
    },
    {
      "train_step": 3774,
      "sup_loss": 0.5053888469447737
    },
    {
      "train_step": 3775,
      "sup_loss": 0.5305498035328051
    },
    {
      "train_step": 3776,
      "sup_loss": 0.7765801976532389
    },
    {
      "train_step": 3777,
      "sup_loss": 0.6826974854225181
    },
    {
      "train_step": 3778,
      "sup_loss": 0.6807030697735302
    },
    {
      "train_step": 3779,
      "sup_loss": 0.6080565917250583
    },
    {
      "train_step": 3780,
      "sup_loss": 0.6699555121934283
    },
    {
      "train_step": 3781,
      "sup_loss": 0.6092459450081472
    },
    {
      "train_step": 3782,
      "sup_loss": 0.5725642292531489
    },
    {
      "train_step": 3783,
      "sup_loss": 0.5919734069974883
    },
    {
      "train_step": 3784,
      "sup_loss": 0.6183920014992491
    },
    {
      "train_step": 3785,
      "sup_loss": 0.47506632356578143
    },
    {
      "train_step": 3786,
      "sup_loss": 0.6315632596622367
    },
    {
      "train_step": 3787,
      "sup_loss": 0.6929230110341748
    },
    {
      "train_step": 3788,
      "sup_loss": 0.7397989755431972
    },
    {
      "train_step": 3789,
      "sup_loss": 0.5656695380874722
    },
    {
      "train_step": 3790,
      "sup_loss": 0.6366630614769864
    },
    {
      "train_step": 3791,
      "sup_loss": 0.4614853391295057
    },
    {
      "train_step": 3792,
      "sup_loss": 0.3601190509846972
    },
    {
      "train_step": 3793,
      "sup_loss": 0.26494746459707463
    },
    {
      "train_step": 3794,
      "sup_loss": 0.40622050113484887
    },
    {
      "train_step": 3795,
      "sup_loss": 0.6473383079872668
    },
    {
      "train_step": 3796,
      "sup_loss": 0.4663231459708493
    },
    {
      "train_step": 3797,
      "sup_loss": 0.6186971707611718
    },
    {
      "train_step": 3798,
      "sup_loss": 0.7537793739558711
    },
    {
      "train_step": 3799,
      "sup_loss": 0.5891919963632201
    },
    {
      "train_step": 3800,
      "sup_loss": 0.5866209654386086
    },
    {
      "train_step": 3801,
      "sup_loss": 0.715362339094964
    },
    {
      "train_step": 3802,
      "sup_loss": 0.511907184144895
    },
    {
      "train_step": 3803,
      "sup_loss": 0.46775465818600065
    },
    {
      "train_step": 3804,
      "sup_loss": 0.5967905392662233
    },
    {
      "train_step": 3805,
      "sup_loss": 0.6242571739857
    },
    {
      "train_step": 3806,
      "sup_loss": 0.6323910225806572
    },
    {
      "train_step": 3807,
      "sup_loss": 0.6534035834622404
    },
    {
      "train_step": 3808,
      "sup_loss": 0.5031822769412145
    },
    {
      "train_step": 3809,
      "sup_loss": 0.6152144329001701
    },
    {
      "train_step": 3810,
      "sup_loss": 0.65553490646094
    },
    {
      "train_step": 3811,
      "sup_loss": 0.515838086543996
    },
    {
      "train_step": 3812,
      "sup_loss": 0.5463833061026602
    },
    {
      "train_step": 3813,
      "sup_loss": 0.6637877453264726
    },
    {
      "train_step": 3814,
      "sup_loss": 0.6949907533064922
    },
    {
      "train_step": 3815,
      "sup_loss": 0.4357474460570614
    },
    {
      "train_step": 3816,
      "sup_loss": 0.5208218120101865
    },
    {
      "train_step": 3817,
      "sup_loss": 0.3126796838027333
    },
    {
      "train_step": 3818,
      "sup_loss": 0.8003961758968202
    },
    {
      "train_step": 3819,
      "sup_loss": 0.6443934482975122
    },
    {
      "train_step": 3820,
      "sup_loss": 0.4100160183622468
    },
    {
      "train_step": 3821,
      "sup_loss": 0.3850174570391023
    },
    {
      "train_step": 3822,
      "sup_loss": 0.3673162731616326
    },
    {
      "train_step": 3823,
      "sup_loss": 0.7094405764656304
    },
    {
      "train_step": 3824,
      "sup_loss": 0.45502952004699365
    },
    {
      "train_step": 3825,
      "sup_loss": 0.510402586513276
    },
    {
      "train_step": 3826,
      "sup_loss": 0.6067924847809514
    },
    {
      "train_step": 3827,
      "sup_loss": 0.6789766504246593
    },
    {
      "train_step": 3828,
      "sup_loss": 0.5890438053245137
    },
    {
      "train_step": 3829,
      "sup_loss": 0.654355553671909
    },
    {
      "train_step": 3830,
      "sup_loss": 0.6727127471804563
    },
    {
      "train_step": 3831,
      "sup_loss": 0.7373569319170554
    },
    {
      "train_step": 3832,
      "sup_loss": 0.4310527336325125
    },
    {
      "train_step": 3833,
      "sup_loss": 0.5573193087179874
    },
    {
      "train_step": 3834,
      "sup_loss": 0.5584103115612626
    },
    {
      "train_step": 3835,
      "sup_loss": 0.4535840933913134
    },
    {
      "train_step": 3836,
      "sup_loss": 0.6183884739456027
    },
    {
      "train_step": 3837,
      "sup_loss": 0.5176228667115652
    },
    {
      "train_step": 3838,
      "sup_loss": 0.5376734898979812
    },
    {
      "train_step": 3839,
      "sup_loss": 0.7383754035759057
    },
    {
      "train_step": 3840,
      "sup_loss": 0.5691658589868007
    },
    {
      "train_step": 3841,
      "sup_loss": 0.6034860232384889
    },
    {
      "train_step": 3842,
      "sup_loss": 0.6659176371985216
    },
    {
      "train_step": 3843,
      "sup_loss": 0.5843838478836687
    },
    {
      "train_step": 3844,
      "sup_loss": 0.6358895206427798
    },
    {
      "train_step": 3845,
      "sup_loss": 0.6349518877358988
    },
    {
      "train_step": 3846,
      "sup_loss": 0.6161252643212848
    },
    {
      "train_step": 3847,
      "sup_loss": 0.5077823818826169
    },
    {
      "train_step": 3848,
      "sup_loss": 0.5986453682881842
    },
    {
      "train_step": 3849,
      "sup_loss": 0.5606494484958
    },
    {
      "train_step": 3850,
      "sup_loss": 0.3605194167590618
    },
    {
      "train_step": 3851,
      "sup_loss": 0.6271315843710906
    },
    {
      "train_step": 3852,
      "sup_loss": 0.4012167901133037
    },
    {
      "train_step": 3853,
      "sup_loss": 0.6692465160512372
    },
    {
      "train_step": 3854,
      "sup_loss": 0.511254283187476
    },
    {
      "train_step": 3855,
      "sup_loss": 0.5430642071135532
    },
    {
      "train_step": 3856,
      "sup_loss": 0.618450377277807
    },
    {
      "train_step": 3857,
      "sup_loss": 0.6110599087733337
    },
    {
      "train_step": 3858,
      "sup_loss": 0.5446801240393768
    },
    {
      "train_step": 3859,
      "sup_loss": 0.5075895395791241
    },
    {
      "train_step": 3860,
      "sup_loss": 0.24430417518836248
    },
    {
      "train_step": 3861,
      "sup_loss": 0.4074866594799518
    },
    {
      "train_step": 3862,
      "sup_loss": 0.6786378063406515
    },
    {
      "train_step": 3863,
      "sup_loss": 0.6904779757046503
    },
    {
      "train_step": 3864,
      "sup_loss": 0.5040777782441114
    },
    {
      "train_step": 3865,
      "sup_loss": 0.6665753890496411
    },
    {
      "train_step": 3866,
      "sup_loss": 0.5294631033448821
    },
    {
      "train_step": 3867,
      "sup_loss": 0.7143073296884722
    },
    {
      "train_step": 3868,
      "sup_loss": 0.7610437075643782
    },
    {
      "train_step": 3869,
      "sup_loss": 0.4920164001128726
    },
    {
      "train_step": 3870,
      "sup_loss": 0.6555628276115224
    },
    {
      "train_step": 3871,
      "sup_loss": 0.5390297802079204
    },
    {
      "train_step": 3872,
      "sup_loss": 0.3224837075004928
    },
    {
      "train_step": 3873,
      "sup_loss": 0.46946519199537035
    },
    {
      "train_step": 3874,
      "sup_loss": 0.5092981475516928
    },
    {
      "train_step": 3875,
      "sup_loss": 0.6025293138566807
    },
    {
      "train_step": 3876,
      "sup_loss": 0.4700407243974832
    },
    {
      "train_step": 3877,
      "sup_loss": 0.5701296540733184
    },
    {
      "train_step": 3878,
      "sup_loss": 0.4478692843590045
    },
    {
      "train_step": 3879,
      "sup_loss": 0.5493187475000101
    },
    {
      "train_step": 3880,
      "sup_loss": 0.6092299599175673
    },
    {
      "train_step": 3881,
      "sup_loss": 0.7158243588460915
    },
    {
      "train_step": 3882,
      "sup_loss": 0.6003569416828783
    },
    {
      "train_step": 3883,
      "sup_loss": 0.5678198968868647
    },
    {
      "train_step": 3884,
      "sup_loss": 0.5342014148581647
    },
    {
      "train_step": 3885,
      "sup_loss": 0.6719932364207551
    },
    {
      "train_step": 3886,
      "sup_loss": 0.6069001383290302
    },
    {
      "train_step": 3887,
      "sup_loss": 0.514779276721508
    },
    {
      "train_step": 3888,
      "sup_loss": 0.5961669188714116
    },
    {
      "train_step": 3889,
      "sup_loss": 0.6300441094538651
    },
    {
      "train_step": 3890,
      "sup_loss": 0.7376715844410657
    },
    {
      "train_step": 3891,
      "sup_loss": 0.7921201287402079
    },
    {
      "train_step": 3892,
      "sup_loss": 0.6988229566996986
    },
    {
      "train_step": 3893,
      "sup_loss": 0.7604788780594836
    },
    {
      "train_step": 3894,
      "sup_loss": 0.6418123913277655
    },
    {
      "train_step": 3895,
      "sup_loss": 0.5590900483133442
    },
    {
      "train_step": 3896,
      "sup_loss": 0.6117725605566573
    },
    {
      "train_step": 3897,
      "sup_loss": 0.5619783956413955
    },
    {
      "train_step": 3898,
      "sup_loss": 0.6597763094805043
    },
    {
      "train_step": 3899,
      "sup_loss": 0.674762651652259
    },
    {
      "train_step": 3900,
      "sup_loss": 0.26701771841120325
    },
    {
      "train_step": 3901,
      "sup_loss": 0.7590355613379441
    },
    {
      "train_step": 3902,
      "sup_loss": 0.5749102817821885
    },
    {
      "train_step": 3903,
      "sup_loss": 0.5942905292427588
    },
    {
      "train_step": 3904,
      "sup_loss": 0.693104213278496
    },
    {
      "train_step": 3905,
      "sup_loss": 0.5605042572879336
    },
    {
      "train_step": 3906,
      "sup_loss": 0.4881553106456233
    },
    {
      "train_step": 3907,
      "sup_loss": 0.5852050340664242
    },
    {
      "train_step": 3908,
      "sup_loss": 0.7883035888593422
    },
    {
      "train_step": 3909,
      "sup_loss": 0.5376289595591799
    },
    {
      "train_step": 3910,
      "sup_loss": 0.6576614966669305
    },
    {
      "train_step": 3911,
      "sup_loss": 0.5833428518841335
    },
    {
      "train_step": 3912,
      "sup_loss": 0.5060685072435571
    },
    {
      "train_step": 3913,
      "sup_loss": 0.8485003284539787
    },
    {
      "train_step": 3914,
      "sup_loss": 0.6382821448778262
    },
    {
      "train_step": 3915,
      "sup_loss": 0.7821326723781438
    },
    {
      "train_step": 3916,
      "sup_loss": 0.5570708820369096
    },
    {
      "train_step": 3917,
      "sup_loss": 0.5450661461951999
    },
    {
      "train_step": 3918,
      "sup_loss": 0.4947208983136943
    },
    {
      "train_step": 3919,
      "sup_loss": 0.6011338943636876
    },
    {
      "train_step": 3920,
      "sup_loss": 0.5393336241330238
    },
    {
      "train_step": 3921,
      "sup_loss": 0.6359178076348948
    },
    {
      "train_step": 3922,
      "sup_loss": 0.5264883545913142
    },
    {
      "train_step": 3923,
      "sup_loss": 0.627901929333606
    },
    {
      "train_step": 3924,
      "sup_loss": 0.6591146463257499
    },
    {
      "train_step": 3925,
      "sup_loss": 0.4920924556172379
    },
    {
      "train_step": 3926,
      "sup_loss": 0.7850497812082695
    },
    {
      "train_step": 3927,
      "sup_loss": 0.8519306068767452
    },
    {
      "train_step": 3928,
      "sup_loss": 0.7901754597694413
    },
    {
      "train_step": 3929,
      "sup_loss": 0.5670961192156846
    },
    {
      "train_step": 3930,
      "sup_loss": 0.6250244744225678
    },
    {
      "train_step": 3931,
      "sup_loss": 0.4096473637283782
    },
    {
      "train_step": 3932,
      "sup_loss": 0.6379303665081998
    },
    {
      "train_step": 3933,
      "sup_loss": 0.7672165408509526
    },
    {
      "train_step": 3934,
      "sup_loss": 0.5279299382007628
    },
    {
      "train_step": 3935,
      "sup_loss": 0.6416317942018597
    },
    {
      "train_step": 3936,
      "sup_loss": 0.46121021939364093
    },
    {
      "train_step": 3937,
      "sup_loss": 0.5746375075137605
    },
    {
      "train_step": 3938,
      "sup_loss": 0.29001784226999217
    },
    {
      "train_step": 3939,
      "sup_loss": 0.5792592988194751
    },
    {
      "train_step": 3940,
      "sup_loss": 0.6286667345873344
    },
    {
      "train_step": 3941,
      "sup_loss": 0.7519613624827735
    },
    {
      "train_step": 3942,
      "sup_loss": 0.5534791845555201
    },
    {
      "train_step": 3943,
      "sup_loss": 0.5600321535627609
    },
    {
      "train_step": 3944,
      "sup_loss": 0.7075733225494723
    },
    {
      "train_step": 3945,
      "sup_loss": 0.7599264536263804
    },
    {
      "train_step": 3946,
      "sup_loss": 0.5892218318591763
    },
    {
      "train_step": 3947,
      "sup_loss": 0.6078920744357541
    },
    {
      "train_step": 3948,
      "sup_loss": 0.3264827154940459
    },
    {
      "train_step": 3949,
      "sup_loss": 0.5676362556616913
    },
    {
      "train_step": 3950,
      "sup_loss": 0.5721787968104398
    },
    {
      "train_step": 3951,
      "sup_loss": 0.6973828234977438
    },
    {
      "train_step": 3952,
      "sup_loss": 0.4477212558772421
    },
    {
      "train_step": 3953,
      "sup_loss": 0.3806602434860605
    },
    {
      "train_step": 3954,
      "sup_loss": 0.661238083892636
    },
    {
      "train_step": 3955,
      "sup_loss": 0.51595507777614
    },
    {
      "train_step": 3956,
      "sup_loss": 0.6009133042496599
    },
    {
      "train_step": 3957,
      "sup_loss": 0.5567236240211325
    },
    {
      "train_step": 3958,
      "sup_loss": 0.7205812976553079
    },
    {
      "train_step": 3959,
      "sup_loss": 0.39339153710328206
    },
    {
      "train_step": 3960,
      "sup_loss": 0.665212337185576
    },
    {
      "train_step": 3961,
      "sup_loss": 0.5207475047430232
    },
    {
      "train_step": 3962,
      "sup_loss": 0.6966251053994601
    },
    {
      "train_step": 3963,
      "sup_loss": 0.756467705044177
    },
    {
      "train_step": 3964,
      "sup_loss": 0.6528906801784585
    },
    {
      "train_step": 3965,
      "sup_loss": 0.6191353246118121
    },
    {
      "train_step": 3966,
      "sup_loss": 0.3693642653220873
    },
    {
      "train_step": 3967,
      "sup_loss": 0.5890362832363857
    },
    {
      "train_step": 3968,
      "sup_loss": 0.8000556781761672
    },
    {
      "train_step": 3969,
      "sup_loss": 0.676265602651029
    },
    {
      "train_step": 3970,
      "sup_loss": 0.4792473660453872
    },
    {
      "train_step": 3971,
      "sup_loss": 0.7923448830916026
    },
    {
      "train_step": 3972,
      "sup_loss": 0.6179819282169144
    },
    {
      "train_step": 3973,
      "sup_loss": 0.5878968871968848
    },
    {
      "train_step": 3974,
      "sup_loss": 0.6536605995207936
    },
    {
      "train_step": 3975,
      "sup_loss": 0.6538120136054513
    },
    {
      "train_step": 3976,
      "sup_loss": 0.4727040932176494
    },
    {
      "train_step": 3977,
      "sup_loss": 0.6523685172639322
    },
    {
      "train_step": 3978,
      "sup_loss": 0.7286676190255522
    },
    {
      "train_step": 3979,
      "sup_loss": 0.9092483528629336
    },
    {
      "train_step": 3980,
      "sup_loss": 0.5479995864864452
    },
    {
      "train_step": 3981,
      "sup_loss": 0.5697945961731861
    },
    {
      "train_step": 3982,
      "sup_loss": 0.4209776241464425
    },
    {
      "train_step": 3983,
      "sup_loss": 0.6585665726157859
    },
    {
      "train_step": 3984,
      "sup_loss": 0.632041941654329
    },
    {
      "train_step": 3985,
      "sup_loss": 0.6065771677925026
    },
    {
      "train_step": 3986,
      "sup_loss": 0.6685092909593608
    },
    {
      "train_step": 3987,
      "sup_loss": 0.6486891338837073
    },
    {
      "train_step": 3988,
      "sup_loss": 0.7062611557596714
    },
    {
      "train_step": 3989,
      "sup_loss": 0.5309283608248727
    },
    {
      "train_step": 3990,
      "sup_loss": 0.6458398559849953
    },
    {
      "train_step": 3991,
      "sup_loss": 0.675080066306953
    },
    {
      "train_step": 3992,
      "sup_loss": 0.6026752386846262
    },
    {
      "train_step": 3993,
      "sup_loss": 0.4658441497227151
    },
    {
      "train_step": 3994,
      "sup_loss": 0.6191851816692066
    },
    {
      "train_step": 3995,
      "sup_loss": 1.2551060021140044
    },
    {
      "train_step": 3996,
      "sup_loss": 0.516315835763302
    },
    {
      "train_step": 3997,
      "sup_loss": 0.7225879785691336
    },
    {
      "train_step": 3998,
      "sup_loss": 0.5858276184656149
    },
    {
      "train_step": 3999,
      "sup_loss": 0.4344032473487623
    },
    {
      "train_step": 4000,
      "sup_loss": 0.7517799129832043
    },
    {
      "train_step": 4001,
      "sup_loss": 0.4944569329007691
    },
    {
      "train_step": 4002,
      "sup_loss": 0.7202795254420771
    },
    {
      "train_step": 4003,
      "sup_loss": 0.5596390169641918
    },
    {
      "train_step": 4004,
      "sup_loss": 0.5409869630248861
    },
    {
      "train_step": 4005,
      "sup_loss": 0.6909301331898439
    },
    {
      "train_step": 4006,
      "sup_loss": 0.5873447160506057
    },
    {
      "train_step": 4007,
      "sup_loss": 0.6167236957403059
    },
    {
      "train_step": 4008,
      "sup_loss": 0.5034108499231632
    },
    {
      "train_step": 4009,
      "sup_loss": 0.5146520037741243
    },
    {
      "train_step": 4010,
      "sup_loss": 0.6811286557319731
    },
    {
      "train_step": 4011,
      "sup_loss": 0.5176444316150587
    },
    {
      "train_step": 4012,
      "sup_loss": 0.7751491958280696
    },
    {
      "train_step": 4013,
      "sup_loss": 0.5279305488841279
    },
    {
      "train_step": 4014,
      "sup_loss": 0.7137930870311437
    },
    {
      "train_step": 4015,
      "sup_loss": 0.5532679602128333
    },
    {
      "train_step": 4016,
      "sup_loss": 0.667866898374617
    },
    {
      "train_step": 4017,
      "sup_loss": 0.6356471432329389
    },
    {
      "train_step": 4018,
      "sup_loss": 0.5489081428847772
    },
    {
      "train_step": 4019,
      "sup_loss": 0.5856812348978756
    },
    {
      "train_step": 4020,
      "sup_loss": 0.4725550798288661
    },
    {
      "train_step": 4021,
      "sup_loss": 0.6911419026423796
    },
    {
      "train_step": 4022,
      "sup_loss": 0.40075599009889795
    },
    {
      "train_step": 4023,
      "sup_loss": 0.4491016552750284
    },
    {
      "train_step": 4024,
      "sup_loss": 0.5053067447799139
    },
    {
      "train_step": 4025,
      "sup_loss": 0.5346064293286135
    },
    {
      "train_step": 4026,
      "sup_loss": 0.8206967856161276
    },
    {
      "train_step": 4027,
      "sup_loss": 0.6694665366954546
    },
    {
      "train_step": 4028,
      "sup_loss": 0.6714102352519905
    },
    {
      "train_step": 4029,
      "sup_loss": 0.3594314851089599
    },
    {
      "train_step": 4030,
      "sup_loss": 0.8779678406457542
    },
    {
      "train_step": 4031,
      "sup_loss": 0.4239919207249446
    },
    {
      "train_step": 4032,
      "sup_loss": 0.3996927918235755
    },
    {
      "train_step": 4033,
      "sup_loss": 0.6435027793447327
    },
    {
      "train_step": 4034,
      "sup_loss": 0.6018515260168947
    },
    {
      "train_step": 4035,
      "sup_loss": 0.4212487550873747
    },
    {
      "train_step": 4036,
      "sup_loss": 0.754236810764008
    },
    {
      "train_step": 4037,
      "sup_loss": 0.5975066375728998
    },
    {
      "train_step": 4038,
      "sup_loss": 0.5373694269568783
    },
    {
      "train_step": 4039,
      "sup_loss": 0.47556024293942456
    },
    {
      "train_step": 4040,
      "sup_loss": 0.7461692921448924
    },
    {
      "train_step": 4041,
      "sup_loss": 0.43250992284042816
    },
    {
      "train_step": 4042,
      "sup_loss": 0.5945520763514385
    },
    {
      "train_step": 4043,
      "sup_loss": 0.7379939514811443
    },
    {
      "train_step": 4044,
      "sup_loss": 0.5520486617724366
    },
    {
      "train_step": 4045,
      "sup_loss": 0.23267999817008161
    },
    {
      "train_step": 4046,
      "sup_loss": 0.6752109923603247
    },
    {
      "train_step": 4047,
      "sup_loss": 0.5286182506739425
    },
    {
      "train_step": 4048,
      "sup_loss": 0.6554488841953946
    },
    {
      "train_step": 4049,
      "sup_loss": 0.5549652627649239
    },
    {
      "train_step": 4050,
      "sup_loss": 0.7932884942314891
    },
    {
      "train_step": 4051,
      "sup_loss": 0.5977022617087547
    },
    {
      "train_step": 4052,
      "sup_loss": 0.3888931388447699
    },
    {
      "train_step": 4053,
      "sup_loss": 0.6956296763289227
    },
    {
      "train_step": 4054,
      "sup_loss": 0.5485149076594068
    },
    {
      "train_step": 4055,
      "sup_loss": 0.6312086711994345
    },
    {
      "train_step": 4056,
      "sup_loss": 0.46428945314306075
    },
    {
      "train_step": 4057,
      "sup_loss": 0.6748408418392781
    },
    {
      "train_step": 4058,
      "sup_loss": 0.6832985168771601
    },
    {
      "train_step": 4059,
      "sup_loss": 0.6565117954794892
    },
    {
      "train_step": 4060,
      "sup_loss": 0.41564316564849013
    },
    {
      "train_step": 4061,
      "sup_loss": 0.6453862519179039
    },
    {
      "train_step": 4062,
      "sup_loss": 0.6086728359072976
    },
    {
      "train_step": 4063,
      "sup_loss": 0.6118810095713414
    },
    {
      "train_step": 4064,
      "sup_loss": 0.62551754432764
    },
    {
      "train_step": 4065,
      "sup_loss": 0.5440811146560208
    },
    {
      "train_step": 4066,
      "sup_loss": 0.591810096292244
    },
    {
      "train_step": 4067,
      "sup_loss": 0.7320776971318868
    },
    {
      "train_step": 4068,
      "sup_loss": 0.5318495119413775
    },
    {
      "train_step": 4069,
      "sup_loss": 0.6007060228900473
    },
    {
      "train_step": 4070,
      "sup_loss": 0.5448652183031406
    },
    {
      "train_step": 4071,
      "sup_loss": 0.47892983136324807
    },
    {
      "train_step": 4072,
      "sup_loss": 0.4928940564133053
    },
    {
      "train_step": 4073,
      "sup_loss": 0.6275835422746099
    },
    {
      "train_step": 4074,
      "sup_loss": 0.7468900069919252
    },
    {
      "train_step": 4075,
      "sup_loss": 0.7231659613349228
    },
    {
      "train_step": 4076,
      "sup_loss": 0.4761743121508599
    },
    {
      "train_step": 4077,
      "sup_loss": 0.5368847277054176
    },
    {
      "train_step": 4078,
      "sup_loss": 0.8143781165833603
    },
    {
      "train_step": 4079,
      "sup_loss": 0.4887550598854397
    },
    {
      "train_step": 4080,
      "sup_loss": 0.6397394618808143
    },
    {
      "train_step": 4081,
      "sup_loss": 0.52912437116342
    },
    {
      "train_step": 4082,
      "sup_loss": 0.32267222376319504
    },
    {
      "train_step": 4083,
      "sup_loss": 0.8361453703747255
    },
    {
      "train_step": 4084,
      "sup_loss": 0.7205195344848131
    },
    {
      "train_step": 4085,
      "sup_loss": 0.5551952124491734
    },
    {
      "train_step": 4086,
      "sup_loss": 0.4716103057184
    },
    {
      "train_step": 4087,
      "sup_loss": 0.618113428796541
    },
    {
      "train_step": 4088,
      "sup_loss": 0.7209061572336211
    },
    {
      "train_step": 4089,
      "sup_loss": 0.6210540820286494
    },
    {
      "train_step": 4090,
      "sup_loss": 0.7254819866705638
    },
    {
      "train_step": 4091,
      "sup_loss": 0.6081726537285413
    },
    {
      "train_step": 4092,
      "sup_loss": 0.4708732185219391
    },
    {
      "train_step": 4093,
      "sup_loss": 0.7283321047817899
    },
    {
      "train_step": 4094,
      "sup_loss": 0.6269952298350271
    },
    {
      "train_step": 4095,
      "sup_loss": 0.6038726604780074
    },
    {
      "train_step": 4096,
      "sup_loss": 0.6038820191603251
    },
    {
      "train_step": 4097,
      "sup_loss": 0.4139951462710961
    },
    {
      "train_step": 4098,
      "sup_loss": 0.388718774108273
    },
    {
      "train_step": 4099,
      "sup_loss": 0.4048647182002212
    },
    {
      "train_step": 4100,
      "sup_loss": 0.6444801696468532
    },
    {
      "train_step": 4101,
      "sup_loss": 0.563314887291658
    },
    {
      "train_step": 4102,
      "sup_loss": 0.2676875323793311
    },
    {
      "train_step": 4103,
      "sup_loss": 0.5372396907567102
    },
    {
      "train_step": 4104,
      "sup_loss": 0.7277213797473585
    },
    {
      "train_step": 4105,
      "sup_loss": 0.5286789448427146
    },
    {
      "train_step": 4106,
      "sup_loss": 0.2629337109758288
    },
    {
      "train_step": 4107,
      "sup_loss": 0.7047940738744137
    },
    {
      "train_step": 4108,
      "sup_loss": 0.39766155000309694
    },
    {
      "train_step": 4109,
      "sup_loss": 0.5992585363794471
    },
    {
      "train_step": 4110,
      "sup_loss": 0.7796843946081453
    },
    {
      "train_step": 4111,
      "sup_loss": 0.6295681976214553
    },
    {
      "train_step": 4112,
      "sup_loss": 0.4324282503626048
    },
    {
      "train_step": 4113,
      "sup_loss": 0.5048461227363179
    },
    {
      "train_step": 4114,
      "sup_loss": 0.7599798220849475
    },
    {
      "train_step": 4115,
      "sup_loss": 0.5398380086951949
    },
    {
      "train_step": 4116,
      "sup_loss": 0.5217545135890644
    },
    {
      "train_step": 4117,
      "sup_loss": 0.5719550723335172
    },
    {
      "train_step": 4118,
      "sup_loss": 0.6379215834675591
    },
    {
      "train_step": 4119,
      "sup_loss": 0.45767084165837335
    },
    {
      "train_step": 4120,
      "sup_loss": 0.7678301277935663
    },
    {
      "train_step": 4121,
      "sup_loss": 0.3737239260189399
    },
    {
      "train_step": 4122,
      "sup_loss": 0.5270928349683889
    },
    {
      "train_step": 4123,
      "sup_loss": 0.7278796290059584
    },
    {
      "train_step": 4124,
      "sup_loss": 0.4187924420446267
    },
    {
      "train_step": 4125,
      "sup_loss": 0.7257868327591217
    },
    {
      "train_step": 4126,
      "sup_loss": 0.7197186847483924
    },
    {
      "train_step": 4127,
      "sup_loss": 0.4998785984480997
    },
    {
      "train_step": 4128,
      "sup_loss": 0.7973723498892318
    },
    {
      "train_step": 4129,
      "sup_loss": 0.7345595991366621
    },
    {
      "train_step": 4130,
      "sup_loss": 0.5460029733173896
    },
    {
      "train_step": 4131,
      "sup_loss": 0.30964322753259116
    },
    {
      "train_step": 4132,
      "sup_loss": 0.563369715439954
    },
    {
      "train_step": 4133,
      "sup_loss": 0.5287972532048627
    },
    {
      "train_step": 4134,
      "sup_loss": 0.793681031919753
    },
    {
      "train_step": 4135,
      "sup_loss": 0.8475591069915637
    },
    {
      "train_step": 4136,
      "sup_loss": 0.3161977440788332
    },
    {
      "train_step": 4137,
      "sup_loss": 0.5548762775202478
    },
    {
      "train_step": 4138,
      "sup_loss": 0.4159663354067992
    },
    {
      "train_step": 4139,
      "sup_loss": 0.35369429075811504
    },
    {
      "train_step": 4140,
      "sup_loss": 0.7305925242081647
    },
    {
      "train_step": 4141,
      "sup_loss": 0.6232823825455627
    },
    {
      "train_step": 4142,
      "sup_loss": 0.3994485963634385
    },
    {
      "train_step": 4143,
      "sup_loss": 0.37976726657727194
    },
    {
      "train_step": 4144,
      "sup_loss": 0.7098513772937998
    },
    {
      "train_step": 4145,
      "sup_loss": 0.5614451026294679
    },
    {
      "train_step": 4146,
      "sup_loss": 0.3870722532842018
    },
    {
      "train_step": 4147,
      "sup_loss": 0.49009728236744887
    },
    {
      "train_step": 4148,
      "sup_loss": 0.5883435591699342
    },
    {
      "train_step": 4149,
      "sup_loss": 0.4608332587756434
    },
    {
      "train_step": 4150,
      "sup_loss": 0.6688931226947089
    },
    {
      "train_step": 4151,
      "sup_loss": 0.5941659217309389
    },
    {
      "train_step": 4152,
      "sup_loss": 0.5407391554194568
    },
    {
      "train_step": 4153,
      "sup_loss": 0.5518751000453735
    },
    {
      "train_step": 4154,
      "sup_loss": 0.6086920761223085
    },
    {
      "train_step": 4155,
      "sup_loss": 0.6460330098355739
    },
    {
      "train_step": 4156,
      "sup_loss": 0.5066960284448362
    },
    {
      "train_step": 4157,
      "sup_loss": 0.5281217841284096
    },
    {
      "train_step": 4158,
      "sup_loss": 0.648505033812452
    },
    {
      "train_step": 4159,
      "sup_loss": 0.5070061379492713
    },
    {
      "train_step": 4160,
      "sup_loss": 0.44655095366507896
    },
    {
      "train_step": 4161,
      "sup_loss": 0.8699363401159538
    },
    {
      "train_step": 4162,
      "sup_loss": 0.3864350787436231
    },
    {
      "train_step": 4163,
      "sup_loss": 0.41598667731201655
    },
    {
      "train_step": 4164,
      "sup_loss": 0.6882499600450508
    },
    {
      "train_step": 4165,
      "sup_loss": 0.48749979160772927
    },
    {
      "train_step": 4166,
      "sup_loss": 0.5640477694076621
    },
    {
      "train_step": 4167,
      "sup_loss": 0.6710833134157067
    },
    {
      "train_step": 4168,
      "sup_loss": 0.6267924057876139
    },
    {
      "train_step": 4169,
      "sup_loss": 0.6364334589457448
    },
    {
      "train_step": 4170,
      "sup_loss": 0.3944007311773222
    },
    {
      "train_step": 4171,
      "sup_loss": 0.4541893340447123
    },
    {
      "train_step": 4172,
      "sup_loss": 0.6445118530111166
    },
    {
      "train_step": 4173,
      "sup_loss": 0.6651631277959311
    },
    {
      "train_step": 4174,
      "sup_loss": 0.5857498337586129
    },
    {
      "train_step": 4175,
      "sup_loss": 0.4552033120394341
    },
    {
      "train_step": 4176,
      "sup_loss": 0.723330191272274
    },
    {
      "train_step": 4177,
      "sup_loss": 0.44466136419592506
    },
    {
      "train_step": 4178,
      "sup_loss": 0.5289418980813073
    },
    {
      "train_step": 4179,
      "sup_loss": 0.7261938468931474
    },
    {
      "train_step": 4180,
      "sup_loss": 0.46632464161995835
    },
    {
      "train_step": 4181,
      "sup_loss": 0.6207120020070915
    },
    {
      "train_step": 4182,
      "sup_loss": 0.5679079003021766
    },
    {
      "train_step": 4183,
      "sup_loss": 0.40288179212013325
    },
    {
      "train_step": 4184,
      "sup_loss": 0.7569281681005361
    },
    {
      "train_step": 4185,
      "sup_loss": 0.5117554143506968
    },
    {
      "train_step": 4186,
      "sup_loss": 0.5403215967969517
    },
    {
      "train_step": 4187,
      "sup_loss": 0.6191493414717794
    },
    {
      "train_step": 4188,
      "sup_loss": 0.5387817799009091
    },
    {
      "train_step": 4189,
      "sup_loss": 0.6922842582133437
    },
    {
      "train_step": 4190,
      "sup_loss": 0.6572800724931328
    },
    {
      "train_step": 4191,
      "sup_loss": 0.4174504003796423
    },
    {
      "train_step": 4192,
      "sup_loss": 0.5936883144337173
    },
    {
      "train_step": 4193,
      "sup_loss": 0.5492091389923992
    },
    {
      "train_step": 4194,
      "sup_loss": 0.6593407226999716
    },
    {
      "train_step": 4195,
      "sup_loss": 0.597879110586368
    },
    {
      "train_step": 4196,
      "sup_loss": 0.45520692041903543
    },
    {
      "train_step": 4197,
      "sup_loss": 0.640643309851111
    },
    {
      "train_step": 4198,
      "sup_loss": 0.7623555377329762
    },
    {
      "train_step": 4199,
      "sup_loss": 0.5894486737922675
    },
    {
      "train_step": 4200,
      "sup_loss": 0.5267521127540807
    },
    {
      "train_step": 4201,
      "sup_loss": 0.607110838515437
    },
    {
      "train_step": 4202,
      "sup_loss": 0.6970967999985827
    },
    {
      "train_step": 4203,
      "sup_loss": 0.5514011572547344
    },
    {
      "train_step": 4204,
      "sup_loss": 0.6340453051500158
    },
    {
      "train_step": 4205,
      "sup_loss": 0.5222354177879318
    },
    {
      "train_step": 4206,
      "sup_loss": 0.6052355600102519
    },
    {
      "train_step": 4207,
      "sup_loss": 0.4598635495192086
    },
    {
      "train_step": 4208,
      "sup_loss": 0.5618325870242573
    },
    {
      "train_step": 4209,
      "sup_loss": 0.5730280824874301
    },
    {
      "train_step": 4210,
      "sup_loss": 0.5986815531287117
    },
    {
      "train_step": 4211,
      "sup_loss": 0.520565727078399
    },
    {
      "train_step": 4212,
      "sup_loss": 0.34048927375570737
    },
    {
      "train_step": 4213,
      "sup_loss": 0.4773863629754449
    },
    {
      "train_step": 4214,
      "sup_loss": 0.7083218122261468
    },
    {
      "train_step": 4215,
      "sup_loss": 0.5778265863532178
    },
    {
      "train_step": 4216,
      "sup_loss": 0.6879533166479215
    },
    {
      "train_step": 4217,
      "sup_loss": 0.4785187933739991
    },
    {
      "train_step": 4218,
      "sup_loss": 0.4620257900673145
    },
    {
      "train_step": 4219,
      "sup_loss": 0.6118263996187714
    },
    {
      "train_step": 4220,
      "sup_loss": 0.6295381632272609
    },
    {
      "train_step": 4221,
      "sup_loss": 0.55982400227831
    },
    {
      "train_step": 4222,
      "sup_loss": 0.5748361866102031
    },
    {
      "train_step": 4223,
      "sup_loss": 0.6409690582656975
    },
    {
      "train_step": 4224,
      "sup_loss": 0.6382575347125371
    },
    {
      "train_step": 4225,
      "sup_loss": 0.5795071489377474
    },
    {
      "train_step": 4226,
      "sup_loss": 0.5082313812305927
    },
    {
      "train_step": 4227,
      "sup_loss": 0.5960434019728803
    },
    {
      "train_step": 4228,
      "sup_loss": 0.5802998048179228
    },
    {
      "train_step": 4229,
      "sup_loss": 0.4632969498302407
    },
    {
      "train_step": 4230,
      "sup_loss": 0.5954710342878939
    },
    {
      "train_step": 4231,
      "sup_loss": 0.6648424601988575
    },
    {
      "train_step": 4232,
      "sup_loss": 0.5387320167745738
    },
    {
      "train_step": 4233,
      "sup_loss": 0.5912400190457745
    },
    {
      "train_step": 4234,
      "sup_loss": 1.004176563747531
    },
    {
      "train_step": 4235,
      "sup_loss": 0.7439518982654325
    },
    {
      "train_step": 4236,
      "sup_loss": 0.7125012797958827
    },
    {
      "train_step": 4237,
      "sup_loss": 0.5265249258996905
    },
    {
      "train_step": 4238,
      "sup_loss": 0.6235659358505427
    },
    {
      "train_step": 4239,
      "sup_loss": 0.6442193251741992
    },
    {
      "train_step": 4240,
      "sup_loss": 0.6158607036390547
    },
    {
      "train_step": 4241,
      "sup_loss": 0.6229792037818619
    },
    {
      "train_step": 4242,
      "sup_loss": 0.5546469958937708
    },
    {
      "train_step": 4243,
      "sup_loss": 0.5791131062374628
    },
    {
      "train_step": 4244,
      "sup_loss": 0.48048107098019743
    },
    {
      "train_step": 4245,
      "sup_loss": 0.6178033263928338
    },
    {
      "train_step": 4246,
      "sup_loss": 0.638703318289825
    },
    {
      "train_step": 4247,
      "sup_loss": 0.6430756811950367
    },
    {
      "train_step": 4248,
      "sup_loss": 0.6533702087539426
    },
    {
      "train_step": 4249,
      "sup_loss": 0.4251366270196019
    },
    {
      "train_step": 4250,
      "sup_loss": 0.6386028878974707
    },
    {
      "train_step": 4251,
      "sup_loss": 0.24540758218757144
    },
    {
      "train_step": 4252,
      "sup_loss": 0.5275628002600128
    },
    {
      "train_step": 4253,
      "sup_loss": 0.618811526380544
    },
    {
      "train_step": 4254,
      "sup_loss": 0.8002682017184695
    },
    {
      "train_step": 4255,
      "sup_loss": 0.6539523705582784
    },
    {
      "train_step": 4256,
      "sup_loss": 0.5422161358056103
    },
    {
      "train_step": 4257,
      "sup_loss": 0.7555336826917537
    },
    {
      "train_step": 4258,
      "sup_loss": 0.5675417793575542
    },
    {
      "train_step": 4259,
      "sup_loss": 0.46390622324021624
    },
    {
      "train_step": 4260,
      "sup_loss": 0.6409554193094982
    },
    {
      "train_step": 4261,
      "sup_loss": 0.4842675053540223
    },
    {
      "train_step": 4262,
      "sup_loss": 0.8173651373479777
    },
    {
      "train_step": 4263,
      "sup_loss": 0.6107953703515773
    },
    {
      "train_step": 4264,
      "sup_loss": 0.5626951156355985
    },
    {
      "train_step": 4265,
      "sup_loss": 0.5638315898078099
    },
    {
      "train_step": 4266,
      "sup_loss": 0.4306782824383916
    },
    {
      "train_step": 4267,
      "sup_loss": 0.6150164335213073
    },
    {
      "train_step": 4268,
      "sup_loss": 0.9158692360966761
    },
    {
      "train_step": 4269,
      "sup_loss": 0.4699789509896858
    },
    {
      "train_step": 4270,
      "sup_loss": 0.6991134329229418
    },
    {
      "train_step": 4271,
      "sup_loss": 0.7563354232452394
    },
    {
      "train_step": 4272,
      "sup_loss": 0.5683281106199547
    },
    {
      "train_step": 4273,
      "sup_loss": 0.7501494702655367
    },
    {
      "train_step": 4274,
      "sup_loss": 0.5156759837075324
    },
    {
      "train_step": 4275,
      "sup_loss": 0.6564252058899238
    },
    {
      "train_step": 4276,
      "sup_loss": 0.4491305829884359
    },
    {
      "train_step": 4277,
      "sup_loss": 0.7568904841784028
    },
    {
      "train_step": 4278,
      "sup_loss": 0.740228115741076
    },
    {
      "train_step": 4279,
      "sup_loss": 0.7508280224522499
    },
    {
      "train_step": 4280,
      "sup_loss": 0.5286570304226188
    },
    {
      "train_step": 4281,
      "sup_loss": 0.4588994122214206
    },
    {
      "train_step": 4282,
      "sup_loss": 0.6207077557161214
    },
    {
      "train_step": 4283,
      "sup_loss": 0.38299895415009516
    },
    {
      "train_step": 4284,
      "sup_loss": 0.5340732843870546
    },
    {
      "train_step": 4285,
      "sup_loss": 0.5804049431798732
    },
    {
      "train_step": 4286,
      "sup_loss": 0.5148731324509922
    },
    {
      "train_step": 4287,
      "sup_loss": 0.5242914222101217
    },
    {
      "train_step": 4288,
      "sup_loss": 0.6696004520889118
    },
    {
      "train_step": 4289,
      "sup_loss": 0.5707609442629857
    },
    {
      "train_step": 4290,
      "sup_loss": 0.7043120253477532
    },
    {
      "train_step": 4291,
      "sup_loss": 0.7092326538518687
    },
    {
      "train_step": 4292,
      "sup_loss": 0.374808938533191
    },
    {
      "train_step": 4293,
      "sup_loss": 0.6681503389523928
    },
    {
      "train_step": 4294,
      "sup_loss": 0.716651648428501
    },
    {
      "train_step": 4295,
      "sup_loss": 0.6967861710169957
    },
    {
      "train_step": 4296,
      "sup_loss": 0.5109502290657244
    },
    {
      "train_step": 4297,
      "sup_loss": 0.49690364039727936
    },
    {
      "train_step": 4298,
      "sup_loss": 0.6036687671884804
    },
    {
      "train_step": 4299,
      "sup_loss": 0.46956620606520827
    },
    {
      "train_step": 4300,
      "sup_loss": 0.5073584216467345
    },
    {
      "train_step": 4301,
      "sup_loss": 0.383335821935938
    },
    {
      "train_step": 4302,
      "sup_loss": 0.5326432611674866
    },
    {
      "train_step": 4303,
      "sup_loss": 0.6016087080142735
    },
    {
      "train_step": 4304,
      "sup_loss": 0.48371864195307485
    },
    {
      "train_step": 4305,
      "sup_loss": 0.493400184300453
    },
    {
      "train_step": 4306,
      "sup_loss": 0.6741285943405504
    },
    {
      "train_step": 4307,
      "sup_loss": 0.8257656194021613
    },
    {
      "train_step": 4308,
      "sup_loss": 0.5012205582319315
    },
    {
      "train_step": 4309,
      "sup_loss": 0.5931894487892956
    },
    {
      "train_step": 4310,
      "sup_loss": 0.41918374549287596
    },
    {
      "train_step": 4311,
      "sup_loss": 0.568224123568165
    },
    {
      "train_step": 4312,
      "sup_loss": 0.4980405585081235
    },
    {
      "train_step": 4313,
      "sup_loss": 0.5761685994050985
    },
    {
      "train_step": 4314,
      "sup_loss": 0.6673369382806146
    },
    {
      "train_step": 4315,
      "sup_loss": 0.5497210323127819
    },
    {
      "train_step": 4316,
      "sup_loss": 0.4764506014673764
    },
    {
      "train_step": 4317,
      "sup_loss": 0.33975654999031046
    },
    {
      "train_step": 4318,
      "sup_loss": 0.49528571543556676
    },
    {
      "train_step": 4319,
      "sup_loss": 0.48505257241259747
    },
    {
      "train_step": 4320,
      "sup_loss": 0.596380423211957
    },
    {
      "train_step": 4321,
      "sup_loss": 0.6440917755790921
    },
    {
      "train_step": 4322,
      "sup_loss": 0.5314911752501044
    },
    {
      "train_step": 4323,
      "sup_loss": 0.5664113495852414
    },
    {
      "train_step": 4324,
      "sup_loss": 0.6064885107547542
    },
    {
      "train_step": 4325,
      "sup_loss": 0.7280495485863424
    },
    {
      "train_step": 4326,
      "sup_loss": 0.649210516965383
    },
    {
      "train_step": 4327,
      "sup_loss": 0.3818550121732392
    },
    {
      "train_step": 4328,
      "sup_loss": 0.43887812481072
    },
    {
      "train_step": 4329,
      "sup_loss": 0.5323765032372646
    },
    {
      "train_step": 4330,
      "sup_loss": 0.5344035381008163
    },
    {
      "train_step": 4331,
      "sup_loss": 0.784848985758154
    },
    {
      "train_step": 4332,
      "sup_loss": 0.5084786483627458
    },
    {
      "train_step": 4333,
      "sup_loss": 0.5952674222536364
    },
    {
      "train_step": 4334,
      "sup_loss": 0.5101422933828206
    },
    {
      "train_step": 4335,
      "sup_loss": 0.8619388760541906
    },
    {
      "train_step": 4336,
      "sup_loss": 0.6050606452201837
    },
    {
      "train_step": 4337,
      "sup_loss": 0.4873360703299982
    },
    {
      "train_step": 4338,
      "sup_loss": 0.6355558617654702
    },
    {
      "train_step": 4339,
      "sup_loss": 0.7268955031794494
    },
    {
      "train_step": 4340,
      "sup_loss": 0.5672188626106793
    },
    {
      "train_step": 4341,
      "sup_loss": 0.7389119578844903
    },
    {
      "train_step": 4342,
      "sup_loss": 0.5804349675056716
    },
    {
      "train_step": 4343,
      "sup_loss": 0.48692417722872056
    },
    {
      "train_step": 4344,
      "sup_loss": 0.5069755271585848
    },
    {
      "train_step": 4345,
      "sup_loss": 0.7459072008538926
    },
    {
      "train_step": 4346,
      "sup_loss": 0.5080167323462554
    },
    {
      "train_step": 4347,
      "sup_loss": 0.7027730113916903
    },
    {
      "train_step": 4348,
      "sup_loss": 0.5032098529613702
    },
    {
      "train_step": 4349,
      "sup_loss": 0.7036994421002691
    },
    {
      "train_step": 4350,
      "sup_loss": 0.6844319059952878
    },
    {
      "train_step": 4351,
      "sup_loss": 0.44653714598563643
    },
    {
      "train_step": 4352,
      "sup_loss": 0.41883677876937986
    },
    {
      "train_step": 4353,
      "sup_loss": 0.48879950479446765
    },
    {
      "train_step": 4354,
      "sup_loss": 0.7150800135469965
    },
    {
      "train_step": 4355,
      "sup_loss": 0.5153916478785856
    },
    {
      "train_step": 4356,
      "sup_loss": 0.4505472959650801
    },
    {
      "train_step": 4357,
      "sup_loss": 0.6351468074397097
    },
    {
      "train_step": 4358,
      "sup_loss": 0.6899401819101243
    },
    {
      "train_step": 4359,
      "sup_loss": 0.56081254519393
    },
    {
      "train_step": 4360,
      "sup_loss": 0.6285211824441438
    },
    {
      "train_step": 4361,
      "sup_loss": 0.6842340133573317
    },
    {
      "train_step": 4362,
      "sup_loss": 0.510557842673987
    },
    {
      "train_step": 4363,
      "sup_loss": 0.7702686208891965
    },
    {
      "train_step": 4364,
      "sup_loss": 0.40482803666948985
    },
    {
      "train_step": 4365,
      "sup_loss": 0.610974032127679
    },
    {
      "train_step": 4366,
      "sup_loss": 0.40305421185662904
    },
    {
      "train_step": 4367,
      "sup_loss": 0.46033762040527404
    },
    {
      "train_step": 4368,
      "sup_loss": 0.714976725167373
    },
    {
      "train_step": 4369,
      "sup_loss": 0.6048065289613568
    },
    {
      "train_step": 4370,
      "sup_loss": 0.6102689813698073
    },
    {
      "train_step": 4371,
      "sup_loss": 0.5383447568762915
    },
    {
      "train_step": 4372,
      "sup_loss": 0.46584306620170485
    },
    {
      "train_step": 4373,
      "sup_loss": 0.39280207904373315
    },
    {
      "train_step": 4374,
      "sup_loss": 0.6896251263481684
    },
    {
      "train_step": 4375,
      "sup_loss": 0.696198535818688
    },
    {
      "train_step": 4376,
      "sup_loss": 0.3794233388173346
    },
    {
      "train_step": 4377,
      "sup_loss": 0.8901974669443008
    },
    {
      "train_step": 4378,
      "sup_loss": 0.7340346453066178
    },
    {
      "train_step": 4379,
      "sup_loss": 0.6591816038904594
    },
    {
      "train_step": 4380,
      "sup_loss": 0.5719082829321669
    },
    {
      "train_step": 4381,
      "sup_loss": 0.46483759902214217
    },
    {
      "train_step": 4382,
      "sup_loss": 0.4961884452408603
    },
    {
      "train_step": 4383,
      "sup_loss": 0.3999968369528457
    },
    {
      "train_step": 4384,
      "sup_loss": 0.5939004904400044
    },
    {
      "train_step": 4385,
      "sup_loss": 0.5298919793258633
    },
    {
      "train_step": 4386,
      "sup_loss": 0.5622232001530145
    },
    {
      "train_step": 4387,
      "sup_loss": 0.8170239660111001
    },
    {
      "train_step": 4388,
      "sup_loss": 0.7235829222506337
    },
    {
      "train_step": 4389,
      "sup_loss": 0.498134963133015
    },
    {
      "train_step": 4390,
      "sup_loss": 0.5208208613089337
    },
    {
      "train_step": 4391,
      "sup_loss": 0.5789164985949232
    },
    {
      "train_step": 4392,
      "sup_loss": 0.40851224155163757
    },
    {
      "train_step": 4393,
      "sup_loss": 0.635313061316645
    },
    {
      "train_step": 4394,
      "sup_loss": 0.5951359348509188
    },
    {
      "train_step": 4395,
      "sup_loss": 0.6004219998127428
    },
    {
      "train_step": 4396,
      "sup_loss": 0.45267337210000086
    },
    {
      "train_step": 4397,
      "sup_loss": 0.8669845555773861
    },
    {
      "train_step": 4398,
      "sup_loss": 0.6318707319930486
    },
    {
      "train_step": 4399,
      "sup_loss": 0.5741503939510413
    },
    {
      "train_step": 4400,
      "sup_loss": 0.6609636417641032
    },
    {
      "train_step": 4401,
      "sup_loss": 0.6093354472522187
    },
    {
      "train_step": 4402,
      "sup_loss": 0.42128457208177805
    },
    {
      "train_step": 4403,
      "sup_loss": 0.5788503393215445
    },
    {
      "train_step": 4404,
      "sup_loss": 0.5964402491226357
    },
    {
      "train_step": 4405,
      "sup_loss": 0.5024511376889831
    },
    {
      "train_step": 4406,
      "sup_loss": 0.55152303508773
    },
    {
      "train_step": 4407,
      "sup_loss": 0.7600553758574363
    },
    {
      "train_step": 4408,
      "sup_loss": 0.4175363692426586
    },
    {
      "train_step": 4409,
      "sup_loss": 0.5376264243986625
    },
    {
      "train_step": 4410,
      "sup_loss": 0.35784596509436906
    },
    {
      "train_step": 4411,
      "sup_loss": 0.6933956100406994
    },
    {
      "train_step": 4412,
      "sup_loss": 0.7037824531182914
    },
    {
      "train_step": 4413,
      "sup_loss": 0.5004073024103372
    },
    {
      "train_step": 4414,
      "sup_loss": 0.42465096781394673
    },
    {
      "train_step": 4415,
      "sup_loss": 0.6643489482838087
    },
    {
      "train_step": 4416,
      "sup_loss": 0.5402313437110279
    },
    {
      "train_step": 4417,
      "sup_loss": 0.6941612826206529
    },
    {
      "train_step": 4418,
      "sup_loss": 0.36439274073895256
    },
    {
      "train_step": 4419,
      "sup_loss": 0.5608643697254383
    },
    {
      "train_step": 4420,
      "sup_loss": 0.5577414969397518
    },
    {
      "train_step": 4421,
      "sup_loss": 0.5005107359572364
    },
    {
      "train_step": 4422,
      "sup_loss": 0.6111429198231519
    },
    {
      "train_step": 4423,
      "sup_loss": 0.571646899900639
    },
    {
      "train_step": 4424,
      "sup_loss": 0.6255576652851937
    },
    {
      "train_step": 4425,
      "sup_loss": 0.6994301449322914
    },
    {
      "train_step": 4426,
      "sup_loss": 0.7459502586852919
    },
    {
      "train_step": 4427,
      "sup_loss": 0.49658784009589985
    },
    {
      "train_step": 4428,
      "sup_loss": 0.6169582382057771
    },
    {
      "train_step": 4429,
      "sup_loss": 0.5628487219998275
    },
    {
      "train_step": 4430,
      "sup_loss": 0.5824765635018215
    },
    {
      "train_step": 4431,
      "sup_loss": 0.6296111268494703
    },
    {
      "train_step": 4432,
      "sup_loss": 0.694174329087293
    },
    {
      "train_step": 4433,
      "sup_loss": 0.7103531105175156
    },
    {
      "train_step": 4434,
      "sup_loss": 0.5576347433752035
    },
    {
      "train_step": 4435,
      "sup_loss": 0.5002324541746991
    },
    {
      "train_step": 4436,
      "sup_loss": 0.6980099219366156
    },
    {
      "train_step": 4437,
      "sup_loss": 0.4071935654909076
    },
    {
      "train_step": 4438,
      "sup_loss": 0.754616307626528
    },
    {
      "train_step": 4439,
      "sup_loss": 0.6093003099353576
    },
    {
      "train_step": 4440,
      "sup_loss": 0.5415657430690902
    },
    {
      "train_step": 4441,
      "sup_loss": 0.2842082999310983
    },
    {
      "train_step": 4442,
      "sup_loss": 0.5236481425756377
    },
    {
      "train_step": 4443,
      "sup_loss": 0.6801717516580418
    },
    {
      "train_step": 4444,
      "sup_loss": 0.5209619587926129
    },
    {
      "train_step": 4445,
      "sup_loss": 0.6586008061932759
    },
    {
      "train_step": 4446,
      "sup_loss": 0.46559846714776887
    },
    {
      "train_step": 4447,
      "sup_loss": 0.7600310573268527
    },
    {
      "train_step": 4448,
      "sup_loss": 0.339488415707297
    },
    {
      "train_step": 4449,
      "sup_loss": 0.4460132983548145
    },
    {
      "train_step": 4450,
      "sup_loss": 0.7822180287306034
    },
    {
      "train_step": 4451,
      "sup_loss": 0.7427193979777945
    },
    {
      "train_step": 4452,
      "sup_loss": 0.6041613152377108
    },
    {
      "train_step": 4453,
      "sup_loss": 0.5645281832078531
    },
    {
      "train_step": 4454,
      "sup_loss": 0.6981670298758341
    },
    {
      "train_step": 4455,
      "sup_loss": 0.6520238941950719
    },
    {
      "train_step": 4456,
      "sup_loss": 0.604373780990268
    },
    {
      "train_step": 4457,
      "sup_loss": 0.47732816088511587
    },
    {
      "train_step": 4458,
      "sup_loss": 0.5052255657724731
    },
    {
      "train_step": 4459,
      "sup_loss": 0.5785242880757753
    },
    {
      "train_step": 4460,
      "sup_loss": 0.48195807256196105
    },
    {
      "train_step": 4461,
      "sup_loss": 0.5138365351839186
    },
    {
      "train_step": 4462,
      "sup_loss": 0.5084197196553716
    },
    {
      "train_step": 4463,
      "sup_loss": 0.5128654377900195
    },
    {
      "train_step": 4464,
      "sup_loss": 0.5198409262536582
    },
    {
      "train_step": 4465,
      "sup_loss": 0.6623478690514353
    },
    {
      "train_step": 4466,
      "sup_loss": 0.4725117936090352
    },
    {
      "train_step": 4467,
      "sup_loss": 0.6651631693550896
    },
    {
      "train_step": 4468,
      "sup_loss": 0.4988359518437189
    },
    {
      "train_step": 4469,
      "sup_loss": 0.36252031866917384
    },
    {
      "train_step": 4470,
      "sup_loss": 0.5943268500112344
    },
    {
      "train_step": 4471,
      "sup_loss": 0.2468957272961137
    },
    {
      "train_step": 4472,
      "sup_loss": 0.7243050735648268
    },
    {
      "train_step": 4473,
      "sup_loss": 0.6047245989308156
    },
    {
      "train_step": 4474,
      "sup_loss": 0.5572766917315969
    },
    {
      "train_step": 4475,
      "sup_loss": 0.5214264243070152
    },
    {
      "train_step": 4476,
      "sup_loss": 0.8093799361624341
    },
    {
      "train_step": 4477,
      "sup_loss": 0.6909272599417562
    },
    {
      "train_step": 4478,
      "sup_loss": 0.5888524391959595
    },
    {
      "train_step": 4479,
      "sup_loss": 0.5796775786133743
    },
    {
      "train_step": 4480,
      "sup_loss": 0.4655593966097883
    },
    {
      "train_step": 4481,
      "sup_loss": 0.6579615660892209
    },
    {
      "train_step": 4482,
      "sup_loss": 0.5843668443655134
    },
    {
      "train_step": 4483,
      "sup_loss": 0.8214732777632981
    },
    {
      "train_step": 4484,
      "sup_loss": 0.6939026964600301
    },
    {
      "train_step": 4485,
      "sup_loss": 0.4455917350517831
    },
    {
      "train_step": 4486,
      "sup_loss": 0.5175886316256153
    },
    {
      "train_step": 4487,
      "sup_loss": 0.5274666148441297
    },
    {
      "train_step": 4488,
      "sup_loss": 0.660758209056754
    },
    {
      "train_step": 4489,
      "sup_loss": 0.7464532117255416
    },
    {
      "train_step": 4490,
      "sup_loss": 0.3110909361033631
    },
    {
      "train_step": 4491,
      "sup_loss": 0.6179614992730835
    },
    {
      "train_step": 4492,
      "sup_loss": 0.7433551531669066
    },
    {
      "train_step": 4493,
      "sup_loss": 0.7253796153612856
    },
    {
      "train_step": 4494,
      "sup_loss": 0.6747076903398526
    },
    {
      "train_step": 4495,
      "sup_loss": 0.5998554676920914
    },
    {
      "train_step": 4496,
      "sup_loss": 0.5135270510594189
    },
    {
      "train_step": 4497,
      "sup_loss": 0.7102711813623225
    },
    {
      "train_step": 4498,
      "sup_loss": 0.7757930702582351
    },
    {
      "train_step": 4499,
      "sup_loss": 0.4014256352935637
    },
    {
      "train_step": 4500,
      "sup_loss": 0.6204527149435132
    },
    {
      "train_step": 4501,
      "sup_loss": 0.5217221525450528
    },
    {
      "train_step": 4502,
      "sup_loss": 0.7377614585713786
    },
    {
      "train_step": 4503,
      "sup_loss": 0.558442116442782
    },
    {
      "train_step": 4504,
      "sup_loss": 0.6413273876383678
    },
    {
      "train_step": 4505,
      "sup_loss": 0.6651265964483565
    },
    {
      "train_step": 4506,
      "sup_loss": 0.4860212712709667
    },
    {
      "train_step": 4507,
      "sup_loss": 0.6986187657549718
    },
    {
      "train_step": 4508,
      "sup_loss": 0.43524869057987164
    },
    {
      "train_step": 4509,
      "sup_loss": 0.6368630625737446
    },
    {
      "train_step": 4510,
      "sup_loss": 0.5166901333867537
    },
    {
      "train_step": 4511,
      "sup_loss": 0.7160562814851605
    },
    {
      "train_step": 4512,
      "sup_loss": 0.5707297271060884
    },
    {
      "train_step": 4513,
      "sup_loss": 0.7023243672791553
    },
    {
      "train_step": 4514,
      "sup_loss": 0.5019908377542692
    },
    {
      "train_step": 4515,
      "sup_loss": 0.6497307015160482
    },
    {
      "train_step": 4516,
      "sup_loss": 0.32683038391602465
    },
    {
      "train_step": 4517,
      "sup_loss": 0.7504413872933593
    },
    {
      "train_step": 4518,
      "sup_loss": 0.5948107230372962
    },
    {
      "train_step": 4519,
      "sup_loss": 0.5572892178004951
    },
    {
      "train_step": 4520,
      "sup_loss": 0.5767538971769384
    },
    {
      "train_step": 4521,
      "sup_loss": 0.4049030087915831
    },
    {
      "train_step": 4522,
      "sup_loss": 0.4600475939824006
    },
    {
      "train_step": 4523,
      "sup_loss": 0.7661641123028328
    },
    {
      "train_step": 4524,
      "sup_loss": 0.4972333142555696
    },
    {
      "train_step": 4525,
      "sup_loss": 0.5932332178486447
    },
    {
      "train_step": 4526,
      "sup_loss": 0.3443990333307067
    },
    {
      "train_step": 4527,
      "sup_loss": 0.4155325455219635
    },
    {
      "train_step": 4528,
      "sup_loss": 0.5835248003274826
    },
    {
      "train_step": 4529,
      "sup_loss": 0.7168004783438845
    },
    {
      "train_step": 4530,
      "sup_loss": 0.4576621729917546
    },
    {
      "train_step": 4531,
      "sup_loss": 0.747703344109412
    },
    {
      "train_step": 4532,
      "sup_loss": 0.4919693089137649
    },
    {
      "train_step": 4533,
      "sup_loss": 0.5550688328859089
    },
    {
      "train_step": 4534,
      "sup_loss": 0.6477973621742025
    },
    {
      "train_step": 4535,
      "sup_loss": 0.42971407362903163
    },
    {
      "train_step": 4536,
      "sup_loss": 0.5739253281641674
    },
    {
      "train_step": 4537,
      "sup_loss": 0.5647103714040618
    },
    {
      "train_step": 4538,
      "sup_loss": 0.49106208280087404
    },
    {
      "train_step": 4539,
      "sup_loss": 0.6738992542914757
    },
    {
      "train_step": 4540,
      "sup_loss": 0.44326953492689153
    },
    {
      "train_step": 4541,
      "sup_loss": 0.5402199418657898
    },
    {
      "train_step": 4542,
      "sup_loss": 0.40443670528883024
    },
    {
      "train_step": 4543,
      "sup_loss": 0.5075171971900587
    },
    {
      "train_step": 4544,
      "sup_loss": 0.6217684846900936
    },
    {
      "train_step": 4545,
      "sup_loss": 0.3183481416188661
    },
    {
      "train_step": 4546,
      "sup_loss": 0.443336355575417
    },
    {
      "train_step": 4547,
      "sup_loss": 0.5227673241712317
    },
    {
      "train_step": 4548,
      "sup_loss": 0.7017891840603749
    },
    {
      "train_step": 4549,
      "sup_loss": 0.5779921153858845
    },
    {
      "train_step": 4550,
      "sup_loss": 0.4507244761435352
    },
    {
      "train_step": 4551,
      "sup_loss": 0.6483942786741772
    },
    {
      "train_step": 4552,
      "sup_loss": 0.6032618681565518
    },
    {
      "train_step": 4553,
      "sup_loss": 0.40810316723797097
    },
    {
      "train_step": 4554,
      "sup_loss": 0.6160672038094823
    },
    {
      "train_step": 4555,
      "sup_loss": 0.5752034257805791
    },
    {
      "train_step": 4556,
      "sup_loss": 0.48336889473564026
    },
    {
      "train_step": 4557,
      "sup_loss": 0.7467961918882589
    },
    {
      "train_step": 4558,
      "sup_loss": 0.5546092151392467
    },
    {
      "train_step": 4559,
      "sup_loss": 0.5558642566276791
    },
    {
      "train_step": 4560,
      "sup_loss": 0.6952224843402393
    },
    {
      "train_step": 4561,
      "sup_loss": 0.6196507367847941
    },
    {
      "train_step": 4562,
      "sup_loss": 0.5627496620860801
    },
    {
      "train_step": 4563,
      "sup_loss": 0.7169840296483333
    },
    {
      "train_step": 4564,
      "sup_loss": 0.5867057062679254
    },
    {
      "train_step": 4565,
      "sup_loss": 0.5764611437160317
    },
    {
      "train_step": 4566,
      "sup_loss": 0.19474359059336854
    },
    {
      "train_step": 4567,
      "sup_loss": 0.4708634452310851
    },
    {
      "train_step": 4568,
      "sup_loss": 0.7188334549502085
    },
    {
      "train_step": 4569,
      "sup_loss": 0.30415086073613146
    },
    {
      "train_step": 4570,
      "sup_loss": 0.5390715422532992
    },
    {
      "train_step": 4571,
      "sup_loss": 0.7523442372242417
    },
    {
      "train_step": 4572,
      "sup_loss": 0.4540495587639851
    },
    {
      "train_step": 4573,
      "sup_loss": 0.6277397918663781
    },
    {
      "train_step": 4574,
      "sup_loss": 0.47370539366245223
    },
    {
      "train_step": 4575,
      "sup_loss": 0.6748438220254173
    },
    {
      "train_step": 4576,
      "sup_loss": 0.6541526983807611
    },
    {
      "train_step": 4577,
      "sup_loss": 0.5581979143618142
    },
    {
      "train_step": 4578,
      "sup_loss": 0.712616028258018
    },
    {
      "train_step": 4579,
      "sup_loss": 0.6061017292663028
    },
    {
      "train_step": 4580,
      "sup_loss": 0.6036248085171092
    },
    {
      "train_step": 4581,
      "sup_loss": 0.47476554693811557
    },
    {
      "train_step": 4582,
      "sup_loss": 0.5037506607530398
    },
    {
      "train_step": 4583,
      "sup_loss": 0.5127206464900811
    },
    {
      "train_step": 4584,
      "sup_loss": 0.6628096203504601
    },
    {
      "train_step": 4585,
      "sup_loss": 0.7683901588282902
    },
    {
      "train_step": 4586,
      "sup_loss": 0.7459600917715082
    },
    {
      "train_step": 4587,
      "sup_loss": 0.6085020575574768
    },
    {
      "train_step": 4588,
      "sup_loss": 0.7048214299550585
    },
    {
      "train_step": 4589,
      "sup_loss": 0.6552619614951645
    },
    {
      "train_step": 4590,
      "sup_loss": 0.5669628352653686
    },
    {
      "train_step": 4591,
      "sup_loss": 0.6086166949526864
    },
    {
      "train_step": 4592,
      "sup_loss": 0.7098806409435612
    },
    {
      "train_step": 4593,
      "sup_loss": 0.7323981419229908
    },
    {
      "train_step": 4594,
      "sup_loss": 0.5432206735732401
    },
    {
      "train_step": 4595,
      "sup_loss": 0.6334024315460207
    },
    {
      "train_step": 4596,
      "sup_loss": 0.6360026912074649
    },
    {
      "train_step": 4597,
      "sup_loss": 0.5957961119073977
    },
    {
      "train_step": 4598,
      "sup_loss": 0.6114994565004164
    },
    {
      "train_step": 4599,
      "sup_loss": 0.41331227190314795
    },
    {
      "train_step": 4600,
      "sup_loss": 0.5034804568506474
    },
    {
      "train_step": 4601,
      "sup_loss": 0.8113884745155352
    },
    {
      "train_step": 4602,
      "sup_loss": 0.5841459102468077
    },
    {
      "train_step": 4603,
      "sup_loss": 0.48229353288348825
    },
    {
      "train_step": 4604,
      "sup_loss": 0.7092933863067679
    },
    {
      "train_step": 4605,
      "sup_loss": 0.6031672773671123
    },
    {
      "train_step": 4606,
      "sup_loss": 0.7173226666010468
    },
    {
      "train_step": 4607,
      "sup_loss": 0.6062577743798901
    },
    {
      "train_step": 4608,
      "sup_loss": 0.8008568892628148
    },
    {
      "train_step": 4609,
      "sup_loss": 0.7267162451711028
    },
    {
      "train_step": 4610,
      "sup_loss": 0.7169216947882092
    },
    {
      "train_step": 4611,
      "sup_loss": 0.6703617077920675
    },
    {
      "train_step": 4612,
      "sup_loss": 0.7808805554236218
    },
    {
      "train_step": 4613,
      "sup_loss": 0.7188404500997311
    },
    {
      "train_step": 4614,
      "sup_loss": 0.6211931638757653
    },
    {
      "train_step": 4615,
      "sup_loss": 0.6633252865918318
    },
    {
      "train_step": 4616,
      "sup_loss": 0.4855885613306505
    },
    {
      "train_step": 4617,
      "sup_loss": 0.4946829693398747
    },
    {
      "train_step": 4618,
      "sup_loss": 0.46156411841539846
    },
    {
      "train_step": 4619,
      "sup_loss": 0.5198530996486949
    },
    {
      "train_step": 4620,
      "sup_loss": 0.5383149684829113
    },
    {
      "train_step": 4621,
      "sup_loss": 0.3908719481248151
    },
    {
      "train_step": 4622,
      "sup_loss": 0.25991686604395575
    },
    {
      "train_step": 4623,
      "sup_loss": 0.673592598941774
    },
    {
      "train_step": 4624,
      "sup_loss": 0.72171204243645
    },
    {
      "train_step": 4625,
      "sup_loss": 0.4998412379199468
    },
    {
      "train_step": 4626,
      "sup_loss": 0.5738230210903915
    },
    {
      "train_step": 4627,
      "sup_loss": 0.6664610516620532
    },
    {
      "train_step": 4628,
      "sup_loss": 0.7954253320174615
    },
    {
      "train_step": 4629,
      "sup_loss": 0.5467251882989499
    },
    {
      "train_step": 4630,
      "sup_loss": 0.7913193200186299
    },
    {
      "train_step": 4631,
      "sup_loss": 0.49624110048599657
    },
    {
      "train_step": 4632,
      "sup_loss": 0.5298136135445104
    },
    {
      "train_step": 4633,
      "sup_loss": 0.6574240929931794
    },
    {
      "train_step": 4634,
      "sup_loss": 0.6209152962994746
    },
    {
      "train_step": 4635,
      "sup_loss": 0.5419707384325587
    },
    {
      "train_step": 4636,
      "sup_loss": 0.5953045342186852
    },
    {
      "train_step": 4637,
      "sup_loss": 0.5554191975659664
    },
    {
      "train_step": 4638,
      "sup_loss": 0.6597897454026109
    },
    {
      "train_step": 4639,
      "sup_loss": 0.5424145320818875
    },
    {
      "train_step": 4640,
      "sup_loss": 0.5834988659233128
    },
    {
      "train_step": 4641,
      "sup_loss": 0.5601923110600886
    },
    {
      "train_step": 4642,
      "sup_loss": 0.5036124321420001
    },
    {
      "train_step": 4643,
      "sup_loss": 0.7086088236873994
    },
    {
      "train_step": 4644,
      "sup_loss": 0.8360839899940076
    },
    {
      "train_step": 4645,
      "sup_loss": 0.7384659636564469
    },
    {
      "train_step": 4646,
      "sup_loss": 0.5173622129052394
    },
    {
      "train_step": 4647,
      "sup_loss": 0.5599607585498659
    },
    {
      "train_step": 4648,
      "sup_loss": 0.6639760461419003
    },
    {
      "train_step": 4649,
      "sup_loss": 0.588583231490555
    },
    {
      "train_step": 4650,
      "sup_loss": 0.7256871522109084
    },
    {
      "train_step": 4651,
      "sup_loss": 0.6499518936709949
    },
    {
      "train_step": 4652,
      "sup_loss": 0.6169496320968025
    },
    {
      "train_step": 4653,
      "sup_loss": 0.5067959372034183
    },
    {
      "train_step": 4654,
      "sup_loss": 0.524590814350213
    },
    {
      "train_step": 4655,
      "sup_loss": 0.4458591674245024
    },
    {
      "train_step": 4656,
      "sup_loss": 0.6232210847281747
    },
    {
      "train_step": 4657,
      "sup_loss": 0.7857100226701997
    },
    {
      "train_step": 4658,
      "sup_loss": 0.7728970740989171
    },
    {
      "train_step": 4659,
      "sup_loss": 0.6196672810375422
    },
    {
      "train_step": 4660,
      "sup_loss": 0.6689159693423087
    },
    {
      "train_step": 4661,
      "sup_loss": 0.656094676369403
    },
    {
      "train_step": 4662,
      "sup_loss": 0.40573422744260984
    },
    {
      "train_step": 4663,
      "sup_loss": 0.3618916786707808
    },
    {
      "train_step": 4664,
      "sup_loss": 0.4096536071594558
    },
    {
      "train_step": 4665,
      "sup_loss": 0.6377894340658837
    },
    {
      "train_step": 4666,
      "sup_loss": 0.6678565442876706
    },
    {
      "train_step": 4667,
      "sup_loss": 0.3862166837868758
    },
    {
      "train_step": 4668,
      "sup_loss": 0.7113571433322975
    },
    {
      "train_step": 4669,
      "sup_loss": 0.4815998480266612
    },
    {
      "train_step": 4670,
      "sup_loss": 0.6401821802715653
    },
    {
      "train_step": 4671,
      "sup_loss": 0.5844480789490082
    },
    {
      "train_step": 4672,
      "sup_loss": 0.6134140761933862
    },
    {
      "train_step": 4673,
      "sup_loss": 0.43463099311113906
    },
    {
      "train_step": 4674,
      "sup_loss": 0.6994569264763364
    },
    {
      "train_step": 4675,
      "sup_loss": 0.48077007034679015
    },
    {
      "train_step": 4676,
      "sup_loss": 0.5220209543040423
    },
    {
      "train_step": 4677,
      "sup_loss": 0.5391382109562
    },
    {
      "train_step": 4678,
      "sup_loss": 0.4328138796687799
    },
    {
      "train_step": 4679,
      "sup_loss": 0.5886020489006911
    },
    {
      "train_step": 4680,
      "sup_loss": 0.6254410500504806
    },
    {
      "train_step": 4681,
      "sup_loss": 0.7675169511519515
    },
    {
      "train_step": 4682,
      "sup_loss": 0.5143578717313362
    },
    {
      "train_step": 4683,
      "sup_loss": 0.7078001998031676
    },
    {
      "train_step": 4684,
      "sup_loss": 0.6826333869492425
    },
    {
      "train_step": 4685,
      "sup_loss": 0.8838377514213531
    },
    {
      "train_step": 4686,
      "sup_loss": 0.6377849101776227
    },
    {
      "train_step": 4687,
      "sup_loss": 0.660457122376116
    },
    {
      "train_step": 4688,
      "sup_loss": 0.7440374355916638
    },
    {
      "train_step": 4689,
      "sup_loss": 0.6768074261593169
    },
    {
      "train_step": 4690,
      "sup_loss": 0.5167852242822412
    },
    {
      "train_step": 4691,
      "sup_loss": 0.5400047365925339
    },
    {
      "train_step": 4692,
      "sup_loss": 0.8101319412543417
    },
    {
      "train_step": 4693,
      "sup_loss": 0.6513300153398875
    },
    {
      "train_step": 4694,
      "sup_loss": 0.53998840026403
    },
    {
      "train_step": 4695,
      "sup_loss": 0.7567675044861315
    },
    {
      "train_step": 4696,
      "sup_loss": 0.5513592778185195
    },
    {
      "train_step": 4697,
      "sup_loss": 0.5766294814118783
    },
    {
      "train_step": 4698,
      "sup_loss": 0.729999796462824
    },
    {
      "train_step": 4699,
      "sup_loss": 0.655707073939634
    },
    {
      "train_step": 4700,
      "sup_loss": 0.5410398227042312
    },
    {
      "train_step": 4701,
      "sup_loss": 0.3588592560646339
    },
    {
      "train_step": 4702,
      "sup_loss": 0.7323481341850415
    },
    {
      "train_step": 4703,
      "sup_loss": 0.4155107171432505
    },
    {
      "train_step": 4704,
      "sup_loss": 0.6537496363941653
    },
    {
      "train_step": 4705,
      "sup_loss": 0.6240027771555617
    },
    {
      "train_step": 4706,
      "sup_loss": 0.47824435510814867
    },
    {
      "train_step": 4707,
      "sup_loss": 0.3974752460486183
    },
    {
      "train_step": 4708,
      "sup_loss": 0.4644208140663279
    },
    {
      "train_step": 4709,
      "sup_loss": 0.43516152448298595
    },
    {
      "train_step": 4710,
      "sup_loss": 0.659692942986645
    },
    {
      "train_step": 4711,
      "sup_loss": 0.4066320315794655
    },
    {
      "train_step": 4712,
      "sup_loss": 0.4916447992813487
    },
    {
      "train_step": 4713,
      "sup_loss": 0.30111593216460536
    },
    {
      "train_step": 4714,
      "sup_loss": 0.35551861499425974
    },
    {
      "train_step": 4715,
      "sup_loss": 0.4671029036549823
    },
    {
      "train_step": 4716,
      "sup_loss": 0.5812072502308215
    },
    {
      "train_step": 4717,
      "sup_loss": 0.2570094400805053
    },
    {
      "train_step": 4718,
      "sup_loss": 0.8164558534548583
    },
    {
      "train_step": 4719,
      "sup_loss": 0.637228568072697
    },
    {
      "train_step": 4720,
      "sup_loss": 0.6496546439513605
    },
    {
      "train_step": 4721,
      "sup_loss": 0.4002491220905038
    },
    {
      "train_step": 4722,
      "sup_loss": 0.5643394510857801
    },
    {
      "train_step": 4723,
      "sup_loss": 0.43187457378754773
    },
    {
      "train_step": 4724,
      "sup_loss": 0.5934885714333069
    },
    {
      "train_step": 4725,
      "sup_loss": 0.3895713715988491
    },
    {
      "train_step": 4726,
      "sup_loss": 0.48327809111740694
    },
    {
      "train_step": 4727,
      "sup_loss": 0.32170403559311644
    },
    {
      "train_step": 4728,
      "sup_loss": 0.40533288799098277
    },
    {
      "train_step": 4729,
      "sup_loss": 0.5422683834486244
    },
    {
      "train_step": 4730,
      "sup_loss": 0.4653691213811729
    },
    {
      "train_step": 4731,
      "sup_loss": 0.7587738055230253
    },
    {
      "train_step": 4732,
      "sup_loss": 0.5717171015161147
    },
    {
      "train_step": 4733,
      "sup_loss": 0.5929266511825845
    },
    {
      "train_step": 4734,
      "sup_loss": 0.42857890448569946
    },
    {
      "train_step": 4735,
      "sup_loss": 0.6400351045059072
    },
    {
      "train_step": 4736,
      "sup_loss": 0.5618010135899199
    },
    {
      "train_step": 4737,
      "sup_loss": 0.50573583952006
    },
    {
      "train_step": 4738,
      "sup_loss": 0.5699600835510884
    },
    {
      "train_step": 4739,
      "sup_loss": 0.7257426813790312
    },
    {
      "train_step": 4740,
      "sup_loss": 0.5801768801822988
    },
    {
      "train_step": 4741,
      "sup_loss": 0.7293895220057275
    },
    {
      "train_step": 4742,
      "sup_loss": 0.7405845699792298
    },
    {
      "train_step": 4743,
      "sup_loss": 0.8986671735750392
    },
    {
      "train_step": 4744,
      "sup_loss": 1.0329967000846143
    },
    {
      "train_step": 4745,
      "sup_loss": 0.41051958833397645
    },
    {
      "train_step": 4746,
      "sup_loss": 0.6678543928398113
    },
    {
      "train_step": 4747,
      "sup_loss": 0.6605560256863976
    },
    {
      "train_step": 4748,
      "sup_loss": 0.8479190735592442
    },
    {
      "train_step": 4749,
      "sup_loss": 0.5656907234978362
    },
    {
      "train_step": 4750,
      "sup_loss": 0.5475215440713999
    },
    {
      "train_step": 4751,
      "sup_loss": 0.623911391030647
    },
    {
      "train_step": 4752,
      "sup_loss": 0.4324353369324858
    },
    {
      "train_step": 4753,
      "sup_loss": 0.876207425889081
    },
    {
      "train_step": 4754,
      "sup_loss": 0.6648243161036023
    },
    {
      "train_step": 4755,
      "sup_loss": 0.577695566028366
    },
    {
      "train_step": 4756,
      "sup_loss": 0.6304289245795724
    },
    {
      "train_step": 4757,
      "sup_loss": 0.6421825146288903
    },
    {
      "train_step": 4758,
      "sup_loss": 0.6386300614363384
    },
    {
      "train_step": 4759,
      "sup_loss": 0.48608256036706665
    },
    {
      "train_step": 4760,
      "sup_loss": 0.771098235128847
    },
    {
      "train_step": 4761,
      "sup_loss": 0.6278597018101607
    },
    {
      "train_step": 4762,
      "sup_loss": 0.6477430995062892
    },
    {
      "train_step": 4763,
      "sup_loss": 0.7540685715545944
    },
    {
      "train_step": 4764,
      "sup_loss": 0.6546913842568471
    },
    {
      "train_step": 4765,
      "sup_loss": 0.49906948497103615
    },
    {
      "train_step": 4766,
      "sup_loss": 0.6604770861060466
    },
    {
      "train_step": 4767,
      "sup_loss": 0.5493858478353238
    },
    {
      "train_step": 4768,
      "sup_loss": 0.6536240896354418
    },
    {
      "train_step": 4769,
      "sup_loss": 0.6164788949013913
    },
    {
      "train_step": 4770,
      "sup_loss": 0.6172347788534629
    },
    {
      "train_step": 4771,
      "sup_loss": 0.6489234728432568
    },
    {
      "train_step": 4772,
      "sup_loss": 0.3541790642656875
    },
    {
      "train_step": 4773,
      "sup_loss": 0.591129794661068
    },
    {
      "train_step": 4774,
      "sup_loss": 0.7600620281972251
    },
    {
      "train_step": 4775,
      "sup_loss": 0.49631689435952286
    },
    {
      "train_step": 4776,
      "sup_loss": 0.6733799993066092
    },
    {
      "train_step": 4777,
      "sup_loss": 0.47579813512199287
    },
    {
      "train_step": 4778,
      "sup_loss": 0.5523620734738828
    },
    {
      "train_step": 4779,
      "sup_loss": 0.7911320839725132
    },
    {
      "train_step": 4780,
      "sup_loss": 0.5785635632640644
    },
    {
      "train_step": 4781,
      "sup_loss": 0.6534983804138195
    },
    {
      "train_step": 4782,
      "sup_loss": 0.7884477525575453
    },
    {
      "train_step": 4783,
      "sup_loss": 0.6118175158345963
    },
    {
      "train_step": 4784,
      "sup_loss": 0.7145003435349183
    },
    {
      "train_step": 4785,
      "sup_loss": 0.6347612374409626
    },
    {
      "train_step": 4786,
      "sup_loss": 0.5838303689683197
    },
    {
      "train_step": 4787,
      "sup_loss": 0.4762259303115375
    },
    {
      "train_step": 4788,
      "sup_loss": 0.7550064704927857
    },
    {
      "train_step": 4789,
      "sup_loss": 0.3270247506956226
    },
    {
      "train_step": 4790,
      "sup_loss": 0.46159021659814087
    },
    {
      "train_step": 4791,
      "sup_loss": 0.5905616100725712
    },
    {
      "train_step": 4792,
      "sup_loss": 0.5378780066730842
    },
    {
      "train_step": 4793,
      "sup_loss": 0.3702697457553981
    },
    {
      "train_step": 4794,
      "sup_loss": 0.5354717724460982
    },
    {
      "train_step": 4795,
      "sup_loss": 0.5132624873838662
    },
    {
      "train_step": 4796,
      "sup_loss": 0.591140209814827
    },
    {
      "train_step": 4797,
      "sup_loss": 0.6227940600149827
    },
    {
      "train_step": 4798,
      "sup_loss": 0.5626756674323998
    },
    {
      "train_step": 4799,
      "sup_loss": 0.6203705744278717
    },
    {
      "train_step": 4800,
      "sup_loss": 0.4404351854638217
    },
    {
      "train_step": 4801,
      "sup_loss": 0.5180164641931184
    },
    {
      "train_step": 4802,
      "sup_loss": 0.5285014727273291
    },
    {
      "train_step": 4803,
      "sup_loss": 0.7341554239328297
    },
    {
      "train_step": 4804,
      "sup_loss": 0.6715334699170747
    },
    {
      "train_step": 4805,
      "sup_loss": 0.7288789085247778
    },
    {
      "train_step": 4806,
      "sup_loss": 0.6268554721127158
    },
    {
      "train_step": 4807,
      "sup_loss": 0.40667053424099653
    },
    {
      "train_step": 4808,
      "sup_loss": 0.668306727910227
    },
    {
      "train_step": 4809,
      "sup_loss": 0.5941400829623289
    },
    {
      "train_step": 4810,
      "sup_loss": 0.7757445948114656
    },
    {
      "train_step": 4811,
      "sup_loss": 0.5680853832412082
    },
    {
      "train_step": 4812,
      "sup_loss": 0.6324718842411586
    },
    {
      "train_step": 4813,
      "sup_loss": 0.5485032157522169
    },
    {
      "train_step": 4814,
      "sup_loss": 0.44225394827601067
    },
    {
      "train_step": 4815,
      "sup_loss": 0.4977659632072386
    },
    {
      "train_step": 4816,
      "sup_loss": 0.3640058491898527
    },
    {
      "train_step": 4817,
      "sup_loss": 0.6252312110429955
    },
    {
      "train_step": 4818,
      "sup_loss": 0.31987633605313687
    },
    {
      "train_step": 4819,
      "sup_loss": 0.5312147712426417
    },
    {
      "train_step": 4820,
      "sup_loss": 0.5621649438715202
    },
    {
      "train_step": 4821,
      "sup_loss": 0.7547048583453301
    },
    {
      "train_step": 4822,
      "sup_loss": 0.5042116768016112
    },
    {
      "train_step": 4823,
      "sup_loss": 0.6855397477872266
    },
    {
      "train_step": 4824,
      "sup_loss": 0.4929596588139971
    },
    {
      "train_step": 4825,
      "sup_loss": 0.5031257187433709
    },
    {
      "train_step": 4826,
      "sup_loss": 0.5711096166926641
    },
    {
      "train_step": 4827,
      "sup_loss": 0.6907643870913698
    },
    {
      "train_step": 4828,
      "sup_loss": 0.4431255937661927
    },
    {
      "train_step": 4829,
      "sup_loss": 0.644858008744396
    },
    {
      "train_step": 4830,
      "sup_loss": 0.7052802829433381
    },
    {
      "train_step": 4831,
      "sup_loss": 0.5135953977004822
    },
    {
      "train_step": 4832,
      "sup_loss": 0.14003253935344379
    },
    {
      "train_step": 4833,
      "sup_loss": 0.5587462981168207
    },
    {
      "train_step": 4834,
      "sup_loss": 0.47418000303523156
    },
    {
      "train_step": 4835,
      "sup_loss": 0.37814556927311044
    },
    {
      "train_step": 4836,
      "sup_loss": 0.3460605842972018
    },
    {
      "train_step": 4837,
      "sup_loss": 0.368343549857194
    },
    {
      "train_step": 4838,
      "sup_loss": 0.5520102558954296
    },
    {
      "train_step": 4839,
      "sup_loss": 0.6359456917207522
    },
    {
      "train_step": 4840,
      "sup_loss": 0.5537809187585299
    },
    {
      "train_step": 4841,
      "sup_loss": 0.7605109765300023
    },
    {
      "train_step": 4842,
      "sup_loss": 0.46066190558305986
    },
    {
      "train_step": 4843,
      "sup_loss": 0.5055703322109752
    },
    {
      "train_step": 4844,
      "sup_loss": 0.6608196462788186
    },
    {
      "train_step": 4845,
      "sup_loss": 0.6560572981031394
    },
    {
      "train_step": 4846,
      "sup_loss": 0.6724432964553806
    },
    {
      "train_step": 4847,
      "sup_loss": 0.5394069438585645
    },
    {
      "train_step": 4848,
      "sup_loss": 0.5356972765643336
    },
    {
      "train_step": 4849,
      "sup_loss": 0.4613020310245883
    },
    {
      "train_step": 4850,
      "sup_loss": 0.456630389054091
    },
    {
      "train_step": 4851,
      "sup_loss": 0.6979748619016933
    },
    {
      "train_step": 4852,
      "sup_loss": 0.812597860021073
    },
    {
      "train_step": 4853,
      "sup_loss": 0.5058190842605161
    },
    {
      "train_step": 4854,
      "sup_loss": 0.5748040832691742
    },
    {
      "train_step": 4855,
      "sup_loss": 0.5297652993950878
    },
    {
      "train_step": 4856,
      "sup_loss": 0.3811434268203523
    },
    {
      "train_step": 4857,
      "sup_loss": 0.671912187028153
    },
    {
      "train_step": 4858,
      "sup_loss": 0.4844606318765083
    },
    {
      "train_step": 4859,
      "sup_loss": 0.5892964553112944
    },
    {
      "train_step": 4860,
      "sup_loss": 0.6604827945373741
    },
    {
      "train_step": 4861,
      "sup_loss": 0.5698910629886301
    },
    {
      "train_step": 4862,
      "sup_loss": 0.6090962862000407
    },
    {
      "train_step": 4863,
      "sup_loss": 0.41765066562599373
    },
    {
      "train_step": 4864,
      "sup_loss": 0.5328440904938136
    },
    {
      "train_step": 4865,
      "sup_loss": 0.5534829450041622
    },
    {
      "train_step": 4866,
      "sup_loss": 0.9144587093419313
    },
    {
      "train_step": 4867,
      "sup_loss": 0.39600278507428627
    },
    {
      "train_step": 4868,
      "sup_loss": 0.39663396605929735
    },
    {
      "train_step": 4869,
      "sup_loss": 0.4958254055528653
    },
    {
      "train_step": 4870,
      "sup_loss": 0.6890347256570183
    },
    {
      "train_step": 4871,
      "sup_loss": 0.22387446464203448
    },
    {
      "train_step": 4872,
      "sup_loss": 0.7126257437122483
    },
    {
      "train_step": 4873,
      "sup_loss": 0.5837748612603779
    },
    {
      "train_step": 4874,
      "sup_loss": 0.5712066654539585
    },
    {
      "train_step": 4875,
      "sup_loss": 0.7313289610948854
    },
    {
      "train_step": 4876,
      "sup_loss": 0.6325889968984256
    },
    {
      "train_step": 4877,
      "sup_loss": 0.3742650284891792
    },
    {
      "train_step": 4878,
      "sup_loss": 0.5568165433702519
    },
    {
      "train_step": 4879,
      "sup_loss": 0.42732121087115327
    },
    {
      "train_step": 4880,
      "sup_loss": 0.6612110736296912
    },
    {
      "train_step": 4881,
      "sup_loss": 0.6610726298440808
    },
    {
      "train_step": 4882,
      "sup_loss": 0.6431005121082607
    },
    {
      "train_step": 4883,
      "sup_loss": 0.740474410507684
    },
    {
      "train_step": 4884,
      "sup_loss": 0.5083623035970135
    },
    {
      "train_step": 4885,
      "sup_loss": 0.6516251528628619
    },
    {
      "train_step": 4886,
      "sup_loss": 0.5438638882422518
    },
    {
      "train_step": 4887,
      "sup_loss": 0.3517764239594805
    },
    {
      "train_step": 4888,
      "sup_loss": 0.787509020832659
    },
    {
      "train_step": 4889,
      "sup_loss": 0.8159949614345328
    },
    {
      "train_step": 4890,
      "sup_loss": 0.7873921589182655
    },
    {
      "train_step": 4891,
      "sup_loss": 0.5199156419935399
    },
    {
      "train_step": 4892,
      "sup_loss": 0.5565676763498206
    },
    {
      "train_step": 4893,
      "sup_loss": 0.7253254966998155
    },
    {
      "train_step": 4894,
      "sup_loss": 0.5701015215425693
    },
    {
      "train_step": 4895,
      "sup_loss": 0.6633783175584327
    },
    {
      "train_step": 4896,
      "sup_loss": 0.5176018801958601
    },
    {
      "train_step": 4897,
      "sup_loss": 0.5196673126585872
    },
    {
      "train_step": 4898,
      "sup_loss": 0.5929586928788652
    },
    {
      "train_step": 4899,
      "sup_loss": 0.5632154048378206
    },
    {
      "train_step": 4900,
      "sup_loss": 0.6456851889460486
    },
    {
      "train_step": 4901,
      "sup_loss": 0.6997896362711329
    },
    {
      "train_step": 4902,
      "sup_loss": 0.5702517862865732
    },
    {
      "train_step": 4903,
      "sup_loss": 0.5218581393217046
    },
    {
      "train_step": 4904,
      "sup_loss": 0.3773737759370157
    },
    {
      "train_step": 4905,
      "sup_loss": 0.5243804279577395
    },
    {
      "train_step": 4906,
      "sup_loss": 0.6481179403223329
    },
    {
      "train_step": 4907,
      "sup_loss": 0.6608692809289191
    },
    {
      "train_step": 4908,
      "sup_loss": 0.2640124058907753
    },
    {
      "train_step": 4909,
      "sup_loss": 0.6713350208656632
    },
    {
      "train_step": 4910,
      "sup_loss": 0.3813869936685184
    },
    {
      "train_step": 4911,
      "sup_loss": 0.7025693322479235
    },
    {
      "train_step": 4912,
      "sup_loss": 0.5528249770362779
    },
    {
      "train_step": 4913,
      "sup_loss": 0.7368569025856341
    },
    {
      "train_step": 4914,
      "sup_loss": 0.6856937073272843
    },
    {
      "train_step": 4915,
      "sup_loss": 0.43461250541622326
    },
    {
      "train_step": 4916,
      "sup_loss": 0.5708041197341981
    },
    {
      "train_step": 4917,
      "sup_loss": 0.4078470046076376
    },
    {
      "train_step": 4918,
      "sup_loss": 0.5340125566388072
    },
    {
      "train_step": 4919,
      "sup_loss": 0.5020849937841628
    },
    {
      "train_step": 4920,
      "sup_loss": 0.6778333444516386
    },
    {
      "train_step": 4921,
      "sup_loss": 0.6738417482741095
    },
    {
      "train_step": 4922,
      "sup_loss": 0.601561237285558
    },
    {
      "train_step": 4923,
      "sup_loss": 0.4988249347663138
    },
    {
      "train_step": 4924,
      "sup_loss": 0.4502242760768478
    },
    {
      "train_step": 4925,
      "sup_loss": 0.2860670592865612
    },
    {
      "train_step": 4926,
      "sup_loss": 0.5355883378457899
    },
    {
      "train_step": 4927,
      "sup_loss": 0.5832692825090935
    },
    {
      "train_step": 4928,
      "sup_loss": 0.512536592742037
    },
    {
      "train_step": 4929,
      "sup_loss": 0.6335656768329755
    },
    {
      "train_step": 4930,
      "sup_loss": 0.6724375686303005
    },
    {
      "train_step": 4931,
      "sup_loss": 0.5660928717941992
    },
    {
      "train_step": 4932,
      "sup_loss": 0.4330090171466526
    },
    {
      "train_step": 4933,
      "sup_loss": 0.6040795002510869
    },
    {
      "train_step": 4934,
      "sup_loss": 0.6141185812160946
    },
    {
      "train_step": 4935,
      "sup_loss": 0.6407469325848026
    },
    {
      "train_step": 4936,
      "sup_loss": 0.6001878262311533
    },
    {
      "train_step": 4937,
      "sup_loss": 0.5852434059644651
    },
    {
      "train_step": 4938,
      "sup_loss": 0.4414902772217767
    },
    {
      "train_step": 4939,
      "sup_loss": 0.6412567838825549
    },
    {
      "train_step": 4940,
      "sup_loss": 0.46209207651842765
    },
    {
      "train_step": 4941,
      "sup_loss": 0.6596957391596371
    },
    {
      "train_step": 4942,
      "sup_loss": 0.7267925950363909
    },
    {
      "train_step": 4943,
      "sup_loss": 0.5212869345685834
    },
    {
      "train_step": 4944,
      "sup_loss": 0.5822803119265003
    },
    {
      "train_step": 4945,
      "sup_loss": 0.695209231828373
    },
    {
      "train_step": 4946,
      "sup_loss": 0.6832644444236973
    },
    {
      "train_step": 4947,
      "sup_loss": 0.6282428085100695
    },
    {
      "train_step": 4948,
      "sup_loss": 0.6229191382755759
    },
    {
      "train_step": 4949,
      "sup_loss": 0.470448930640974
    },
    {
      "train_step": 4950,
      "sup_loss": 0.6605238055429432
    },
    {
      "train_step": 4951,
      "sup_loss": 0.8678242332618878
    },
    {
      "train_step": 4952,
      "sup_loss": 0.7040210475469381
    },
    {
      "train_step": 4953,
      "sup_loss": 0.4130704217044159
    },
    {
      "train_step": 4954,
      "sup_loss": 0.5619553882258242
    },
    {
      "train_step": 4955,
      "sup_loss": 0.616855922580221
    },
    {
      "train_step": 4956,
      "sup_loss": 0.5928692413624579
    },
    {
      "train_step": 4957,
      "sup_loss": 0.684457099337536
    },
    {
      "train_step": 4958,
      "sup_loss": 0.5861028493028199
    },
    {
      "train_step": 4959,
      "sup_loss": 0.5246561173354194
    },
    {
      "train_step": 4960,
      "sup_loss": 0.7630032133592013
    },
    {
      "train_step": 4961,
      "sup_loss": 0.39520948557576924
    },
    {
      "train_step": 4962,
      "sup_loss": 0.6389940778913944
    },
    {
      "train_step": 4963,
      "sup_loss": 0.85911691936446
    },
    {
      "train_step": 4964,
      "sup_loss": 0.6599179848222633
    },
    {
      "train_step": 4965,
      "sup_loss": 0.5872717687785898
    },
    {
      "train_step": 4966,
      "sup_loss": 0.6708698936978816
    },
    {
      "train_step": 4967,
      "sup_loss": 0.5322473996145024
    },
    {
      "train_step": 4968,
      "sup_loss": 0.4963038719000675
    },
    {
      "train_step": 4969,
      "sup_loss": 0.6044828800910543
    },
    {
      "train_step": 4970,
      "sup_loss": 0.39559551428656187
    },
    {
      "train_step": 4971,
      "sup_loss": 0.4125533026715166
    },
    {
      "train_step": 4972,
      "sup_loss": 0.3540910369979439
    },
    {
      "train_step": 4973,
      "sup_loss": 0.5715529571651755
    },
    {
      "train_step": 4974,
      "sup_loss": 0.4923395164018048
    },
    {
      "train_step": 4975,
      "sup_loss": 0.6299591128177952
    },
    {
      "train_step": 4976,
      "sup_loss": 0.46725827221286337
    },
    {
      "train_step": 4977,
      "sup_loss": 0.6258885403018433
    },
    {
      "train_step": 4978,
      "sup_loss": 0.6468255767433428
    },
    {
      "train_step": 4979,
      "sup_loss": 0.6813989895055508
    },
    {
      "train_step": 4980,
      "sup_loss": 0.5980610548281393
    },
    {
      "train_step": 4981,
      "sup_loss": 0.5242005860726537
    },
    {
      "train_step": 4982,
      "sup_loss": 0.8176974964780452
    },
    {
      "train_step": 4983,
      "sup_loss": 0.6056309923545562
    },
    {
      "train_step": 4984,
      "sup_loss": 0.4189390243112042
    },
    {
      "train_step": 4985,
      "sup_loss": 0.7017485951541574
    },
    {
      "train_step": 4986,
      "sup_loss": 0.41297910391851517
    },
    {
      "train_step": 4987,
      "sup_loss": 0.6706816343094841
    },
    {
      "train_step": 4988,
      "sup_loss": 0.6630610446326969
    },
    {
      "train_step": 4989,
      "sup_loss": 0.7449776996565453
    },
    {
      "train_step": 4990,
      "sup_loss": 0.6234924890880544
    },
    {
      "train_step": 4991,
      "sup_loss": 0.22767542712791328
    },
    {
      "train_step": 4992,
      "sup_loss": 0.5957265111389303
    },
    {
      "train_step": 4993,
      "sup_loss": 0.5989464958262789
    },
    {
      "train_step": 4994,
      "sup_loss": 0.39631754661250784
    },
    {
      "train_step": 4995,
      "sup_loss": 0.4253714990928627
    },
    {
      "train_step": 4996,
      "sup_loss": 0.762796570468641
    },
    {
      "train_step": 4997,
      "sup_loss": 0.5626805677999814
    },
    {
      "train_step": 4998,
      "sup_loss": 0.7629846107703029
    },
    {
      "train_step": 4999,
      "sup_loss": 0.5882478124954535
    },
    {
      "train_step": 5000,
      "sup_loss": 0.3544659660186096
    },
    {
      "train_step": 5001,
      "sup_loss": 0.5613737611514108
    },
    {
      "train_step": 5002,
      "sup_loss": 0.6436963530004968
    },
    {
      "train_step": 5003,
      "sup_loss": 0.7205929722712819
    },
    {
      "train_step": 5004,
      "sup_loss": 0.6411002582808276
    },
    {
      "train_step": 5005,
      "sup_loss": 0.521605765073385
    },
    {
      "train_step": 5006,
      "sup_loss": 0.7010091715285505
    },
    {
      "train_step": 5007,
      "sup_loss": 0.7153773302292342
    },
    {
      "train_step": 5008,
      "sup_loss": 0.45377457567905405
    },
    {
      "train_step": 5009,
      "sup_loss": 0.6487628257669489
    },
    {
      "train_step": 5010,
      "sup_loss": 0.6097753751128226
    },
    {
      "train_step": 5011,
      "sup_loss": 0.42485310729517395
    },
    {
      "train_step": 5012,
      "sup_loss": 0.5022314221888516
    },
    {
      "train_step": 5013,
      "sup_loss": 0.28327660497761203
    },
    {
      "train_step": 5014,
      "sup_loss": 0.5645672598548821
    },
    {
      "train_step": 5015,
      "sup_loss": 0.745697138779809
    },
    {
      "train_step": 5016,
      "sup_loss": 0.7879842420709552
    },
    {
      "train_step": 5017,
      "sup_loss": 0.45474758952681565
    },
    {
      "train_step": 5018,
      "sup_loss": 0.66034855292184
    },
    {
      "train_step": 5019,
      "sup_loss": 0.689250105132889
    },
    {
      "train_step": 5020,
      "sup_loss": 0.6672214963195112
    },
    {
      "train_step": 5021,
      "sup_loss": 0.4328131776005238
    },
    {
      "train_step": 5022,
      "sup_loss": 0.48962420072066787
    },
    {
      "train_step": 5023,
      "sup_loss": 0.7359705632885872
    },
    {
      "train_step": 5024,
      "sup_loss": 0.5181246420040282
    },
    {
      "train_step": 5025,
      "sup_loss": 0.8595020732794217
    },
    {
      "train_step": 5026,
      "sup_loss": 0.5437201735088982
    },
    {
      "train_step": 5027,
      "sup_loss": 0.5634723032509154
    },
    {
      "train_step": 5028,
      "sup_loss": 0.6687435750897939
    },
    {
      "train_step": 5029,
      "sup_loss": 0.6842103401974754
    },
    {
      "train_step": 5030,
      "sup_loss": 0.635394534082086
    },
    {
      "train_step": 5031,
      "sup_loss": 0.1945692213456766
    },
    {
      "train_step": 5032,
      "sup_loss": 0.6109287704231907
    },
    {
      "train_step": 5033,
      "sup_loss": 0.5031497347793309
    },
    {
      "train_step": 5034,
      "sup_loss": 0.6883506530739967
    },
    {
      "train_step": 5035,
      "sup_loss": 0.3524574533535607
    },
    {
      "train_step": 5036,
      "sup_loss": 0.681731217493975
    },
    {
      "train_step": 5037,
      "sup_loss": 0.42804733902250064
    },
    {
      "train_step": 5038,
      "sup_loss": 0.7239880681367336
    },
    {
      "train_step": 5039,
      "sup_loss": 0.6954973778598748
    },
    {
      "train_step": 5040,
      "sup_loss": 0.4559350649238881
    },
    {
      "train_step": 5041,
      "sup_loss": 0.4010137344123723
    },
    {
      "train_step": 5042,
      "sup_loss": 0.5530331802908957
    },
    {
      "train_step": 5043,
      "sup_loss": 0.5609668081004204
    },
    {
      "train_step": 5044,
      "sup_loss": 0.625140947158055
    },
    {
      "train_step": 5045,
      "sup_loss": 0.5987285813767281
    },
    {
      "train_step": 5046,
      "sup_loss": 0.6626207498120595
    },
    {
      "train_step": 5047,
      "sup_loss": 0.6408685037371981
    },
    {
      "train_step": 5048,
      "sup_loss": 0.583759866161012
    },
    {
      "train_step": 5049,
      "sup_loss": 0.9603524944612382
    },
    {
      "train_step": 5050,
      "sup_loss": 0.5852908539741953
    },
    {
      "train_step": 5051,
      "sup_loss": 0.7281008647202459
    },
    {
      "train_step": 5052,
      "sup_loss": 0.583763215333356
    },
    {
      "train_step": 5053,
      "sup_loss": 0.643666569158082
    },
    {
      "train_step": 5054,
      "sup_loss": 0.49638227394693896
    },
    {
      "train_step": 5055,
      "sup_loss": 0.6751790303972915
    },
    {
      "train_step": 5056,
      "sup_loss": 0.4447126508835269
    },
    {
      "train_step": 5057,
      "sup_loss": 0.7553304890513848
    },
    {
      "train_step": 5058,
      "sup_loss": 0.5560902730020607
    },
    {
      "train_step": 5059,
      "sup_loss": 0.5439206479891027
    },
    {
      "train_step": 5060,
      "sup_loss": 0.4906017158950764
    },
    {
      "train_step": 5061,
      "sup_loss": 0.7391798009679438
    },
    {
      "train_step": 5062,
      "sup_loss": 0.5711016241482968
    },
    {
      "train_step": 5063,
      "sup_loss": 0.5047082544093598
    },
    {
      "train_step": 5064,
      "sup_loss": 0.589755968938977
    },
    {
      "train_step": 5065,
      "sup_loss": 0.7420094733799995
    },
    {
      "train_step": 5066,
      "sup_loss": 0.6684673200998191
    },
    {
      "train_step": 5067,
      "sup_loss": 0.546548083053138
    },
    {
      "train_step": 5068,
      "sup_loss": 0.5340828055352416
    },
    {
      "train_step": 5069,
      "sup_loss": 0.7209092239548313
    },
    {
      "train_step": 5070,
      "sup_loss": 0.5376181081248511
    },
    {
      "train_step": 5071,
      "sup_loss": 0.4594819607150527
    },
    {
      "train_step": 5072,
      "sup_loss": 0.5888819612072341
    },
    {
      "train_step": 5073,
      "sup_loss": 0.5118028052972164
    },
    {
      "train_step": 5074,
      "sup_loss": 0.41218274702558855
    },
    {
      "train_step": 5075,
      "sup_loss": 0.5378734407744022
    },
    {
      "train_step": 5076,
      "sup_loss": 0.6193336342779725
    },
    {
      "train_step": 5077,
      "sup_loss": 0.47994667653700007
    },
    {
      "train_step": 5078,
      "sup_loss": 0.5852231431989747
    },
    {
      "train_step": 5079,
      "sup_loss": 0.4806462603900332
    },
    {
      "train_step": 5080,
      "sup_loss": 0.6411285409529622
    },
    {
      "train_step": 5081,
      "sup_loss": 0.5870614270334938
    },
    {
      "train_step": 5082,
      "sup_loss": 0.45816809137788306
    },
    {
      "train_step": 5083,
      "sup_loss": 0.5808166058163706
    },
    {
      "train_step": 5084,
      "sup_loss": 0.45899500857974995
    },
    {
      "train_step": 5085,
      "sup_loss": 0.5286513895174249
    },
    {
      "train_step": 5086,
      "sup_loss": 0.3890489107531
    },
    {
      "train_step": 5087,
      "sup_loss": 0.5278986040219354
    },
    {
      "train_step": 5088,
      "sup_loss": 0.7028382826287614
    },
    {
      "train_step": 5089,
      "sup_loss": 0.4272131648704039
    },
    {
      "train_step": 5090,
      "sup_loss": 0.5862224350890922
    },
    {
      "train_step": 5091,
      "sup_loss": 0.6745587636988668
    },
    {
      "train_step": 5092,
      "sup_loss": 0.6479823379971955
    },
    {
      "train_step": 5093,
      "sup_loss": 0.5188212084485594
    },
    {
      "train_step": 5094,
      "sup_loss": 0.2903672856348565
    },
    {
      "train_step": 5095,
      "sup_loss": 0.6911175288707941
    },
    {
      "train_step": 5096,
      "sup_loss": 0.5774270120245744
    },
    {
      "train_step": 5097,
      "sup_loss": 0.576087467450566
    },
    {
      "train_step": 5098,
      "sup_loss": 0.4885979061541573
    },
    {
      "train_step": 5099,
      "sup_loss": 0.4386021360254987
    },
    {
      "train_step": 5100,
      "sup_loss": 0.4212361821276087
    },
    {
      "train_step": 5101,
      "sup_loss": 0.4937091987590343
    },
    {
      "train_step": 5102,
      "sup_loss": 0.7542557035874304
    },
    {
      "train_step": 5103,
      "sup_loss": 0.5179707593553894
    },
    {
      "train_step": 5104,
      "sup_loss": 0.5707959372275457
    },
    {
      "train_step": 5105,
      "sup_loss": 0.6305482757441886
    },
    {
      "train_step": 5106,
      "sup_loss": 0.5687442990815986
    },
    {
      "train_step": 5107,
      "sup_loss": 0.7863559236846551
    },
    {
      "train_step": 5108,
      "sup_loss": 0.5905243812753865
    },
    {
      "train_step": 5109,
      "sup_loss": 0.6091803716340032
    },
    {
      "train_step": 5110,
      "sup_loss": 0.5717491821484383
    },
    {
      "train_step": 5111,
      "sup_loss": 0.7018076922891318
    },
    {
      "train_step": 5112,
      "sup_loss": 0.6132443926893498
    },
    {
      "train_step": 5113,
      "sup_loss": 0.46226471134984043
    },
    {
      "train_step": 5114,
      "sup_loss": 0.8295802210026936
    },
    {
      "train_step": 5115,
      "sup_loss": 0.4386479089049253
    },
    {
      "train_step": 5116,
      "sup_loss": 0.7368104281854848
    },
    {
      "train_step": 5117,
      "sup_loss": 0.7231299425879854
    },
    {
      "train_step": 5118,
      "sup_loss": 0.2639872138031314
    },
    {
      "train_step": 5119,
      "sup_loss": 0.6803880696798582
    },
    {
      "train_step": 5120,
      "sup_loss": 0.5636848198068435
    },
    {
      "train_step": 5121,
      "sup_loss": 0.6826278101561041
    },
    {
      "train_step": 5122,
      "sup_loss": 0.3745477352174143
    },
    {
      "train_step": 5123,
      "sup_loss": 0.6034186110199755
    },
    {
      "train_step": 5124,
      "sup_loss": 0.5477584193728686
    },
    {
      "train_step": 5125,
      "sup_loss": 0.5885545299940471
    },
    {
      "train_step": 5126,
      "sup_loss": 0.6291531516759418
    },
    {
      "train_step": 5127,
      "sup_loss": 0.6696453372956804
    },
    {
      "train_step": 5128,
      "sup_loss": 0.6778723263676023
    },
    {
      "train_step": 5129,
      "sup_loss": 0.3918575677843299
    },
    {
      "train_step": 5130,
      "sup_loss": 0.6537481266173386
    },
    {
      "train_step": 5131,
      "sup_loss": 0.7244683853582089
    },
    {
      "train_step": 5132,
      "sup_loss": 0.5440892679892735
    },
    {
      "train_step": 5133,
      "sup_loss": 0.6255300865846967
    },
    {
      "train_step": 5134,
      "sup_loss": 0.6502139098687935
    },
    {
      "train_step": 5135,
      "sup_loss": 0.6323550619393155
    },
    {
      "train_step": 5136,
      "sup_loss": 0.38079238181825126
    },
    {
      "train_step": 5137,
      "sup_loss": 0.4319378518317796
    },
    {
      "train_step": 5138,
      "sup_loss": 0.7887819266013533
    },
    {
      "train_step": 5139,
      "sup_loss": 0.5651377417066864
    },
    {
      "train_step": 5140,
      "sup_loss": 0.7395132771776702
    },
    {
      "train_step": 5141,
      "sup_loss": 0.4017942340671617
    },
    {
      "train_step": 5142,
      "sup_loss": 0.7580681893093861
    },
    {
      "train_step": 5143,
      "sup_loss": 0.5906251379706352
    },
    {
      "train_step": 5144,
      "sup_loss": 0.5621213418933899
    },
    {
      "train_step": 5145,
      "sup_loss": 0.5987287827327642
    },
    {
      "train_step": 5146,
      "sup_loss": 0.5105475733678041
    },
    {
      "train_step": 5147,
      "sup_loss": 0.5793410608723187
    },
    {
      "train_step": 5148,
      "sup_loss": 0.5978134634209451
    },
    {
      "train_step": 5149,
      "sup_loss": 0.5352670400280454
    },
    {
      "train_step": 5150,
      "sup_loss": 0.5402373546370509
    },
    {
      "train_step": 5151,
      "sup_loss": 0.7342354049403845
    },
    {
      "train_step": 5152,
      "sup_loss": 0.805202657556492
    },
    {
      "train_step": 5153,
      "sup_loss": 0.8794129955224871
    },
    {
      "train_step": 5154,
      "sup_loss": 0.507677353440144
    },
    {
      "train_step": 5155,
      "sup_loss": 0.5913752679136771
    },
    {
      "train_step": 5156,
      "sup_loss": 0.41587008906798517
    },
    {
      "train_step": 5157,
      "sup_loss": 0.6101591024157693
    },
    {
      "train_step": 5158,
      "sup_loss": 0.7710708177527756
    },
    {
      "train_step": 5159,
      "sup_loss": 0.35521677395193546
    },
    {
      "train_step": 5160,
      "sup_loss": 0.5235465773110607
    },
    {
      "train_step": 5161,
      "sup_loss": 0.5419257478456341
    },
    {
      "train_step": 5162,
      "sup_loss": 0.6742631825534539
    },
    {
      "train_step": 5163,
      "sup_loss": 0.5643987591018091
    },
    {
      "train_step": 5164,
      "sup_loss": 0.48681526595334296
    },
    {
      "train_step": 5165,
      "sup_loss": 0.5727693892818768
    },
    {
      "train_step": 5166,
      "sup_loss": 0.5937773532939685
    },
    {
      "train_step": 5167,
      "sup_loss": 0.6176605209592428
    },
    {
      "train_step": 5168,
      "sup_loss": 0.4230679486235493
    },
    {
      "train_step": 5169,
      "sup_loss": 0.5057096021018732
    },
    {
      "train_step": 5170,
      "sup_loss": 0.4818755440298491
    },
    {
      "train_step": 5171,
      "sup_loss": 0.7259015934538232
    },
    {
      "train_step": 5172,
      "sup_loss": 0.3454941439266386
    },
    {
      "train_step": 5173,
      "sup_loss": 0.8306312233242883
    },
    {
      "train_step": 5174,
      "sup_loss": 0.68859193278217
    },
    {
      "train_step": 5175,
      "sup_loss": 0.5237353828280007
    },
    {
      "train_step": 5176,
      "sup_loss": 0.7191879279078864
    },
    {
      "train_step": 5177,
      "sup_loss": 0.6542711980170929
    },
    {
      "train_step": 5178,
      "sup_loss": 0.4609203620313286
    },
    {
      "train_step": 5179,
      "sup_loss": 0.2876407079266673
    },
    {
      "train_step": 5180,
      "sup_loss": 0.7250749070603206
    },
    {
      "train_step": 5181,
      "sup_loss": 0.4404354845951604
    },
    {
      "train_step": 5182,
      "sup_loss": 0.4212095427367454
    },
    {
      "train_step": 5183,
      "sup_loss": 0.7390106365121376
    },
    {
      "train_step": 5184,
      "sup_loss": 0.6518322731009906
    },
    {
      "train_step": 5185,
      "sup_loss": 0.540805618874634
    },
    {
      "train_step": 5186,
      "sup_loss": 0.6020390926678957
    },
    {
      "train_step": 5187,
      "sup_loss": 1.1174845829170237
    },
    {
      "train_step": 5188,
      "sup_loss": 0.4880044374143887
    },
    {
      "train_step": 5189,
      "sup_loss": 0.7686922766952918
    },
    {
      "train_step": 5190,
      "sup_loss": 0.6264308230774523
    },
    {
      "train_step": 5191,
      "sup_loss": 0.6266398626001619
    },
    {
      "train_step": 5192,
      "sup_loss": 0.6634330202532103
    },
    {
      "train_step": 5193,
      "sup_loss": 0.6757313293144042
    },
    {
      "train_step": 5194,
      "sup_loss": 0.5547096430369057
    },
    {
      "train_step": 5195,
      "sup_loss": 0.5002099290018
    },
    {
      "train_step": 5196,
      "sup_loss": 0.5769174436898086
    },
    {
      "train_step": 5197,
      "sup_loss": 0.5570535996644588
    },
    {
      "train_step": 5198,
      "sup_loss": 0.6337624035539646
    },
    {
      "train_step": 5199,
      "sup_loss": 0.512651735877638
    },
    {
      "train_step": 5200,
      "sup_loss": 0.7482372795266291
    },
    {
      "train_step": 5201,
      "sup_loss": 0.5634989986478391
    },
    {
      "train_step": 5202,
      "sup_loss": 0.78738338788158
    },
    {
      "train_step": 5203,
      "sup_loss": 0.7482238812848618
    },
    {
      "train_step": 5204,
      "sup_loss": 0.41998279866105165
    },
    {
      "train_step": 5205,
      "sup_loss": 0.3985262390299965
    },
    {
      "train_step": 5206,
      "sup_loss": 0.6001084972500308
    },
    {
      "train_step": 5207,
      "sup_loss": 0.8011206360259238
    },
    {
      "train_step": 5208,
      "sup_loss": 0.6310964523373948
    },
    {
      "train_step": 5209,
      "sup_loss": 0.5543863711634301
    },
    {
      "train_step": 5210,
      "sup_loss": 0.43540085365400366
    },
    {
      "train_step": 5211,
      "sup_loss": 0.5136870793173836
    },
    {
      "train_step": 5212,
      "sup_loss": 0.7634107166983501
    },
    {
      "train_step": 5213,
      "sup_loss": 0.5953623576583911
    },
    {
      "train_step": 5214,
      "sup_loss": 0.6014658376100821
    },
    {
      "train_step": 5215,
      "sup_loss": 0.7663742274096733
    },
    {
      "train_step": 5216,
      "sup_loss": 0.7146487421035942
    },
    {
      "train_step": 5217,
      "sup_loss": 0.4341531628429498
    },
    {
      "train_step": 5218,
      "sup_loss": 0.6223680042994988
    },
    {
      "train_step": 5219,
      "sup_loss": 0.4803683271322496
    },
    {
      "train_step": 5220,
      "sup_loss": 0.7737819947852678
    },
    {
      "train_step": 5221,
      "sup_loss": 0.576537091245736
    },
    {
      "train_step": 5222,
      "sup_loss": 0.4443950385209937
    },
    {
      "train_step": 5223,
      "sup_loss": 0.5879298210399173
    },
    {
      "train_step": 5224,
      "sup_loss": 0.6504693791911398
    },
    {
      "train_step": 5225,
      "sup_loss": 0.3464740119780871
    },
    {
      "train_step": 5226,
      "sup_loss": 0.8405131017037912
    },
    {
      "train_step": 5227,
      "sup_loss": 0.5681289697293637
    },
    {
      "train_step": 5228,
      "sup_loss": 0.6500515034864629
    },
    {
      "train_step": 5229,
      "sup_loss": 0.4946738625881356
    },
    {
      "train_step": 5230,
      "sup_loss": 0.5318634085930009
    },
    {
      "train_step": 5231,
      "sup_loss": 0.7781070840958507
    },
    {
      "train_step": 5232,
      "sup_loss": 0.4983458979048778
    },
    {
      "train_step": 5233,
      "sup_loss": 0.45151762017534003
    },
    {
      "train_step": 5234,
      "sup_loss": 0.29499433702173655
    },
    {
      "train_step": 5235,
      "sup_loss": 0.6444493181757894
    },
    {
      "train_step": 5236,
      "sup_loss": 0.6370587819442114
    },
    {
      "train_step": 5237,
      "sup_loss": 0.5397595199999333
    },
    {
      "train_step": 5238,
      "sup_loss": 0.41926494455838886
    },
    {
      "train_step": 5239,
      "sup_loss": 0.7517097752088207
    },
    {
      "train_step": 5240,
      "sup_loss": 0.4598156542801687
    },
    {
      "train_step": 5241,
      "sup_loss": 0.6697013134574903
    },
    {
      "train_step": 5242,
      "sup_loss": 0.5302153522024414
    },
    {
      "train_step": 5243,
      "sup_loss": 0.5632194032332141
    },
    {
      "train_step": 5244,
      "sup_loss": 0.6034064500825163
    },
    {
      "train_step": 5245,
      "sup_loss": 0.4982959740469933
    },
    {
      "train_step": 5246,
      "sup_loss": 0.5326434023067961
    },
    {
      "train_step": 5247,
      "sup_loss": 0.4774025617169678
    },
    {
      "train_step": 5248,
      "sup_loss": 0.436655451048541
    },
    {
      "train_step": 5249,
      "sup_loss": 0.5406512833673798
    },
    {
      "train_step": 5250,
      "sup_loss": 0.6136493362868828
    },
    {
      "train_step": 5251,
      "sup_loss": 0.8665692232213478
    },
    {
      "train_step": 5252,
      "sup_loss": 0.5800355949608851
    },
    {
      "train_step": 5253,
      "sup_loss": 0.531311103716457
    },
    {
      "train_step": 5254,
      "sup_loss": 0.7844293528939541
    },
    {
      "train_step": 5255,
      "sup_loss": 0.5709328450843467
    },
    {
      "train_step": 5256,
      "sup_loss": 0.38194798240374833
    },
    {
      "train_step": 5257,
      "sup_loss": 0.5878499304784913
    },
    {
      "train_step": 5258,
      "sup_loss": 0.4581891893758826
    },
    {
      "train_step": 5259,
      "sup_loss": 0.6609826531974213
    },
    {
      "train_step": 5260,
      "sup_loss": 0.6733486944454605
    },
    {
      "train_step": 5261,
      "sup_loss": 0.3836357312055753
    },
    {
      "train_step": 5262,
      "sup_loss": 0.42213629179289003
    },
    {
      "train_step": 5263,
      "sup_loss": 0.565345747964421
    },
    {
      "train_step": 5264,
      "sup_loss": 0.5991075040241216
    },
    {
      "train_step": 5265,
      "sup_loss": 0.7226698945542991
    },
    {
      "train_step": 5266,
      "sup_loss": 0.8540869243073647
    },
    {
      "train_step": 5267,
      "sup_loss": 0.5776743853163252
    },
    {
      "train_step": 5268,
      "sup_loss": 0.7952531221258887
    },
    {
      "train_step": 5269,
      "sup_loss": 0.7105696075416714
    },
    {
      "train_step": 5270,
      "sup_loss": 0.48193572382874617
    },
    {
      "train_step": 5271,
      "sup_loss": 0.46647303013279595
    },
    {
      "train_step": 5272,
      "sup_loss": 0.7315574204076516
    },
    {
      "train_step": 5273,
      "sup_loss": 0.7449542409913463
    },
    {
      "train_step": 5274,
      "sup_loss": 0.5019465452379199
    },
    {
      "train_step": 5275,
      "sup_loss": 0.4380405164047066
    },
    {
      "train_step": 5276,
      "sup_loss": 0.49413488289799673
    },
    {
      "train_step": 5277,
      "sup_loss": 0.7749747227644361
    },
    {
      "train_step": 5278,
      "sup_loss": 0.7661674580657678
    },
    {
      "train_step": 5279,
      "sup_loss": 0.461536283218255
    },
    {
      "train_step": 5280,
      "sup_loss": 0.7263744242703958
    },
    {
      "train_step": 5281,
      "sup_loss": 0.5222536366529049
    },
    {
      "train_step": 5282,
      "sup_loss": 0.6366697333540988
    },
    {
      "train_step": 5283,
      "sup_loss": 0.49331133762258295
    },
    {
      "train_step": 5284,
      "sup_loss": 0.5178324429843313
    },
    {
      "train_step": 5285,
      "sup_loss": 0.6066722083614627
    },
    {
      "train_step": 5286,
      "sup_loss": 0.5465741896543058
    },
    {
      "train_step": 5287,
      "sup_loss": 0.6256759241304
    },
    {
      "train_step": 5288,
      "sup_loss": 0.6458775244933681
    },
    {
      "train_step": 5289,
      "sup_loss": 0.6286111605101409
    },
    {
      "train_step": 5290,
      "sup_loss": 0.6276191153702149
    },
    {
      "train_step": 5291,
      "sup_loss": 0.5789980127499981
    },
    {
      "train_step": 5292,
      "sup_loss": 0.36107257427265127
    },
    {
      "train_step": 5293,
      "sup_loss": 0.49180814803126444
    },
    {
      "train_step": 5294,
      "sup_loss": 0.5730932921359673
    },
    {
      "train_step": 5295,
      "sup_loss": 0.5865715193392764
    },
    {
      "train_step": 5296,
      "sup_loss": 0.46437121327350417
    },
    {
      "train_step": 5297,
      "sup_loss": 0.5173281401770248
    },
    {
      "train_step": 5298,
      "sup_loss": 0.8031026543463702
    },
    {
      "train_step": 5299,
      "sup_loss": 0.733052417260817
    },
    {
      "train_step": 5300,
      "sup_loss": 0.5964334678211805
    },
    {
      "train_step": 5301,
      "sup_loss": 0.6660634655133415
    },
    {
      "train_step": 5302,
      "sup_loss": 0.6116556581870402
    },
    {
      "train_step": 5303,
      "sup_loss": 0.6517814577703464
    },
    {
      "train_step": 5304,
      "sup_loss": 0.7194920516359686
    },
    {
      "train_step": 5305,
      "sup_loss": 0.8136122613537253
    },
    {
      "train_step": 5306,
      "sup_loss": 0.562757162340658
    },
    {
      "train_step": 5307,
      "sup_loss": 0.551904709806356
    },
    {
      "train_step": 5308,
      "sup_loss": 0.6237564853660793
    },
    {
      "train_step": 5309,
      "sup_loss": 0.7869545777000556
    },
    {
      "train_step": 5310,
      "sup_loss": 0.7654330173833345
    },
    {
      "train_step": 5311,
      "sup_loss": 0.7246873747736675
    },
    {
      "train_step": 5312,
      "sup_loss": 0.4966238754788756
    },
    {
      "train_step": 5313,
      "sup_loss": 0.45312530084874947
    },
    {
      "train_step": 5314,
      "sup_loss": 0.746251070530985
    },
    {
      "train_step": 5315,
      "sup_loss": 0.5810574544684148
    },
    {
      "train_step": 5316,
      "sup_loss": 0.429717579925116
    },
    {
      "train_step": 5317,
      "sup_loss": 0.7321907381140031
    },
    {
      "train_step": 5318,
      "sup_loss": 0.5353371737717798
    },
    {
      "train_step": 5319,
      "sup_loss": 0.7607290983128666
    },
    {
      "train_step": 5320,
      "sup_loss": 0.5409487025220152
    },
    {
      "train_step": 5321,
      "sup_loss": 0.6974574423031676
    },
    {
      "train_step": 5322,
      "sup_loss": 0.47465410432980254
    },
    {
      "train_step": 5323,
      "sup_loss": 0.8348250945124824
    },
    {
      "train_step": 5324,
      "sup_loss": 0.4015006703876283
    },
    {
      "train_step": 5325,
      "sup_loss": 0.464481749067612
    },
    {
      "train_step": 5326,
      "sup_loss": 0.5229641597047727
    },
    {
      "train_step": 5327,
      "sup_loss": 0.8240945319799421
    },
    {
      "train_step": 5328,
      "sup_loss": 0.471584935679412
    },
    {
      "train_step": 5329,
      "sup_loss": 0.39437187789928035
    },
    {
      "train_step": 5330,
      "sup_loss": 0.6819177052079527
    },
    {
      "train_step": 5331,
      "sup_loss": 0.701049400834173
    },
    {
      "train_step": 5332,
      "sup_loss": 0.5384903251990937
    },
    {
      "train_step": 5333,
      "sup_loss": 0.6287950880678701
    },
    {
      "train_step": 5334,
      "sup_loss": 0.7102428380389678
    },
    {
      "train_step": 5335,
      "sup_loss": 0.5357892917605878
    },
    {
      "train_step": 5336,
      "sup_loss": 0.772070637834205
    },
    {
      "train_step": 5337,
      "sup_loss": 0.7391182019219265
    },
    {
      "train_step": 5338,
      "sup_loss": 0.6339539402903097
    },
    {
      "train_step": 5339,
      "sup_loss": 0.6228689568441317
    },
    {
      "train_step": 5340,
      "sup_loss": 0.41156771784530666
    },
    {
      "train_step": 5341,
      "sup_loss": 0.7405192115450538
    },
    {
      "train_step": 5342,
      "sup_loss": 0.6105990210195957
    },
    {
      "train_step": 5343,
      "sup_loss": 0.742930352033098
    },
    {
      "train_step": 5344,
      "sup_loss": 0.7527440341485607
    },
    {
      "train_step": 5345,
      "sup_loss": 0.5945353724989798
    },
    {
      "train_step": 5346,
      "sup_loss": 0.6671388157872995
    },
    {
      "train_step": 5347,
      "sup_loss": 0.32003242592018966
    },
    {
      "train_step": 5348,
      "sup_loss": 0.4100795842317314
    },
    {
      "train_step": 5349,
      "sup_loss": 0.5996259901040978
    },
    {
      "train_step": 5350,
      "sup_loss": 0.2867316280292235
    },
    {
      "train_step": 5351,
      "sup_loss": 0.3192931666480807
    },
    {
      "train_step": 5352,
      "sup_loss": 0.7112531391330483
    },
    {
      "train_step": 5353,
      "sup_loss": 0.4838630996675694
    },
    {
      "train_step": 5354,
      "sup_loss": 0.7556528030038757
    },
    {
      "train_step": 5355,
      "sup_loss": 0.6052825519616023
    },
    {
      "train_step": 5356,
      "sup_loss": 0.4158259733008862
    },
    {
      "train_step": 5357,
      "sup_loss": 0.5707754219437998
    },
    {
      "train_step": 5358,
      "sup_loss": 0.5748864935031391
    },
    {
      "train_step": 5359,
      "sup_loss": 0.5282517772831787
    },
    {
      "train_step": 5360,
      "sup_loss": 0.6714033824504968
    },
    {
      "train_step": 5361,
      "sup_loss": 0.5832476356072744
    },
    {
      "train_step": 5362,
      "sup_loss": 0.7330250930616403
    },
    {
      "train_step": 5363,
      "sup_loss": 0.5624553101696135
    },
    {
      "train_step": 5364,
      "sup_loss": 0.44197874933989595
    },
    {
      "train_step": 5365,
      "sup_loss": 0.4552603433833176
    },
    {
      "train_step": 5366,
      "sup_loss": 0.480784060608581
    },
    {
      "train_step": 5367,
      "sup_loss": 0.5205919788108743
    },
    {
      "train_step": 5368,
      "sup_loss": 0.560423720980303
    },
    {
      "train_step": 5369,
      "sup_loss": 0.5739761144302049
    },
    {
      "train_step": 5370,
      "sup_loss": 0.5354362295233233
    },
    {
      "train_step": 5371,
      "sup_loss": 0.7895326481096749
    },
    {
      "train_step": 5372,
      "sup_loss": 0.7009652008805303
    },
    {
      "train_step": 5373,
      "sup_loss": 0.6382535818987917
    },
    {
      "train_step": 5374,
      "sup_loss": 0.6047022899822742
    },
    {
      "train_step": 5375,
      "sup_loss": 0.6009521092423867
    },
    {
      "train_step": 5376,
      "sup_loss": 0.5874826270871064
    },
    {
      "train_step": 5377,
      "sup_loss": 0.8147885247415882
    },
    {
      "train_step": 5378,
      "sup_loss": 0.5886645917475193
    },
    {
      "train_step": 5379,
      "sup_loss": 0.507910766721363
    },
    {
      "train_step": 5380,
      "sup_loss": 0.6048184464969419
    },
    {
      "train_step": 5381,
      "sup_loss": 0.21631804155831186
    },
    {
      "train_step": 5382,
      "sup_loss": 0.581568745675681
    },
    {
      "train_step": 5383,
      "sup_loss": 0.706504062958881
    },
    {
      "train_step": 5384,
      "sup_loss": 0.6689293696181573
    },
    {
      "train_step": 5385,
      "sup_loss": 0.560993357716401
    },
    {
      "train_step": 5386,
      "sup_loss": 0.7052773718462877
    },
    {
      "train_step": 5387,
      "sup_loss": 0.46931492396579083
    },
    {
      "train_step": 5388,
      "sup_loss": 0.6616255150555783
    },
    {
      "train_step": 5389,
      "sup_loss": 0.2539094306546979
    },
    {
      "train_step": 5390,
      "sup_loss": 0.7701085121017415
    },
    {
      "train_step": 5391,
      "sup_loss": 0.5993614623633832
    },
    {
      "train_step": 5392,
      "sup_loss": 0.690710769766441
    },
    {
      "train_step": 5393,
      "sup_loss": 0.7541091707989238
    },
    {
      "train_step": 5394,
      "sup_loss": 0.49825844348424586
    },
    {
      "train_step": 5395,
      "sup_loss": 0.45300208613592197
    },
    {
      "train_step": 5396,
      "sup_loss": 0.7647488450483312
    },
    {
      "train_step": 5397,
      "sup_loss": 0.6639181385997445
    },
    {
      "train_step": 5398,
      "sup_loss": 0.4858757149746072
    },
    {
      "train_step": 5399,
      "sup_loss": 0.5853728043015404
    },
    {
      "train_step": 5400,
      "sup_loss": 0.5457754144387683
    },
    {
      "train_step": 5401,
      "sup_loss": 0.617119236264332
    },
    {
      "train_step": 5402,
      "sup_loss": 0.7381397563997766
    },
    {
      "train_step": 5403,
      "sup_loss": 0.5544916551533947
    },
    {
      "train_step": 5404,
      "sup_loss": 0.7465623362887268
    },
    {
      "train_step": 5405,
      "sup_loss": 0.6863571104112093
    },
    {
      "train_step": 5406,
      "sup_loss": 0.4887403017116032
    },
    {
      "train_step": 5407,
      "sup_loss": 0.5296074627070582
    },
    {
      "train_step": 5408,
      "sup_loss": 0.5403578880100589
    },
    {
      "train_step": 5409,
      "sup_loss": 0.6440389212435181
    },
    {
      "train_step": 5410,
      "sup_loss": 0.5918814615707544
    },
    {
      "train_step": 5411,
      "sup_loss": 0.44512841496905736
    },
    {
      "train_step": 5412,
      "sup_loss": 0.8078561923947102
    },
    {
      "train_step": 5413,
      "sup_loss": 0.6706285346475382
    },
    {
      "train_step": 5414,
      "sup_loss": 0.4171654489154926
    },
    {
      "train_step": 5415,
      "sup_loss": 0.4407783584687936
    },
    {
      "train_step": 5416,
      "sup_loss": 0.6480213337963574
    },
    {
      "train_step": 5417,
      "sup_loss": 0.599175213103149
    },
    {
      "train_step": 5418,
      "sup_loss": 0.7494353059905435
    },
    {
      "train_step": 5419,
      "sup_loss": 0.728065221456685
    },
    {
      "train_step": 5420,
      "sup_loss": 0.5244453715047787
    },
    {
      "train_step": 5421,
      "sup_loss": 0.5387422556337222
    },
    {
      "train_step": 5422,
      "sup_loss": 0.5570976000125039
    },
    {
      "train_step": 5423,
      "sup_loss": 0.4553158224476182
    },
    {
      "train_step": 5424,
      "sup_loss": 0.5625540943490066
    },
    {
      "train_step": 5425,
      "sup_loss": 0.7494110249964873
    },
    {
      "train_step": 5426,
      "sup_loss": 0.5295859237046713
    },
    {
      "train_step": 5427,
      "sup_loss": 0.45964852600903333
    },
    {
      "train_step": 5428,
      "sup_loss": 0.4823147650370873
    },
    {
      "train_step": 5429,
      "sup_loss": 0.653714811474575
    },
    {
      "train_step": 5430,
      "sup_loss": 0.4380852896563154
    },
    {
      "train_step": 5431,
      "sup_loss": 0.509173351543317
    },
    {
      "train_step": 5432,
      "sup_loss": 0.6588953486021458
    },
    {
      "train_step": 5433,
      "sup_loss": 0.6736529843148718
    },
    {
      "train_step": 5434,
      "sup_loss": 0.4472580730842246
    },
    {
      "train_step": 5435,
      "sup_loss": 0.7549834155282044
    },
    {
      "train_step": 5436,
      "sup_loss": 0.566107634581418
    },
    {
      "train_step": 5437,
      "sup_loss": 0.6665565339143193
    },
    {
      "train_step": 5438,
      "sup_loss": 0.3166495072384509
    },
    {
      "train_step": 5439,
      "sup_loss": 0.6132576673336373
    },
    {
      "train_step": 5440,
      "sup_loss": 0.5245816551475501
    },
    {
      "train_step": 5441,
      "sup_loss": 0.6423789789457764
    },
    {
      "train_step": 5442,
      "sup_loss": 0.6487270684629174
    },
    {
      "train_step": 5443,
      "sup_loss": 0.654081604181979
    },
    {
      "train_step": 5444,
      "sup_loss": 0.5841568784013127
    },
    {
      "train_step": 5445,
      "sup_loss": 0.5477361645200094
    },
    {
      "train_step": 5446,
      "sup_loss": 0.6902491411085779
    },
    {
      "train_step": 5447,
      "sup_loss": 0.49568148155554986
    },
    {
      "train_step": 5448,
      "sup_loss": 0.5333687273809289
    },
    {
      "train_step": 5449,
      "sup_loss": 0.5501995417074254
    },
    {
      "train_step": 5450,
      "sup_loss": 0.6345821460241488
    },
    {
      "train_step": 5451,
      "sup_loss": 0.5812810296439757
    },
    {
      "train_step": 5452,
      "sup_loss": 0.637805110558443
    },
    {
      "train_step": 5453,
      "sup_loss": 0.6652704493251758
    },
    {
      "train_step": 5454,
      "sup_loss": 0.6159974776821302
    },
    {
      "train_step": 5455,
      "sup_loss": 0.3749715272509326
    },
    {
      "train_step": 5456,
      "sup_loss": 0.5004049501492362
    },
    {
      "train_step": 5457,
      "sup_loss": 0.8246622035951302
    },
    {
      "train_step": 5458,
      "sup_loss": 0.7242288495039632
    },
    {
      "train_step": 5459,
      "sup_loss": 0.6506415155063114
    },
    {
      "train_step": 5460,
      "sup_loss": 0.7042164493626253
    },
    {
      "train_step": 5461,
      "sup_loss": 0.5836448339693034
    },
    {
      "train_step": 5462,
      "sup_loss": 0.5627687650304825
    },
    {
      "train_step": 5463,
      "sup_loss": 0.5105673753448422
    },
    {
      "train_step": 5464,
      "sup_loss": 0.5411674635372606
    },
    {
      "train_step": 5465,
      "sup_loss": 0.5357976556316625
    },
    {
      "train_step": 5466,
      "sup_loss": 0.589027913132095
    },
    {
      "train_step": 5467,
      "sup_loss": 0.33895796365402175
    },
    {
      "train_step": 5468,
      "sup_loss": 0.6525922942022546
    },
    {
      "train_step": 5469,
      "sup_loss": 0.5401111459111626
    },
    {
      "train_step": 5470,
      "sup_loss": 0.47618467649889495
    },
    {
      "train_step": 5471,
      "sup_loss": 0.6683041417067135
    },
    {
      "train_step": 5472,
      "sup_loss": 0.6890062218723265
    },
    {
      "train_step": 5473,
      "sup_loss": 0.515115923775969
    },
    {
      "train_step": 5474,
      "sup_loss": 0.692899933142616
    },
    {
      "train_step": 5475,
      "sup_loss": 0.756578048422741
    },
    {
      "train_step": 5476,
      "sup_loss": 0.5314048405415784
    },
    {
      "train_step": 5477,
      "sup_loss": 0.6142764143245101
    },
    {
      "train_step": 5478,
      "sup_loss": 0.5898381140948695
    },
    {
      "train_step": 5479,
      "sup_loss": 0.41265754784786796
    },
    {
      "train_step": 5480,
      "sup_loss": 0.5414408981434028
    },
    {
      "train_step": 5481,
      "sup_loss": 0.7929756009706239
    },
    {
      "train_step": 5482,
      "sup_loss": 0.6101045690849759
    },
    {
      "train_step": 5483,
      "sup_loss": 0.7171207833554327
    },
    {
      "train_step": 5484,
      "sup_loss": 0.8389347885397258
    },
    {
      "train_step": 5485,
      "sup_loss": 0.24302989995300894
    },
    {
      "train_step": 5486,
      "sup_loss": 0.6881802411377135
    },
    {
      "train_step": 5487,
      "sup_loss": 0.4469524697201517
    },
    {
      "train_step": 5488,
      "sup_loss": 0.6925419132350618
    },
    {
      "train_step": 5489,
      "sup_loss": 0.4108668254785278
    },
    {
      "train_step": 5490,
      "sup_loss": 0.43041664899514603
    },
    {
      "train_step": 5491,
      "sup_loss": 0.6785981685401151
    },
    {
      "train_step": 5492,
      "sup_loss": 0.6551082296711328
    },
    {
      "train_step": 5493,
      "sup_loss": 0.8120273582744914
    },
    {
      "train_step": 5494,
      "sup_loss": 0.5142093933945073
    },
    {
      "train_step": 5495,
      "sup_loss": 0.5670191549158791
    },
    {
      "train_step": 5496,
      "sup_loss": 0.5836423656051404
    },
    {
      "train_step": 5497,
      "sup_loss": 0.681306589802009
    },
    {
      "train_step": 5498,
      "sup_loss": 0.4923769692599191
    },
    {
      "train_step": 5499,
      "sup_loss": 0.6006321069807533
    },
    {
      "train_step": 5500,
      "sup_loss": 0.4458404443137627
    },
    {
      "train_step": 5501,
      "sup_loss": 0.8083015015542636
    },
    {
      "train_step": 5502,
      "sup_loss": 0.48965750147074394
    },
    {
      "train_step": 5503,
      "sup_loss": 0.5884201347252359
    },
    {
      "train_step": 5504,
      "sup_loss": 0.692370749303022
    },
    {
      "train_step": 5505,
      "sup_loss": 0.6850937813536071
    },
    {
      "train_step": 5506,
      "sup_loss": 0.7554604011047917
    },
    {
      "train_step": 5507,
      "sup_loss": 0.5688362888160285
    },
    {
      "train_step": 5508,
      "sup_loss": 0.6459503065151184
    },
    {
      "train_step": 5509,
      "sup_loss": 0.66842768174045
    },
    {
      "train_step": 5510,
      "sup_loss": 0.5154091627159147
    },
    {
      "train_step": 5511,
      "sup_loss": 0.6191281706330456
    },
    {
      "train_step": 5512,
      "sup_loss": 0.6927664043598881
    },
    {
      "train_step": 5513,
      "sup_loss": 0.5939687598274203
    },
    {
      "train_step": 5514,
      "sup_loss": 0.6405964660359925
    },
    {
      "train_step": 5515,
      "sup_loss": 0.7554266020081707
    },
    {
      "train_step": 5516,
      "sup_loss": 0.602746456708375
    },
    {
      "train_step": 5517,
      "sup_loss": 0.6051630374847814
    },
    {
      "train_step": 5518,
      "sup_loss": 0.45840265444863687
    },
    {
      "train_step": 5519,
      "sup_loss": 0.5823809947081457
    },
    {
      "train_step": 5520,
      "sup_loss": 0.6240676307609749
    },
    {
      "train_step": 5521,
      "sup_loss": 0.7536413298264201
    },
    {
      "train_step": 5522,
      "sup_loss": 0.6081968782768595
    },
    {
      "train_step": 5523,
      "sup_loss": 0.48580535089169424
    },
    {
      "train_step": 5524,
      "sup_loss": 0.5115642004062004
    },
    {
      "train_step": 5525,
      "sup_loss": 0.5794177444315948
    },
    {
      "train_step": 5526,
      "sup_loss": 0.5670414694844801
    },
    {
      "train_step": 5527,
      "sup_loss": 0.6266429899882574
    },
    {
      "train_step": 5528,
      "sup_loss": 0.5986186935794224
    },
    {
      "train_step": 5529,
      "sup_loss": 0.5471908354038115
    },
    {
      "train_step": 5530,
      "sup_loss": 0.6091586125981957
    },
    {
      "train_step": 5531,
      "sup_loss": 0.45956225531181527
    },
    {
      "train_step": 5532,
      "sup_loss": 0.5716658562909659
    },
    {
      "train_step": 5533,
      "sup_loss": 0.8166084148837168
    },
    {
      "train_step": 5534,
      "sup_loss": 0.6307688048834621
    },
    {
      "train_step": 5535,
      "sup_loss": 0.3286338067901234
    },
    {
      "train_step": 5536,
      "sup_loss": 0.5584822271793082
    },
    {
      "train_step": 5537,
      "sup_loss": 0.7383660631089842
    },
    {
      "train_step": 5538,
      "sup_loss": 0.7944729786459138
    },
    {
      "train_step": 5539,
      "sup_loss": 0.5946480799409671
    },
    {
      "train_step": 5540,
      "sup_loss": 0.4077384419127615
    },
    {
      "train_step": 5541,
      "sup_loss": 0.53938445401126
    },
    {
      "train_step": 5542,
      "sup_loss": 0.4795099080793475
    },
    {
      "train_step": 5543,
      "sup_loss": 0.698729655176693
    },
    {
      "train_step": 5544,
      "sup_loss": 0.569528634688729
    },
    {
      "train_step": 5545,
      "sup_loss": 0.627331412894001
    },
    {
      "train_step": 5546,
      "sup_loss": 0.5657743569925926
    },
    {
      "train_step": 5547,
      "sup_loss": 0.6657046941104572
    },
    {
      "train_step": 5548,
      "sup_loss": 0.5909123418950407
    },
    {
      "train_step": 5549,
      "sup_loss": 0.6399234624086979
    },
    {
      "train_step": 5550,
      "sup_loss": 0.7034465733041317
    },
    {
      "train_step": 5551,
      "sup_loss": 0.3631693138220749
    },
    {
      "train_step": 5552,
      "sup_loss": 0.5687636945626043
    },
    {
      "train_step": 5553,
      "sup_loss": 0.5391341878860493
    },
    {
      "train_step": 5554,
      "sup_loss": 0.7487849880728266
    },
    {
      "train_step": 5555,
      "sup_loss": 0.6375407516571345
    },
    {
      "train_step": 5556,
      "sup_loss": 0.6370648985797721
    },
    {
      "train_step": 5557,
      "sup_loss": 0.3223937860516702
    },
    {
      "train_step": 5558,
      "sup_loss": 0.516060411639524
    },
    {
      "train_step": 5559,
      "sup_loss": 0.4142823459231694
    },
    {
      "train_step": 5560,
      "sup_loss": 0.5619004436554548
    },
    {
      "train_step": 5561,
      "sup_loss": 0.5181442162991644
    },
    {
      "train_step": 5562,
      "sup_loss": 0.47169821049026256
    },
    {
      "train_step": 5563,
      "sup_loss": 0.49144786116295563
    },
    {
      "train_step": 5564,
      "sup_loss": 0.4916370587754483
    },
    {
      "train_step": 5565,
      "sup_loss": 0.6765451768232181
    },
    {
      "train_step": 5566,
      "sup_loss": 0.303436687588916
    },
    {
      "train_step": 5567,
      "sup_loss": 0.6858396835624229
    },
    {
      "train_step": 5568,
      "sup_loss": 0.5205730654411743
    },
    {
      "train_step": 5569,
      "sup_loss": 0.5080630103514314
    },
    {
      "train_step": 5570,
      "sup_loss": 0.5618983420925611
    },
    {
      "train_step": 5571,
      "sup_loss": 0.4497243702617279
    },
    {
      "train_step": 5572,
      "sup_loss": 0.822558692361833
    },
    {
      "train_step": 5573,
      "sup_loss": 0.6349805653492671
    },
    {
      "train_step": 5574,
      "sup_loss": 0.663945095303747
    },
    {
      "train_step": 5575,
      "sup_loss": 0.6006802266910736
    },
    {
      "train_step": 5576,
      "sup_loss": 0.7433763868569637
    },
    {
      "train_step": 5577,
      "sup_loss": 0.4644064955759949
    },
    {
      "train_step": 5578,
      "sup_loss": 0.691218284340437
    },
    {
      "train_step": 5579,
      "sup_loss": 0.4503289063896076
    },
    {
      "train_step": 5580,
      "sup_loss": 0.616443239667753
    },
    {
      "train_step": 5581,
      "sup_loss": 0.4660673135011002
    },
    {
      "train_step": 5582,
      "sup_loss": 0.5661844289892606
    },
    {
      "train_step": 5583,
      "sup_loss": 0.694604362434037
    },
    {
      "train_step": 5584,
      "sup_loss": 0.42069275162299735
    },
    {
      "train_step": 5585,
      "sup_loss": 0.42433056838842437
    },
    {
      "train_step": 5586,
      "sup_loss": 0.6248053433432174
    },
    {
      "train_step": 5587,
      "sup_loss": 0.7052010040393426
    },
    {
      "train_step": 5588,
      "sup_loss": 0.7232899932525555
    },
    {
      "train_step": 5589,
      "sup_loss": 0.6563074269176753
    },
    {
      "train_step": 5590,
      "sup_loss": 0.6594972061134702
    },
    {
      "train_step": 5591,
      "sup_loss": 0.5379802557079664
    },
    {
      "train_step": 5592,
      "sup_loss": 0.656099014711067
    },
    {
      "train_step": 5593,
      "sup_loss": 0.6152941165791069
    },
    {
      "train_step": 5594,
      "sup_loss": 0.5634776491309559
    },
    {
      "train_step": 5595,
      "sup_loss": 0.580687527590984
    },
    {
      "train_step": 5596,
      "sup_loss": 0.6375769333940677
    },
    {
      "train_step": 5597,
      "sup_loss": 0.5502690414681255
    },
    {
      "train_step": 5598,
      "sup_loss": 0.5090663999209388
    },
    {
      "train_step": 5599,
      "sup_loss": 0.6388566729658294
    },
    {
      "train_step": 5600,
      "sup_loss": 0.5291649785833327
    },
    {
      "train_step": 5601,
      "sup_loss": 0.5700783001665589
    },
    {
      "train_step": 5602,
      "sup_loss": 0.6514962963978068
    },
    {
      "train_step": 5603,
      "sup_loss": 0.7658404916956514
    },
    {
      "train_step": 5604,
      "sup_loss": 0.42447664481452835
    },
    {
      "train_step": 5605,
      "sup_loss": 0.6454861866324353
    },
    {
      "train_step": 5606,
      "sup_loss": 0.6679248499261704
    },
    {
      "train_step": 5607,
      "sup_loss": 0.6500693936991826
    },
    {
      "train_step": 5608,
      "sup_loss": 0.5324564209889735
    },
    {
      "train_step": 5609,
      "sup_loss": 0.5652480294479877
    },
    {
      "train_step": 5610,
      "sup_loss": 0.5332021002373063
    },
    {
      "train_step": 5611,
      "sup_loss": 0.6051879839359084
    },
    {
      "train_step": 5612,
      "sup_loss": 0.4327630963888307
    },
    {
      "train_step": 5613,
      "sup_loss": 0.2812004939468996
    },
    {
      "train_step": 5614,
      "sup_loss": 0.6157828290104416
    },
    {
      "train_step": 5615,
      "sup_loss": 0.7794704907664254
    },
    {
      "train_step": 5616,
      "sup_loss": 0.42098516798134605
    },
    {
      "train_step": 5617,
      "sup_loss": 0.5710418691201459
    },
    {
      "train_step": 5618,
      "sup_loss": 0.9428475944283486
    },
    {
      "train_step": 5619,
      "sup_loss": 0.6383145125995782
    },
    {
      "train_step": 5620,
      "sup_loss": 0.6151782842453349
    },
    {
      "train_step": 5621,
      "sup_loss": 0.44100505958378194
    },
    {
      "train_step": 5622,
      "sup_loss": 0.5281319742680125
    },
    {
      "train_step": 5623,
      "sup_loss": 0.8019350723058818
    },
    {
      "train_step": 5624,
      "sup_loss": 0.5258938918222068
    },
    {
      "train_step": 5625,
      "sup_loss": 0.4586949549727148
    },
    {
      "train_step": 5626,
      "sup_loss": 0.6801165282801658
    },
    {
      "train_step": 5627,
      "sup_loss": 0.5636889880408581
    },
    {
      "train_step": 5628,
      "sup_loss": 0.8610047286275239
    },
    {
      "train_step": 5629,
      "sup_loss": 0.48489959063519544
    },
    {
      "train_step": 5630,
      "sup_loss": 0.6195090838009207
    },
    {
      "train_step": 5631,
      "sup_loss": 0.6240035755074025
    },
    {
      "train_step": 5632,
      "sup_loss": 0.568160191198514
    },
    {
      "train_step": 5633,
      "sup_loss": 0.6484457690874736
    },
    {
      "train_step": 5634,
      "sup_loss": 0.5679876244175555
    },
    {
      "train_step": 5635,
      "sup_loss": 0.6029693129084162
    },
    {
      "train_step": 5636,
      "sup_loss": 0.7045461451541597
    },
    {
      "train_step": 5637,
      "sup_loss": 0.7022910105097132
    },
    {
      "train_step": 5638,
      "sup_loss": 0.6591235555405985
    },
    {
      "train_step": 5639,
      "sup_loss": 0.4701427143898791
    },
    {
      "train_step": 5640,
      "sup_loss": 0.5798996427775709
    },
    {
      "train_step": 5641,
      "sup_loss": 0.7503157355089219
    },
    {
      "train_step": 5642,
      "sup_loss": 0.6861737256488131
    },
    {
      "train_step": 5643,
      "sup_loss": 0.5624088022817391
    },
    {
      "train_step": 5644,
      "sup_loss": 0.6661425005526159
    },
    {
      "train_step": 5645,
      "sup_loss": 0.7172313537291881
    },
    {
      "train_step": 5646,
      "sup_loss": 0.7212147638150098
    },
    {
      "train_step": 5647,
      "sup_loss": 0.5913106856883519
    },
    {
      "train_step": 5648,
      "sup_loss": 0.845227832413445
    },
    {
      "train_step": 5649,
      "sup_loss": 0.7830480366793133
    },
    {
      "train_step": 5650,
      "sup_loss": 0.7638919723766198
    },
    {
      "train_step": 5651,
      "sup_loss": 0.46752792296734186
    },
    {
      "train_step": 5652,
      "sup_loss": 0.8385442444024888
    },
    {
      "train_step": 5653,
      "sup_loss": 0.5558068923470559
    },
    {
      "train_step": 5654,
      "sup_loss": 0.6183646167596921
    },
    {
      "train_step": 5655,
      "sup_loss": 0.6434937900838333
    },
    {
      "train_step": 5656,
      "sup_loss": 0.42507822252132477
    },
    {
      "train_step": 5657,
      "sup_loss": 0.5475122274998253
    },
    {
      "train_step": 5658,
      "sup_loss": 0.5879605168718767
    },
    {
      "train_step": 5659,
      "sup_loss": 0.609299454649649
    },
    {
      "train_step": 5660,
      "sup_loss": 0.6121381609425076
    },
    {
      "train_step": 5661,
      "sup_loss": 0.6465518505052396
    },
    {
      "train_step": 5662,
      "sup_loss": 0.5464195968587396
    },
    {
      "train_step": 5663,
      "sup_loss": 0.7961761578950204
    },
    {
      "train_step": 5664,
      "sup_loss": 0.5833934735155187
    },
    {
      "train_step": 5665,
      "sup_loss": 0.7399911091157146
    },
    {
      "train_step": 5666,
      "sup_loss": 0.7042671225961971
    },
    {
      "train_step": 5667,
      "sup_loss": 0.6448228002767217
    },
    {
      "train_step": 5668,
      "sup_loss": 0.4566071889971578
    },
    {
      "train_step": 5669,
      "sup_loss": 0.4571616771624531
    },
    {
      "train_step": 5670,
      "sup_loss": 0.3095651417508809
    },
    {
      "train_step": 5671,
      "sup_loss": 0.364662061208317
    },
    {
      "train_step": 5672,
      "sup_loss": 0.5960660646383962
    },
    {
      "train_step": 5673,
      "sup_loss": 0.44841061156517575
    },
    {
      "train_step": 5674,
      "sup_loss": 0.592163932973784
    },
    {
      "train_step": 5675,
      "sup_loss": 0.5163928796879336
    },
    {
      "train_step": 5676,
      "sup_loss": 0.666809955289579
    },
    {
      "train_step": 5677,
      "sup_loss": 0.48230709873301447
    },
    {
      "train_step": 5678,
      "sup_loss": 0.641400944399796
    },
    {
      "train_step": 5679,
      "sup_loss": 0.5398327977083658
    },
    {
      "train_step": 5680,
      "sup_loss": 0.4859766259169612
    },
    {
      "train_step": 5681,
      "sup_loss": 0.48695923786894807
    },
    {
      "train_step": 5682,
      "sup_loss": 0.5711623451591863
    },
    {
      "train_step": 5683,
      "sup_loss": 0.6885024257153486
    },
    {
      "train_step": 5684,
      "sup_loss": 0.38819135875213684
    },
    {
      "train_step": 5685,
      "sup_loss": 0.5772149162602306
    },
    {
      "train_step": 5686,
      "sup_loss": 0.5729514513441984
    },
    {
      "train_step": 5687,
      "sup_loss": 0.7040312343684344
    },
    {
      "train_step": 5688,
      "sup_loss": 0.6042055588527183
    },
    {
      "train_step": 5689,
      "sup_loss": 0.6269172547438977
    },
    {
      "train_step": 5690,
      "sup_loss": 0.6295485602712841
    },
    {
      "train_step": 5691,
      "sup_loss": 0.5145583012679444
    },
    {
      "train_step": 5692,
      "sup_loss": 0.53807518032784
    },
    {
      "train_step": 5693,
      "sup_loss": 0.49953031433482564
    },
    {
      "train_step": 5694,
      "sup_loss": 0.5786814860867892
    },
    {
      "train_step": 5695,
      "sup_loss": 0.7019275810974293
    },
    {
      "train_step": 5696,
      "sup_loss": 0.6498339044683171
    },
    {
      "train_step": 5697,
      "sup_loss": 0.5784718499379266
    },
    {
      "train_step": 5698,
      "sup_loss": 0.617080757845663
    },
    {
      "train_step": 5699,
      "sup_loss": 0.5457601302553081
    },
    {
      "train_step": 5700,
      "sup_loss": 0.5851266911262556
    },
    {
      "train_step": 5701,
      "sup_loss": 0.6900138477433583
    },
    {
      "train_step": 5702,
      "sup_loss": 0.6057866399499293
    },
    {
      "train_step": 5703,
      "sup_loss": 0.6014989340191593
    },
    {
      "train_step": 5704,
      "sup_loss": 0.686668376705047
    },
    {
      "train_step": 5705,
      "sup_loss": 0.5113221504263463
    },
    {
      "train_step": 5706,
      "sup_loss": 0.6209879798796221
    },
    {
      "train_step": 5707,
      "sup_loss": 0.6945604966924349
    },
    {
      "train_step": 5708,
      "sup_loss": 0.5284246036425367
    },
    {
      "train_step": 5709,
      "sup_loss": 0.8319098542049936
    },
    {
      "train_step": 5710,
      "sup_loss": 0.3879161090521438
    },
    {
      "train_step": 5711,
      "sup_loss": 0.6000154284775983
    },
    {
      "train_step": 5712,
      "sup_loss": 0.644268990676852
    },
    {
      "train_step": 5713,
      "sup_loss": 0.4820650522927569
    },
    {
      "train_step": 5714,
      "sup_loss": 0.6357451209673869
    },
    {
      "train_step": 5715,
      "sup_loss": 0.6804731560728069
    },
    {
      "train_step": 5716,
      "sup_loss": 0.6039154320637683
    },
    {
      "train_step": 5717,
      "sup_loss": 0.5097942008237866
    },
    {
      "train_step": 5718,
      "sup_loss": 0.4169101194486133
    },
    {
      "train_step": 5719,
      "sup_loss": 0.8147940939933127
    },
    {
      "train_step": 5720,
      "sup_loss": 0.5858307611033085
    },
    {
      "train_step": 5721,
      "sup_loss": 0.5161236885674865
    },
    {
      "train_step": 5722,
      "sup_loss": 0.5487161793740722
    },
    {
      "train_step": 5723,
      "sup_loss": 0.6781389524442499
    },
    {
      "train_step": 5724,
      "sup_loss": 0.7263733637202762
    },
    {
      "train_step": 5725,
      "sup_loss": 0.3587576245081618
    },
    {
      "train_step": 5726,
      "sup_loss": 0.2896208012181759
    },
    {
      "train_step": 5727,
      "sup_loss": 0.37827984110106944
    },
    {
      "train_step": 5728,
      "sup_loss": 0.5359251595434003
    },
    {
      "train_step": 5729,
      "sup_loss": 0.6887180190070595
    },
    {
      "train_step": 5730,
      "sup_loss": 0.5579343637910161
    },
    {
      "train_step": 5731,
      "sup_loss": 0.4988817388116895
    },
    {
      "train_step": 5732,
      "sup_loss": 0.5958936885632153
    },
    {
      "train_step": 5733,
      "sup_loss": 0.7419673358910335
    },
    {
      "train_step": 5734,
      "sup_loss": 0.7867032329502665
    },
    {
      "train_step": 5735,
      "sup_loss": 0.2923532032744569
    },
    {
      "train_step": 5736,
      "sup_loss": 0.5442187408851357
    },
    {
      "train_step": 5737,
      "sup_loss": 0.6049534967289399
    },
    {
      "train_step": 5738,
      "sup_loss": 0.4556681081322689
    },
    {
      "train_step": 5739,
      "sup_loss": 0.4906905343404044
    },
    {
      "train_step": 5740,
      "sup_loss": 0.594423242377205
    },
    {
      "train_step": 5741,
      "sup_loss": 0.5015807321683586
    },
    {
      "train_step": 5742,
      "sup_loss": 0.5222629095915713
    },
    {
      "train_step": 5743,
      "sup_loss": 0.5121337853008769
    },
    {
      "train_step": 5744,
      "sup_loss": 0.40339936114529373
    },
    {
      "train_step": 5745,
      "sup_loss": 0.539168913463623
    },
    {
      "train_step": 5746,
      "sup_loss": 0.6581713964523407
    },
    {
      "train_step": 5747,
      "sup_loss": 0.5897369801196597
    },
    {
      "train_step": 5748,
      "sup_loss": 0.5383890400862381
    },
    {
      "train_step": 5749,
      "sup_loss": 0.8266070440919806
    },
    {
      "train_step": 5750,
      "sup_loss": 0.6695240794210388
    },
    {
      "train_step": 5751,
      "sup_loss": 0.5124593176241691
    },
    {
      "train_step": 5752,
      "sup_loss": 0.6907682841613889
    },
    {
      "train_step": 5753,
      "sup_loss": 0.5473888323439389
    },
    {
      "train_step": 5754,
      "sup_loss": 0.6262641365942032
    },
    {
      "train_step": 5755,
      "sup_loss": 0.25939458255823006
    },
    {
      "train_step": 5756,
      "sup_loss": 0.6094126303097511
    },
    {
      "train_step": 5757,
      "sup_loss": 0.7410859477192345
    },
    {
      "train_step": 5758,
      "sup_loss": 0.6257365538907179
    },
    {
      "train_step": 5759,
      "sup_loss": 0.7660963943276751
    },
    {
      "train_step": 5760,
      "sup_loss": 0.5858063011175062
    },
    {
      "train_step": 5761,
      "sup_loss": 0.6494635348078213
    },
    {
      "train_step": 5762,
      "sup_loss": 0.5158855578191869
    },
    {
      "train_step": 5763,
      "sup_loss": 0.4070477533328482
    },
    {
      "train_step": 5764,
      "sup_loss": 0.49217978259941353
    },
    {
      "train_step": 5765,
      "sup_loss": 0.6367360378784969
    },
    {
      "train_step": 5766,
      "sup_loss": 0.618013520422263
    },
    {
      "train_step": 5767,
      "sup_loss": 0.49113284873362933
    },
    {
      "train_step": 5768,
      "sup_loss": 0.46706999053979437
    },
    {
      "train_step": 5769,
      "sup_loss": 0.565939699756834
    },
    {
      "train_step": 5770,
      "sup_loss": 0.5669120929598606
    },
    {
      "train_step": 5771,
      "sup_loss": 0.44001982379686155
    },
    {
      "train_step": 5772,
      "sup_loss": 0.6218225175245629
    },
    {
      "train_step": 5773,
      "sup_loss": 0.6159946082461936
    },
    {
      "train_step": 5774,
      "sup_loss": 0.7222141135364066
    },
    {
      "train_step": 5775,
      "sup_loss": 0.5583968194183407
    },
    {
      "train_step": 5776,
      "sup_loss": 0.34363860534065405
    },
    {
      "train_step": 5777,
      "sup_loss": 0.7068555671009957
    },
    {
      "train_step": 5778,
      "sup_loss": 0.5351356051439154
    },
    {
      "train_step": 5779,
      "sup_loss": 0.7769240857070592
    },
    {
      "train_step": 5780,
      "sup_loss": 0.674830774505938
    },
    {
      "train_step": 5781,
      "sup_loss": 0.6893501120988901
    },
    {
      "train_step": 5782,
      "sup_loss": 0.413537302948806
    },
    {
      "train_step": 5783,
      "sup_loss": 0.6995043876478648
    },
    {
      "train_step": 5784,
      "sup_loss": 0.4677417974837211
    },
    {
      "train_step": 5785,
      "sup_loss": 0.46713600317651455
    },
    {
      "train_step": 5786,
      "sup_loss": 0.687938930550226
    },
    {
      "train_step": 5787,
      "sup_loss": 0.5357948882859893
    },
    {
      "train_step": 5788,
      "sup_loss": 0.6195268498586183
    },
    {
      "train_step": 5789,
      "sup_loss": 0.692141188785518
    },
    {
      "train_step": 5790,
      "sup_loss": 0.6494696532583579
    },
    {
      "train_step": 5791,
      "sup_loss": 0.5535358718055393
    },
    {
      "train_step": 5792,
      "sup_loss": 0.6867496606012414
    },
    {
      "train_step": 5793,
      "sup_loss": 0.532151272298574
    },
    {
      "train_step": 5794,
      "sup_loss": 0.6965865538508053
    },
    {
      "train_step": 5795,
      "sup_loss": 0.46139116660622886
    },
    {
      "train_step": 5796,
      "sup_loss": 0.6321411880047461
    },
    {
      "train_step": 5797,
      "sup_loss": 0.725005381885861
    },
    {
      "train_step": 5798,
      "sup_loss": 0.5760698297626868
    },
    {
      "train_step": 5799,
      "sup_loss": 0.6159727298794263
    },
    {
      "train_step": 5800,
      "sup_loss": 0.6457165780160277
    },
    {
      "train_step": 5801,
      "sup_loss": 0.7175557003852051
    },
    {
      "train_step": 5802,
      "sup_loss": 0.5787385546933134
    },
    {
      "train_step": 5803,
      "sup_loss": 0.7682007722219647
    },
    {
      "train_step": 5804,
      "sup_loss": 0.6037399219330954
    },
    {
      "train_step": 5805,
      "sup_loss": 0.40414575719990603
    },
    {
      "train_step": 5806,
      "sup_loss": 0.6730796818600269
    },
    {
      "train_step": 5807,
      "sup_loss": 0.6857743920270467
    },
    {
      "train_step": 5808,
      "sup_loss": 0.627495290600819
    },
    {
      "train_step": 5809,
      "sup_loss": 0.8343131894708075
    },
    {
      "train_step": 5810,
      "sup_loss": 0.4223302650675068
    },
    {
      "train_step": 5811,
      "sup_loss": 0.7145053222466077
    },
    {
      "train_step": 5812,
      "sup_loss": 0.5028404898969049
    },
    {
      "train_step": 5813,
      "sup_loss": 0.34587348594838274
    },
    {
      "train_step": 5814,
      "sup_loss": 0.6121519564777228
    },
    {
      "train_step": 5815,
      "sup_loss": 0.647637628897019
    },
    {
      "train_step": 5816,
      "sup_loss": 0.4718313734486893
    },
    {
      "train_step": 5817,
      "sup_loss": 0.6261279842862142
    },
    {
      "train_step": 5818,
      "sup_loss": 0.6018313281541726
    },
    {
      "train_step": 5819,
      "sup_loss": 0.5157350328637395
    },
    {
      "train_step": 5820,
      "sup_loss": 0.6311841055772588
    },
    {
      "train_step": 5821,
      "sup_loss": 0.3817839191455579
    },
    {
      "train_step": 5822,
      "sup_loss": 0.6163392537891216
    },
    {
      "train_step": 5823,
      "sup_loss": 0.4099872356551084
    },
    {
      "train_step": 5824,
      "sup_loss": 0.47162288600955615
    },
    {
      "train_step": 5825,
      "sup_loss": 0.2533721755942552
    },
    {
      "train_step": 5826,
      "sup_loss": 0.2929033171061978
    },
    {
      "train_step": 5827,
      "sup_loss": 0.5671522290312689
    },
    {
      "train_step": 5828,
      "sup_loss": 0.8374482233959804
    },
    {
      "train_step": 5829,
      "sup_loss": 0.46079363712827776
    },
    {
      "train_step": 5830,
      "sup_loss": 0.6560982485169002
    },
    {
      "train_step": 5831,
      "sup_loss": 0.5650684172917076
    },
    {
      "train_step": 5832,
      "sup_loss": 0.4362140371901303
    },
    {
      "train_step": 5833,
      "sup_loss": 0.6533128670058014
    },
    {
      "train_step": 5834,
      "sup_loss": 0.6739370586626694
    },
    {
      "train_step": 5835,
      "sup_loss": 0.5958461814792406
    },
    {
      "train_step": 5836,
      "sup_loss": 0.5646009658926102
    },
    {
      "train_step": 5837,
      "sup_loss": 0.5049471797635037
    },
    {
      "train_step": 5838,
      "sup_loss": 0.4998709725119136
    },
    {
      "train_step": 5839,
      "sup_loss": 0.7235001309859307
    },
    {
      "train_step": 5840,
      "sup_loss": 0.4780567297100746
    },
    {
      "train_step": 5841,
      "sup_loss": 0.5745640555956265
    },
    {
      "train_step": 5842,
      "sup_loss": 0.38901510179571697
    },
    {
      "train_step": 5843,
      "sup_loss": 0.6001785125329113
    },
    {
      "train_step": 5844,
      "sup_loss": 0.7187174451081564
    },
    {
      "train_step": 5845,
      "sup_loss": 0.5545631524301975
    },
    {
      "train_step": 5846,
      "sup_loss": 0.5509076492964416
    },
    {
      "train_step": 5847,
      "sup_loss": 0.6389427021782832
    },
    {
      "train_step": 5848,
      "sup_loss": 0.8094122247922698
    },
    {
      "train_step": 5849,
      "sup_loss": 0.7552548471389379
    },
    {
      "train_step": 5850,
      "sup_loss": 0.5736757214685231
    },
    {
      "train_step": 5851,
      "sup_loss": 0.30654077952543884
    },
    {
      "train_step": 5852,
      "sup_loss": 0.7012585216639762
    },
    {
      "train_step": 5853,
      "sup_loss": 0.4250243190067068
    },
    {
      "train_step": 5854,
      "sup_loss": 0.5743675421829103
    },
    {
      "train_step": 5855,
      "sup_loss": 0.6453766324032388
    },
    {
      "train_step": 5856,
      "sup_loss": 0.6325910585462262
    },
    {
      "train_step": 5857,
      "sup_loss": 0.7713461789778949
    },
    {
      "train_step": 5858,
      "sup_loss": 0.3004127282735108
    },
    {
      "train_step": 5859,
      "sup_loss": 0.7288653397988695
    },
    {
      "train_step": 5860,
      "sup_loss": 0.6338597393288687
    },
    {
      "train_step": 5861,
      "sup_loss": 0.6597107642754665
    },
    {
      "train_step": 5862,
      "sup_loss": 0.7601094076305296
    },
    {
      "train_step": 5863,
      "sup_loss": 0.5863360303463058
    },
    {
      "train_step": 5864,
      "sup_loss": 0.34916000441800343
    },
    {
      "train_step": 5865,
      "sup_loss": 0.30805506940498356
    },
    {
      "train_step": 5866,
      "sup_loss": 0.44229923265023086
    },
    {
      "train_step": 5867,
      "sup_loss": 0.8458615348385465
    },
    {
      "train_step": 5868,
      "sup_loss": 0.5809481679259981
    },
    {
      "train_step": 5869,
      "sup_loss": 0.5085200642595131
    },
    {
      "train_step": 5870,
      "sup_loss": 0.4205285817134881
    },
    {
      "train_step": 5871,
      "sup_loss": 0.4351600918403712
    },
    {
      "train_step": 5872,
      "sup_loss": 0.39840806572631926
    },
    {
      "train_step": 5873,
      "sup_loss": 0.5657993172238485
    },
    {
      "train_step": 5874,
      "sup_loss": 0.5067354340091104
    },
    {
      "train_step": 5875,
      "sup_loss": 0.6479571367564562
    },
    {
      "train_step": 5876,
      "sup_loss": 0.50469162515829
    },
    {
      "train_step": 5877,
      "sup_loss": 0.5580489429868614
    },
    {
      "train_step": 5878,
      "sup_loss": 0.44584240079172305
    },
    {
      "train_step": 5879,
      "sup_loss": 0.8049658061257171
    },
    {
      "train_step": 5880,
      "sup_loss": 0.7157453833009741
    },
    {
      "train_step": 5881,
      "sup_loss": 0.5252308527668987
    },
    {
      "train_step": 5882,
      "sup_loss": 0.456899369637654
    },
    {
      "train_step": 5883,
      "sup_loss": 0.7054555627965644
    },
    {
      "train_step": 5884,
      "sup_loss": 0.5817087713285398
    },
    {
      "train_step": 5885,
      "sup_loss": 0.627112524677339
    },
    {
      "train_step": 5886,
      "sup_loss": 0.5923908466727165
    },
    {
      "train_step": 5887,
      "sup_loss": 0.9906243410452913
    },
    {
      "train_step": 5888,
      "sup_loss": 0.2613351987067238
    },
    {
      "train_step": 5889,
      "sup_loss": 0.6049267847702504
    },
    {
      "train_step": 5890,
      "sup_loss": 0.5603352738953414
    },
    {
      "train_step": 5891,
      "sup_loss": 0.6218621803535779
    },
    {
      "train_step": 5892,
      "sup_loss": 0.6639946187133281
    },
    {
      "train_step": 5893,
      "sup_loss": 0.536499805761121
    },
    {
      "train_step": 5894,
      "sup_loss": 0.33854688762116164
    },
    {
      "train_step": 5895,
      "sup_loss": 0.3335977937005421
    },
    {
      "train_step": 5896,
      "sup_loss": 0.5384757626685069
    },
    {
      "train_step": 5897,
      "sup_loss": 0.6246059513826976
    },
    {
      "train_step": 5898,
      "sup_loss": 0.6303583772461135
    },
    {
      "train_step": 5899,
      "sup_loss": 0.35025063259079314
    },
    {
      "train_step": 5900,
      "sup_loss": 0.3391879473019137
    },
    {
      "train_step": 5901,
      "sup_loss": 0.643007783794678
    },
    {
      "train_step": 5902,
      "sup_loss": 0.5613338475562426
    },
    {
      "train_step": 5903,
      "sup_loss": 0.5971370146017705
    },
    {
      "train_step": 5904,
      "sup_loss": 0.6439672736470127
    },
    {
      "train_step": 5905,
      "sup_loss": 0.5545810916782498
    },
    {
      "train_step": 5906,
      "sup_loss": 0.48548239199277265
    },
    {
      "train_step": 5907,
      "sup_loss": 0.5323855465141806
    },
    {
      "train_step": 5908,
      "sup_loss": 0.6209482455395066
    },
    {
      "train_step": 5909,
      "sup_loss": 0.7746374397320355
    },
    {
      "train_step": 5910,
      "sup_loss": 0.9156887835471191
    },
    {
      "train_step": 5911,
      "sup_loss": 0.734808413279869
    },
    {
      "train_step": 5912,
      "sup_loss": 0.7869355599323722
    },
    {
      "train_step": 5913,
      "sup_loss": 0.6802554537835181
    },
    {
      "train_step": 5914,
      "sup_loss": 0.49894103278328455
    },
    {
      "train_step": 5915,
      "sup_loss": 0.4915929622502331
    },
    {
      "train_step": 5916,
      "sup_loss": 0.4522622817535803
    },
    {
      "train_step": 5917,
      "sup_loss": 0.42579340270471205
    },
    {
      "train_step": 5918,
      "sup_loss": 0.5493117673704944
    },
    {
      "train_step": 5919,
      "sup_loss": 0.7688689774514553
    },
    {
      "train_step": 5920,
      "sup_loss": 0.7562498651718569
    },
    {
      "train_step": 5921,
      "sup_loss": 0.4752966283084578
    },
    {
      "train_step": 5922,
      "sup_loss": 0.750656147486345
    },
    {
      "train_step": 5923,
      "sup_loss": 0.7135266302232529
    },
    {
      "train_step": 5924,
      "sup_loss": 0.6334676476790295
    },
    {
      "train_step": 5925,
      "sup_loss": 0.5735315991917889
    },
    {
      "train_step": 5926,
      "sup_loss": 0.7258510455264265
    },
    {
      "train_step": 5927,
      "sup_loss": 0.6637974932537347
    },
    {
      "train_step": 5928,
      "sup_loss": 0.5542850398523373
    },
    {
      "train_step": 5929,
      "sup_loss": 0.6291079117477912
    },
    {
      "train_step": 5930,
      "sup_loss": 0.47137072120322704
    },
    {
      "train_step": 5931,
      "sup_loss": 0.45673465053455725
    },
    {
      "train_step": 5932,
      "sup_loss": 0.45397611699469986
    },
    {
      "train_step": 5933,
      "sup_loss": 0.7149084885499148
    },
    {
      "train_step": 5934,
      "sup_loss": 0.6740023454498183
    },
    {
      "train_step": 5935,
      "sup_loss": 0.4573980470296857
    },
    {
      "train_step": 5936,
      "sup_loss": 0.7575393918184706
    },
    {
      "train_step": 5937,
      "sup_loss": 0.6083964099984539
    },
    {
      "train_step": 5938,
      "sup_loss": 0.7790651547245466
    },
    {
      "train_step": 5939,
      "sup_loss": 0.4675023481535897
    },
    {
      "train_step": 5940,
      "sup_loss": 0.5383128312296814
    },
    {
      "train_step": 5941,
      "sup_loss": 0.48258622853796546
    },
    {
      "train_step": 5942,
      "sup_loss": 0.6943420326474256
    },
    {
      "train_step": 5943,
      "sup_loss": 0.49684389511925964
    },
    {
      "train_step": 5944,
      "sup_loss": 0.4697866685642738
    },
    {
      "train_step": 5945,
      "sup_loss": 0.3835869619416898
    },
    {
      "train_step": 5946,
      "sup_loss": 0.5850937688426984
    },
    {
      "train_step": 5947,
      "sup_loss": 0.6065974826169295
    },
    {
      "train_step": 5948,
      "sup_loss": 0.3072523783072163
    },
    {
      "train_step": 5949,
      "sup_loss": 0.44826242180887504
    },
    {
      "train_step": 5950,
      "sup_loss": 0.4733775787683517
    },
    {
      "train_step": 5951,
      "sup_loss": 0.6431375594860307
    },
    {
      "train_step": 5952,
      "sup_loss": 0.613605171125369
    },
    {
      "train_step": 5953,
      "sup_loss": 0.5390906321047259
    },
    {
      "train_step": 5954,
      "sup_loss": 0.6810435158252703
    },
    {
      "train_step": 5955,
      "sup_loss": 0.5321686561421736
    },
    {
      "train_step": 5956,
      "sup_loss": 0.46274037505370474
    },
    {
      "train_step": 5957,
      "sup_loss": 0.5385881572270949
    },
    {
      "train_step": 5958,
      "sup_loss": 0.48212621938781836
    },
    {
      "train_step": 5959,
      "sup_loss": 0.706959264189559
    },
    {
      "train_step": 5960,
      "sup_loss": 0.49565966078578144
    },
    {
      "train_step": 5961,
      "sup_loss": 0.725925431437083
    },
    {
      "train_step": 5962,
      "sup_loss": 0.570411120038777
    },
    {
      "train_step": 5963,
      "sup_loss": 0.5837751582867877
    },
    {
      "train_step": 5964,
      "sup_loss": 0.7309270214546572
    },
    {
      "train_step": 5965,
      "sup_loss": 0.5362807794274788
    },
    {
      "train_step": 5966,
      "sup_loss": 0.6069848608422025
    },
    {
      "train_step": 5967,
      "sup_loss": 0.8388645065196584
    },
    {
      "train_step": 5968,
      "sup_loss": 0.5081003236017976
    },
    {
      "train_step": 5969,
      "sup_loss": 0.6094133727625501
    },
    {
      "train_step": 5970,
      "sup_loss": 0.7100939696017122
    },
    {
      "train_step": 5971,
      "sup_loss": 0.7422957079708948
    },
    {
      "train_step": 5972,
      "sup_loss": 0.32578718961527897
    },
    {
      "train_step": 5973,
      "sup_loss": 0.6732131475610947
    },
    {
      "train_step": 5974,
      "sup_loss": 0.5644081460191882
    },
    {
      "train_step": 5975,
      "sup_loss": 0.4308076734280382
    },
    {
      "train_step": 5976,
      "sup_loss": 0.5811498317682844
    },
    {
      "train_step": 5977,
      "sup_loss": 0.5560985201179176
    },
    {
      "train_step": 5978,
      "sup_loss": 0.63204091559171
    },
    {
      "train_step": 5979,
      "sup_loss": 0.41932766317291637
    },
    {
      "train_step": 5980,
      "sup_loss": 0.6747723138518057
    },
    {
      "train_step": 5981,
      "sup_loss": 0.7645773750364407
    },
    {
      "train_step": 5982,
      "sup_loss": 0.6719748897174732
    },
    {
      "train_step": 5983,
      "sup_loss": 0.5886860139846423
    },
    {
      "train_step": 5984,
      "sup_loss": 0.5816779377464186
    },
    {
      "train_step": 5985,
      "sup_loss": 0.40681965698968847
    },
    {
      "train_step": 5986,
      "sup_loss": 0.6607593810226084
    },
    {
      "train_step": 5987,
      "sup_loss": 0.43165183763517345
    },
    {
      "train_step": 5988,
      "sup_loss": 0.5813930961070939
    },
    {
      "train_step": 5989,
      "sup_loss": 0.45696240040689223
    },
    {
      "train_step": 5990,
      "sup_loss": 0.38354223821137806
    },
    {
      "train_step": 5991,
      "sup_loss": 0.4352515322824047
    },
    {
      "train_step": 5992,
      "sup_loss": 0.4149548275506441
    },
    {
      "train_step": 5993,
      "sup_loss": 0.6450665059527578
    },
    {
      "train_step": 5994,
      "sup_loss": 0.7938189720121239
    },
    {
      "train_step": 5995,
      "sup_loss": 0.49892032650738694
    },
    {
      "train_step": 5996,
      "sup_loss": 0.5320306595658684
    },
    {
      "train_step": 5997,
      "sup_loss": 0.4135882878667406
    },
    {
      "train_step": 5998,
      "sup_loss": 0.5188504501777271
    },
    {
      "train_step": 5999,
      "sup_loss": 0.6072799361794045
    },
    {
      "train_step": 6000,
      "sup_loss": 0.5153628324320948
    },
    {
      "train_step": 6001,
      "sup_loss": 0.5282866280473211
    },
    {
      "train_step": 6002,
      "sup_loss": 0.5897945721017257
    },
    {
      "train_step": 6003,
      "sup_loss": 0.5048227118992582
    },
    {
      "train_step": 6004,
      "sup_loss": 0.5520817394466231
    },
    {
      "train_step": 6005,
      "sup_loss": 0.4154540147302516
    },
    {
      "train_step": 6006,
      "sup_loss": 0.6576357639776877
    },
    {
      "train_step": 6007,
      "sup_loss": 0.7138670853015145
    },
    {
      "train_step": 6008,
      "sup_loss": 0.7178425155102265
    },
    {
      "train_step": 6009,
      "sup_loss": 0.6145930160952369
    },
    {
      "train_step": 6010,
      "sup_loss": 0.6100058089080304
    },
    {
      "train_step": 6011,
      "sup_loss": 0.4584996678528286
    },
    {
      "train_step": 6012,
      "sup_loss": 0.6801933943891985
    },
    {
      "train_step": 6013,
      "sup_loss": 0.568556825597282
    },
    {
      "train_step": 6014,
      "sup_loss": 0.6515342447525966
    },
    {
      "train_step": 6015,
      "sup_loss": 0.61476000869188
    },
    {
      "train_step": 6016,
      "sup_loss": 0.37582226463055174
    },
    {
      "train_step": 6017,
      "sup_loss": 0.437479127290189
    },
    {
      "train_step": 6018,
      "sup_loss": 0.616130442701721
    },
    {
      "train_step": 6019,
      "sup_loss": 0.4661684697126862
    },
    {
      "train_step": 6020,
      "sup_loss": 0.38208145350940054
    },
    {
      "train_step": 6021,
      "sup_loss": 0.7345675236834075
    },
    {
      "train_step": 6022,
      "sup_loss": 0.45608821019852813
    },
    {
      "train_step": 6023,
      "sup_loss": 0.4723600089726889
    },
    {
      "train_step": 6024,
      "sup_loss": 0.5334198666606111
    },
    {
      "train_step": 6025,
      "sup_loss": 0.5881602940311513
    },
    {
      "train_step": 6026,
      "sup_loss": 0.7553603641851148
    },
    {
      "train_step": 6027,
      "sup_loss": 0.5835121319778714
    },
    {
      "train_step": 6028,
      "sup_loss": 0.5875814817730342
    },
    {
      "train_step": 6029,
      "sup_loss": 0.7230346983466212
    },
    {
      "train_step": 6030,
      "sup_loss": 0.5531324244889058
    },
    {
      "train_step": 6031,
      "sup_loss": 0.5330239459283906
    },
    {
      "train_step": 6032,
      "sup_loss": 0.36867473380648424
    },
    {
      "train_step": 6033,
      "sup_loss": 0.41638380885083165
    },
    {
      "train_step": 6034,
      "sup_loss": 0.5931002536345197
    },
    {
      "train_step": 6035,
      "sup_loss": 0.5181264051002537
    },
    {
      "train_step": 6036,
      "sup_loss": 0.3455682950878736
    },
    {
      "train_step": 6037,
      "sup_loss": 0.5868287647175955
    },
    {
      "train_step": 6038,
      "sup_loss": 0.3583629983975507
    },
    {
      "train_step": 6039,
      "sup_loss": 0.44699601699110675
    },
    {
      "train_step": 6040,
      "sup_loss": 0.5558458115881395
    },
    {
      "train_step": 6041,
      "sup_loss": 0.36594280499345383
    },
    {
      "train_step": 6042,
      "sup_loss": 0.4111826749664307
    },
    {
      "train_step": 6043,
      "sup_loss": 0.39915547546999863
    },
    {
      "train_step": 6044,
      "sup_loss": 0.6997281318329582
    },
    {
      "train_step": 6045,
      "sup_loss": 0.4049703008667731
    },
    {
      "train_step": 6046,
      "sup_loss": 0.3866197354138447
    },
    {
      "train_step": 6047,
      "sup_loss": 0.7689248544510089
    },
    {
      "train_step": 6048,
      "sup_loss": 0.7383107089912724
    },
    {
      "train_step": 6049,
      "sup_loss": 0.5742381323799866
    },
    {
      "train_step": 6050,
      "sup_loss": 0.4850373020370957
    },
    {
      "train_step": 6051,
      "sup_loss": 0.6221273137743596
    },
    {
      "train_step": 6052,
      "sup_loss": 0.5933325400282508
    },
    {
      "train_step": 6053,
      "sup_loss": 0.5261626727193531
    },
    {
      "train_step": 6054,
      "sup_loss": 0.6632610069957181
    },
    {
      "train_step": 6055,
      "sup_loss": 0.7147527701770234
    },
    {
      "train_step": 6056,
      "sup_loss": 0.7220698968971336
    },
    {
      "train_step": 6057,
      "sup_loss": 0.5585895817135593
    },
    {
      "train_step": 6058,
      "sup_loss": 0.693915985425718
    },
    {
      "train_step": 6059,
      "sup_loss": 0.7378447221474084
    },
    {
      "train_step": 6060,
      "sup_loss": 0.5996530732587831
    },
    {
      "train_step": 6061,
      "sup_loss": 0.627331687332535
    },
    {
      "train_step": 6062,
      "sup_loss": 0.6310560370807362
    },
    {
      "train_step": 6063,
      "sup_loss": 0.6127577309029861
    },
    {
      "train_step": 6064,
      "sup_loss": 0.6647972551355996
    },
    {
      "train_step": 6065,
      "sup_loss": 0.7125166338245688
    },
    {
      "train_step": 6066,
      "sup_loss": 0.7034359202567076
    },
    {
      "train_step": 6067,
      "sup_loss": 0.7823371568058608
    },
    {
      "train_step": 6068,
      "sup_loss": 0.6424416673348627
    },
    {
      "train_step": 6069,
      "sup_loss": 0.37098017990719123
    },
    {
      "train_step": 6070,
      "sup_loss": 0.5218044380008549
    },
    {
      "train_step": 6071,
      "sup_loss": 0.7745028471313905
    },
    {
      "train_step": 6072,
      "sup_loss": 0.19620446851673012
    },
    {
      "train_step": 6073,
      "sup_loss": 0.5557464593431692
    },
    {
      "train_step": 6074,
      "sup_loss": 0.3273591295639151
    },
    {
      "train_step": 6075,
      "sup_loss": 0.7412719653382877
    },
    {
      "train_step": 6076,
      "sup_loss": 0.47947196488626376
    },
    {
      "train_step": 6077,
      "sup_loss": 0.7021127854022591
    },
    {
      "train_step": 6078,
      "sup_loss": 0.7498115357380353
    },
    {
      "train_step": 6079,
      "sup_loss": 0.6733666803645139
    },
    {
      "train_step": 6080,
      "sup_loss": 0.7193912988061497
    },
    {
      "train_step": 6081,
      "sup_loss": 0.5269215531293202
    },
    {
      "train_step": 6082,
      "sup_loss": 0.3647385752406844
    },
    {
      "train_step": 6083,
      "sup_loss": 0.6092490936537263
    },
    {
      "train_step": 6084,
      "sup_loss": 0.6205958589595322
    },
    {
      "train_step": 6085,
      "sup_loss": 0.6688104034095069
    },
    {
      "train_step": 6086,
      "sup_loss": 0.37885784622728347
    },
    {
      "train_step": 6087,
      "sup_loss": 0.4432518024308997
    },
    {
      "train_step": 6088,
      "sup_loss": 0.5094307796881226
    },
    {
      "train_step": 6089,
      "sup_loss": 0.5175316747803115
    },
    {
      "train_step": 6090,
      "sup_loss": 0.6409308550409514
    },
    {
      "train_step": 6091,
      "sup_loss": 0.5522572189001093
    },
    {
      "train_step": 6092,
      "sup_loss": 0.5853165636484188
    },
    {
      "train_step": 6093,
      "sup_loss": 0.325604338541831
    },
    {
      "train_step": 6094,
      "sup_loss": 0.5142459396260338
    },
    {
      "train_step": 6095,
      "sup_loss": 0.6202517281699536
    },
    {
      "train_step": 6096,
      "sup_loss": 0.6387062313884532
    },
    {
      "train_step": 6097,
      "sup_loss": 0.48995049072667207
    },
    {
      "train_step": 6098,
      "sup_loss": 0.7507629099098482
    },
    {
      "train_step": 6099,
      "sup_loss": 0.630130573134662
    },
    {
      "train_step": 6100,
      "sup_loss": 0.5610476539671319
    },
    {
      "train_step": 6101,
      "sup_loss": 0.46495847304387483
    },
    {
      "train_step": 6102,
      "sup_loss": 0.5934515503631338
    },
    {
      "train_step": 6103,
      "sup_loss": 0.675349741970148
    },
    {
      "train_step": 6104,
      "sup_loss": 0.5397010714978709
    },
    {
      "train_step": 6105,
      "sup_loss": 0.7083755114376473
    },
    {
      "train_step": 6106,
      "sup_loss": 0.546549271689977
    },
    {
      "train_step": 6107,
      "sup_loss": 0.4180965802748533
    },
    {
      "train_step": 6108,
      "sup_loss": 0.7104552439956191
    },
    {
      "train_step": 6109,
      "sup_loss": 0.6932393960556402
    },
    {
      "train_step": 6110,
      "sup_loss": 0.5420170024600842
    },
    {
      "train_step": 6111,
      "sup_loss": 0.4869077113132476
    },
    {
      "train_step": 6112,
      "sup_loss": 0.5472200127187382
    },
    {
      "train_step": 6113,
      "sup_loss": 0.4872513629530288
    },
    {
      "train_step": 6114,
      "sup_loss": 0.7159226385445591
    },
    {
      "train_step": 6115,
      "sup_loss": 0.4646975744235759
    },
    {
      "train_step": 6116,
      "sup_loss": 0.48501788172015337
    },
    {
      "train_step": 6117,
      "sup_loss": 0.308293303657486
    },
    {
      "train_step": 6118,
      "sup_loss": 0.544607094118797
    },
    {
      "train_step": 6119,
      "sup_loss": 0.6909205775458525
    },
    {
      "train_step": 6120,
      "sup_loss": 0.5473531679667243
    },
    {
      "train_step": 6121,
      "sup_loss": 0.6133946128969756
    },
    {
      "train_step": 6122,
      "sup_loss": 0.5349191775034361
    },
    {
      "train_step": 6123,
      "sup_loss": 0.3831831782822612
    },
    {
      "train_step": 6124,
      "sup_loss": 0.49741475099179294
    },
    {
      "train_step": 6125,
      "sup_loss": 0.33371923837041495
    },
    {
      "train_step": 6126,
      "sup_loss": 0.5662261354507613
    },
    {
      "train_step": 6127,
      "sup_loss": 0.5344997517530179
    },
    {
      "train_step": 6128,
      "sup_loss": 0.6127687247333884
    },
    {
      "train_step": 6129,
      "sup_loss": 0.6731634355006948
    },
    {
      "train_step": 6130,
      "sup_loss": 0.5072745886314547
    },
    {
      "train_step": 6131,
      "sup_loss": 0.5753892972769004
    },
    {
      "train_step": 6132,
      "sup_loss": 0.37726567101266933
    },
    {
      "train_step": 6133,
      "sup_loss": 0.3724999398924439
    },
    {
      "train_step": 6134,
      "sup_loss": 0.5109037475355324
    },
    {
      "train_step": 6135,
      "sup_loss": 0.720019373383667
    },
    {
      "train_step": 6136,
      "sup_loss": 0.6663517679677105
    },
    {
      "train_step": 6137,
      "sup_loss": 0.45877206973151774
    },
    {
      "train_step": 6138,
      "sup_loss": 0.2854067274969469
    },
    {
      "train_step": 6139,
      "sup_loss": 0.590244549176914
    },
    {
      "train_step": 6140,
      "sup_loss": 0.6280251689994424
    },
    {
      "train_step": 6141,
      "sup_loss": 0.5904478227992593
    },
    {
      "train_step": 6142,
      "sup_loss": 0.5001023308413117
    },
    {
      "train_step": 6143,
      "sup_loss": 0.5196636758613307
    },
    {
      "train_step": 6144,
      "sup_loss": 0.7223501590502257
    },
    {
      "train_step": 6145,
      "sup_loss": 0.4882945033523188
    },
    {
      "train_step": 6146,
      "sup_loss": 0.6340966858495276
    },
    {
      "train_step": 6147,
      "sup_loss": 0.4391302663578804
    },
    {
      "train_step": 6148,
      "sup_loss": 0.7011831511031993
    },
    {
      "train_step": 6149,
      "sup_loss": 0.5561068009600616
    },
    {
      "train_step": 6150,
      "sup_loss": 0.472642315911722
    },
    {
      "train_step": 6151,
      "sup_loss": 0.25945527027767945
    },
    {
      "train_step": 6152,
      "sup_loss": 0.6656799267569178
    },
    {
      "train_step": 6153,
      "sup_loss": 0.6217351266222805
    },
    {
      "train_step": 6154,
      "sup_loss": 0.5031475067882434
    },
    {
      "train_step": 6155,
      "sup_loss": 0.6695601122490386
    },
    {
      "train_step": 6156,
      "sup_loss": 0.5152853805077704
    },
    {
      "train_step": 6157,
      "sup_loss": 0.4238847135524931
    },
    {
      "train_step": 6158,
      "sup_loss": 0.6312902683697033
    },
    {
      "train_step": 6159,
      "sup_loss": 0.7204978121735881
    },
    {
      "train_step": 6160,
      "sup_loss": 0.6663413902736627
    },
    {
      "train_step": 6161,
      "sup_loss": 0.3405781321976802
    },
    {
      "train_step": 6162,
      "sup_loss": 0.5309270218994916
    },
    {
      "train_step": 6163,
      "sup_loss": 0.5309366270474333
    },
    {
      "train_step": 6164,
      "sup_loss": 0.5414836813226018
    },
    {
      "train_step": 6165,
      "sup_loss": 0.7067633388421378
    },
    {
      "train_step": 6166,
      "sup_loss": 0.4487629417429749
    },
    {
      "train_step": 6167,
      "sup_loss": 0.6942553452266106
    },
    {
      "train_step": 6168,
      "sup_loss": 0.45242274252608433
    },
    {
      "train_step": 6169,
      "sup_loss": 0.6962641884021167
    },
    {
      "train_step": 6170,
      "sup_loss": 0.6461125420367592
    },
    {
      "train_step": 6171,
      "sup_loss": 0.7848811406395486
    },
    {
      "train_step": 6172,
      "sup_loss": 0.3395078307100886
    },
    {
      "train_step": 6173,
      "sup_loss": 0.4862434633168865
    },
    {
      "train_step": 6174,
      "sup_loss": 0.606696483602418
    },
    {
      "train_step": 6175,
      "sup_loss": 0.6165872254094197
    },
    {
      "train_step": 6176,
      "sup_loss": 0.624940526425869
    },
    {
      "train_step": 6177,
      "sup_loss": 0.6391431946719733
    },
    {
      "train_step": 6178,
      "sup_loss": 0.6686437386789105
    },
    {
      "train_step": 6179,
      "sup_loss": 0.46430477174796286
    },
    {
      "train_step": 6180,
      "sup_loss": 0.577407519830737
    },
    {
      "train_step": 6181,
      "sup_loss": 0.6469803487240843
    },
    {
      "train_step": 6182,
      "sup_loss": 0.7499797018993477
    },
    {
      "train_step": 6183,
      "sup_loss": 0.4956945996299575
    },
    {
      "train_step": 6184,
      "sup_loss": 0.6024466565229339
    },
    {
      "train_step": 6185,
      "sup_loss": 0.6769039379902102
    },
    {
      "train_step": 6186,
      "sup_loss": 0.5169039349251046
    },
    {
      "train_step": 6187,
      "sup_loss": 0.6635226871555284
    },
    {
      "train_step": 6188,
      "sup_loss": 0.7391362096931269
    },
    {
      "train_step": 6189,
      "sup_loss": 0.6698490666650455
    },
    {
      "train_step": 6190,
      "sup_loss": 0.35231123478872917
    },
    {
      "train_step": 6191,
      "sup_loss": 0.6020918144999727
    },
    {
      "train_step": 6192,
      "sup_loss": 0.603617650140525
    },
    {
      "train_step": 6193,
      "sup_loss": 0.5735724451453768
    },
    {
      "train_step": 6194,
      "sup_loss": 0.5931880689439847
    },
    {
      "train_step": 6195,
      "sup_loss": 0.4680237164068528
    },
    {
      "train_step": 6196,
      "sup_loss": 0.4944516142684543
    },
    {
      "train_step": 6197,
      "sup_loss": 0.4045475627799561
    },
    {
      "train_step": 6198,
      "sup_loss": 0.6186610360304517
    },
    {
      "train_step": 6199,
      "sup_loss": 0.7865438711888698
    },
    {
      "train_step": 6200,
      "sup_loss": 0.5376286439640113
    },
    {
      "train_step": 6201,
      "sup_loss": 0.2585071590685488
    },
    {
      "train_step": 6202,
      "sup_loss": 0.6385855033471837
    },
    {
      "train_step": 6203,
      "sup_loss": 0.6533648392172613
    },
    {
      "train_step": 6204,
      "sup_loss": 0.9994944103513
    },
    {
      "train_step": 6205,
      "sup_loss": 0.5920672579678353
    },
    {
      "train_step": 6206,
      "sup_loss": 0.40036838525112495
    },
    {
      "train_step": 6207,
      "sup_loss": 0.3505858116685657
    },
    {
      "train_step": 6208,
      "sup_loss": 0.730884708423733
    },
    {
      "train_step": 6209,
      "sup_loss": 0.7070007180342038
    },
    {
      "train_step": 6210,
      "sup_loss": 0.5232539175782632
    },
    {
      "train_step": 6211,
      "sup_loss": 0.4291172467434318
    },
    {
      "train_step": 6212,
      "sup_loss": 0.490620900902014
    },
    {
      "train_step": 6213,
      "sup_loss": 0.6071392488919946
    },
    {
      "train_step": 6214,
      "sup_loss": 0.5798826765992356
    },
    {
      "train_step": 6215,
      "sup_loss": 0.5554103952537053
    },
    {
      "train_step": 6216,
      "sup_loss": 0.4835603056270186
    },
    {
      "train_step": 6217,
      "sup_loss": 0.6483503012878795
    },
    {
      "train_step": 6218,
      "sup_loss": 0.5456647450426962
    },
    {
      "train_step": 6219,
      "sup_loss": 0.6473337041695504
    },
    {
      "train_step": 6220,
      "sup_loss": 0.5164054999857329
    },
    {
      "train_step": 6221,
      "sup_loss": 0.7306080789020838
    },
    {
      "train_step": 6222,
      "sup_loss": 0.43802946620195093
    },
    {
      "train_step": 6223,
      "sup_loss": 0.6195042096101304
    },
    {
      "train_step": 6224,
      "sup_loss": 0.3071607796957554
    },
    {
      "train_step": 6225,
      "sup_loss": 0.4516202010354801
    },
    {
      "train_step": 6226,
      "sup_loss": 0.6015567725384122
    },
    {
      "train_step": 6227,
      "sup_loss": 0.6444245833141901
    },
    {
      "train_step": 6228,
      "sup_loss": 0.5961673540822026
    },
    {
      "train_step": 6229,
      "sup_loss": 0.6027377307304649
    },
    {
      "train_step": 6230,
      "sup_loss": 0.6315102441909104
    },
    {
      "train_step": 6231,
      "sup_loss": 0.43886257383220484
    },
    {
      "train_step": 6232,
      "sup_loss": 0.5609455633943248
    },
    {
      "train_step": 6233,
      "sup_loss": 0.6669818765505172
    },
    {
      "train_step": 6234,
      "sup_loss": 0.5757968344359907
    },
    {
      "train_step": 6235,
      "sup_loss": 0.5993264810607822
    },
    {
      "train_step": 6236,
      "sup_loss": 0.49384957392486856
    },
    {
      "train_step": 6237,
      "sup_loss": 0.5014153330742459
    },
    {
      "train_step": 6238,
      "sup_loss": 0.6122824600340679
    },
    {
      "train_step": 6239,
      "sup_loss": 0.5981188255401023
    },
    {
      "train_step": 6240,
      "sup_loss": 0.6463259905253445
    },
    {
      "train_step": 6241,
      "sup_loss": 0.5581165128524769
    },
    {
      "train_step": 6242,
      "sup_loss": 0.3421550975797515
    },
    {
      "train_step": 6243,
      "sup_loss": 0.5855398606503925
    },
    {
      "train_step": 6244,
      "sup_loss": 0.2734943721306753
    },
    {
      "train_step": 6245,
      "sup_loss": 0.5214580089595757
    },
    {
      "train_step": 6246,
      "sup_loss": 0.7266165245758782
    },
    {
      "train_step": 6247,
      "sup_loss": 0.7260517824375254
    },
    {
      "train_step": 6248,
      "sup_loss": 0.684753166150391
    },
    {
      "train_step": 6249,
      "sup_loss": 0.5162743932231297
    },
    {
      "train_step": 6250,
      "sup_loss": 0.6372841955891558
    },
    {
      "train_step": 6251,
      "sup_loss": 0.5371903511236527
    },
    {
      "train_step": 6252,
      "sup_loss": 0.6265108281269371
    },
    {
      "train_step": 6253,
      "sup_loss": 0.77364827371843
    },
    {
      "train_step": 6254,
      "sup_loss": 0.4365161357051444
    },
    {
      "train_step": 6255,
      "sup_loss": 0.5717272172301415
    },
    {
      "train_step": 6256,
      "sup_loss": 0.2988813161752904
    },
    {
      "train_step": 6257,
      "sup_loss": 0.5466544606600109
    },
    {
      "train_step": 6258,
      "sup_loss": 0.6569918692175013
    },
    {
      "train_step": 6259,
      "sup_loss": 0.6526786847137427
    },
    {
      "train_step": 6260,
      "sup_loss": 0.4357979492877544
    },
    {
      "train_step": 6261,
      "sup_loss": 0.6658451697634901
    },
    {
      "train_step": 6262,
      "sup_loss": 0.2509970018868303
    },
    {
      "train_step": 6263,
      "sup_loss": 0.4105170141035407
    },
    {
      "train_step": 6264,
      "sup_loss": 0.5258220405733979
    },
    {
      "train_step": 6265,
      "sup_loss": 0.5926293189017049
    },
    {
      "train_step": 6266,
      "sup_loss": 0.5860886229522403
    },
    {
      "train_step": 6267,
      "sup_loss": 0.753556031660289
    },
    {
      "train_step": 6268,
      "sup_loss": 0.6556668543744623
    },
    {
      "train_step": 6269,
      "sup_loss": 0.5864178563453939
    },
    {
      "train_step": 6270,
      "sup_loss": 0.7045871929084586
    },
    {
      "train_step": 6271,
      "sup_loss": 0.4330900945790692
    },
    {
      "train_step": 6272,
      "sup_loss": 0.44364284146370964
    },
    {
      "train_step": 6273,
      "sup_loss": 0.3675743375713884
    },
    {
      "train_step": 6274,
      "sup_loss": 0.5683207233319709
    },
    {
      "train_step": 6275,
      "sup_loss": 0.6530119048317501
    },
    {
      "train_step": 6276,
      "sup_loss": 0.49160839484417707
    },
    {
      "train_step": 6277,
      "sup_loss": 0.808398541396313
    },
    {
      "train_step": 6278,
      "sup_loss": 0.3371873499207291
    },
    {
      "train_step": 6279,
      "sup_loss": 0.44412029472470393
    },
    {
      "train_step": 6280,
      "sup_loss": 0.45997906184173687
    },
    {
      "train_step": 6281,
      "sup_loss": 0.7172119388823931
    },
    {
      "train_step": 6282,
      "sup_loss": 0.4199051227405742
    },
    {
      "train_step": 6283,
      "sup_loss": 0.5290099199149817
    },
    {
      "train_step": 6284,
      "sup_loss": 0.4173985966932492
    },
    {
      "train_step": 6285,
      "sup_loss": 0.7375470686434303
    },
    {
      "train_step": 6286,
      "sup_loss": 0.5476774636929661
    },
    {
      "train_step": 6287,
      "sup_loss": 0.29608414177292947
    },
    {
      "train_step": 6288,
      "sup_loss": 0.6716340792361328
    },
    {
      "train_step": 6289,
      "sup_loss": 0.59101542894755
    },
    {
      "train_step": 6290,
      "sup_loss": 0.4849352525741908
    },
    {
      "train_step": 6291,
      "sup_loss": 0.5442294499754906
    },
    {
      "train_step": 6292,
      "sup_loss": 0.5102732564595489
    },
    {
      "train_step": 6293,
      "sup_loss": 0.7694224168192818
    },
    {
      "train_step": 6294,
      "sup_loss": 0.48663572432589824
    },
    {
      "train_step": 6295,
      "sup_loss": 0.8484793193013691
    },
    {
      "train_step": 6296,
      "sup_loss": 0.4451790212695699
    },
    {
      "train_step": 6297,
      "sup_loss": 0.8268048698641266
    },
    {
      "train_step": 6298,
      "sup_loss": 0.6706852451170329
    },
    {
      "train_step": 6299,
      "sup_loss": 0.353300964005126
    },
    {
      "train_step": 6300,
      "sup_loss": 0.4151113664212803
    },
    {
      "train_step": 6301,
      "sup_loss": 0.5539317142421768
    },
    {
      "train_step": 6302,
      "sup_loss": 0.37877890634321815
    },
    {
      "train_step": 6303,
      "sup_loss": 0.4270078695078684
    },
    {
      "train_step": 6304,
      "sup_loss": 0.41033559788805113
    },
    {
      "train_step": 6305,
      "sup_loss": 0.6752004285941615
    },
    {
      "train_step": 6306,
      "sup_loss": 0.689804320108659
    },
    {
      "train_step": 6307,
      "sup_loss": 0.5008020756183515
    },
    {
      "train_step": 6308,
      "sup_loss": 0.6249313647687461
    },
    {
      "train_step": 6309,
      "sup_loss": 0.5292331124212338
    },
    {
      "train_step": 6310,
      "sup_loss": 0.6425616079816119
    },
    {
      "train_step": 6311,
      "sup_loss": 0.5780300744270944
    },
    {
      "train_step": 6312,
      "sup_loss": 0.7386086553821689
    },
    {
      "train_step": 6313,
      "sup_loss": 0.6217070618716226
    },
    {
      "train_step": 6314,
      "sup_loss": 0.7568018935066088
    },
    {
      "train_step": 6315,
      "sup_loss": 0.5475646998663656
    },
    {
      "train_step": 6316,
      "sup_loss": 0.5780185018916274
    },
    {
      "train_step": 6317,
      "sup_loss": 0.7717726021024707
    },
    {
      "train_step": 6318,
      "sup_loss": 0.5012507931444525
    },
    {
      "train_step": 6319,
      "sup_loss": 0.5032801780745735
    },
    {
      "train_step": 6320,
      "sup_loss": 0.6123281284530776
    },
    {
      "train_step": 6321,
      "sup_loss": 0.34930412327938126
    },
    {
      "train_step": 6322,
      "sup_loss": 0.7076113865243512
    },
    {
      "train_step": 6323,
      "sup_loss": 0.4611452442314912
    },
    {
      "train_step": 6324,
      "sup_loss": 0.3981403034522312
    },
    {
      "train_step": 6325,
      "sup_loss": 0.574874210181354
    },
    {
      "train_step": 6326,
      "sup_loss": 0.8536859889302878
    },
    {
      "train_step": 6327,
      "sup_loss": 0.6533719381167891
    },
    {
      "train_step": 6328,
      "sup_loss": 0.6034872711650572
    },
    {
      "train_step": 6329,
      "sup_loss": 0.6734407647070878
    },
    {
      "train_step": 6330,
      "sup_loss": 0.7833073731595487
    },
    {
      "train_step": 6331,
      "sup_loss": 0.6299720892906364
    },
    {
      "train_step": 6332,
      "sup_loss": 0.4379628864127065
    },
    {
      "train_step": 6333,
      "sup_loss": 0.7538077961099956
    },
    {
      "train_step": 6334,
      "sup_loss": 0.6352940990222479
    },
    {
      "train_step": 6335,
      "sup_loss": 0.6673527906865974
    },
    {
      "train_step": 6336,
      "sup_loss": 0.4124876110001987
    },
    {
      "train_step": 6337,
      "sup_loss": 0.7325290190206595
    },
    {
      "train_step": 6338,
      "sup_loss": 0.5483041435415732
    },
    {
      "train_step": 6339,
      "sup_loss": 0.646862880795273
    },
    {
      "train_step": 6340,
      "sup_loss": 0.5517811315254859
    },
    {
      "train_step": 6341,
      "sup_loss": 0.7482939598868266
    },
    {
      "train_step": 6342,
      "sup_loss": 0.4337001912151104
    },
    {
      "train_step": 6343,
      "sup_loss": 0.6164543227525031
    },
    {
      "train_step": 6344,
      "sup_loss": 0.5255689747406448
    },
    {
      "train_step": 6345,
      "sup_loss": 0.7949129524580184
    },
    {
      "train_step": 6346,
      "sup_loss": 0.642272624482574
    },
    {
      "train_step": 6347,
      "sup_loss": 0.506658756083346
    },
    {
      "train_step": 6348,
      "sup_loss": 0.5241598460781902
    },
    {
      "train_step": 6349,
      "sup_loss": 0.48731301296623397
    },
    {
      "train_step": 6350,
      "sup_loss": 0.6860752641599958
    },
    {
      "train_step": 6351,
      "sup_loss": 0.7121034956588769
    },
    {
      "train_step": 6352,
      "sup_loss": 0.5364404398565683
    },
    {
      "train_step": 6353,
      "sup_loss": 0.40648472879537634
    },
    {
      "train_step": 6354,
      "sup_loss": 0.5191011780011164
    },
    {
      "train_step": 6355,
      "sup_loss": 1.0373592839176047
    },
    {
      "train_step": 6356,
      "sup_loss": 0.6127227876081576
    },
    {
      "train_step": 6357,
      "sup_loss": 0.6694513722799876
    },
    {
      "train_step": 6358,
      "sup_loss": 0.5110950930865152
    },
    {
      "train_step": 6359,
      "sup_loss": 0.30831793208934066
    },
    {
      "train_step": 6360,
      "sup_loss": 0.5729545666153333
    },
    {
      "train_step": 6361,
      "sup_loss": 0.642397702144674
    },
    {
      "train_step": 6362,
      "sup_loss": 0.41643943671378875
    },
    {
      "train_step": 6363,
      "sup_loss": 0.5712757660351825
    },
    {
      "train_step": 6364,
      "sup_loss": 0.8459085745341011
    },
    {
      "train_step": 6365,
      "sup_loss": 0.5569377055874638
    },
    {
      "train_step": 6366,
      "sup_loss": 0.46630236667941455
    },
    {
      "train_step": 6367,
      "sup_loss": 0.48394377308624525
    },
    {
      "train_step": 6368,
      "sup_loss": 0.5117900641165641
    },
    {
      "train_step": 6369,
      "sup_loss": 0.5797040997701784
    },
    {
      "train_step": 6370,
      "sup_loss": 0.5478889997855988
    },
    {
      "train_step": 6371,
      "sup_loss": 0.5400454361631868
    },
    {
      "train_step": 6372,
      "sup_loss": 0.516260625980461
    },
    {
      "train_step": 6373,
      "sup_loss": 0.4024266640177424
    },
    {
      "train_step": 6374,
      "sup_loss": 0.5536735350329697
    },
    {
      "train_step": 6375,
      "sup_loss": 0.4734930913372212
    },
    {
      "train_step": 6376,
      "sup_loss": 0.3942592391891165
    },
    {
      "train_step": 6377,
      "sup_loss": 0.5724562795401068
    },
    {
      "train_step": 6378,
      "sup_loss": 0.3463451790312318
    },
    {
      "train_step": 6379,
      "sup_loss": 0.6352368731418666
    },
    {
      "train_step": 6380,
      "sup_loss": 0.5926901609828825
    },
    {
      "train_step": 6381,
      "sup_loss": 0.6849386697765365
    },
    {
      "train_step": 6382,
      "sup_loss": 0.6706282879288565
    },
    {
      "train_step": 6383,
      "sup_loss": 0.5007130028427382
    },
    {
      "train_step": 6384,
      "sup_loss": 0.3879099900749409
    },
    {
      "train_step": 6385,
      "sup_loss": 0.7200749999937293
    },
    {
      "train_step": 6386,
      "sup_loss": 0.6116998611775066
    },
    {
      "train_step": 6387,
      "sup_loss": 0.5715247374702854
    },
    {
      "train_step": 6388,
      "sup_loss": 0.6210606258055398
    },
    {
      "train_step": 6389,
      "sup_loss": 0.6870218470172534
    },
    {
      "train_step": 6390,
      "sup_loss": 0.6584062067628959
    },
    {
      "train_step": 6391,
      "sup_loss": 0.7096530366759666
    },
    {
      "train_step": 6392,
      "sup_loss": 0.393852726737842
    },
    {
      "train_step": 6393,
      "sup_loss": 0.3524355578182665
    },
    {
      "train_step": 6394,
      "sup_loss": 0.4513261251339687
    },
    {
      "train_step": 6395,
      "sup_loss": 0.4813421682710174
    },
    {
      "train_step": 6396,
      "sup_loss": 0.7127220301849422
    },
    {
      "train_step": 6397,
      "sup_loss": 0.47443402145713265
    },
    {
      "train_step": 6398,
      "sup_loss": 0.4905592944022601
    },
    {
      "train_step": 6399,
      "sup_loss": 0.666102044094725
    },
    {
      "train_step": 6400,
      "sup_loss": 0.5920636436444331
    },
    {
      "train_step": 6401,
      "sup_loss": 0.5930164109313214
    },
    {
      "train_step": 6402,
      "sup_loss": 0.5794063544436443
    },
    {
      "train_step": 6403,
      "sup_loss": 0.5504409335381786
    },
    {
      "train_step": 6404,
      "sup_loss": 0.7433298982937717
    },
    {
      "train_step": 6405,
      "sup_loss": 0.5488170974559714
    },
    {
      "train_step": 6406,
      "sup_loss": 0.5372455313705351
    },
    {
      "train_step": 6407,
      "sup_loss": 0.6460067549824264
    },
    {
      "train_step": 6408,
      "sup_loss": 0.635780943440313
    },
    {
      "train_step": 6409,
      "sup_loss": 0.7514370020400247
    },
    {
      "train_step": 6410,
      "sup_loss": 0.4257096518486415
    },
    {
      "train_step": 6411,
      "sup_loss": 0.5377411430675931
    },
    {
      "train_step": 6412,
      "sup_loss": 0.572495070901818
    },
    {
      "train_step": 6413,
      "sup_loss": 0.6817239092429721
    },
    {
      "train_step": 6414,
      "sup_loss": 0.6059629346329352
    },
    {
      "train_step": 6415,
      "sup_loss": 0.44156787577647655
    },
    {
      "train_step": 6416,
      "sup_loss": 0.6570748431681721
    },
    {
      "train_step": 6417,
      "sup_loss": 0.766056200952623
    },
    {
      "train_step": 6418,
      "sup_loss": 0.7641237795362407
    },
    {
      "train_step": 6419,
      "sup_loss": 0.5675348111350076
    },
    {
      "train_step": 6420,
      "sup_loss": 0.6727379572914383
    },
    {
      "train_step": 6421,
      "sup_loss": 0.3527155283429894
    },
    {
      "train_step": 6422,
      "sup_loss": 0.6922179710065056
    },
    {
      "train_step": 6423,
      "sup_loss": 0.15696284465180851
    },
    {
      "train_step": 6424,
      "sup_loss": 0.45348679982929685
    },
    {
      "train_step": 6425,
      "sup_loss": 0.30229044724239024
    },
    {
      "train_step": 6426,
      "sup_loss": 0.6158976196864442
    },
    {
      "train_step": 6427,
      "sup_loss": 0.6265941386662006
    },
    {
      "train_step": 6428,
      "sup_loss": 0.300768853050892
    },
    {
      "train_step": 6429,
      "sup_loss": 0.7051647276963452
    },
    {
      "train_step": 6430,
      "sup_loss": 0.5901314385621278
    },
    {
      "train_step": 6431,
      "sup_loss": 0.46855844929673485
    },
    {
      "train_step": 6432,
      "sup_loss": 0.4865086911382395
    },
    {
      "train_step": 6433,
      "sup_loss": 0.6476447019926306
    },
    {
      "train_step": 6434,
      "sup_loss": 0.42362638027373467
    },
    {
      "train_step": 6435,
      "sup_loss": 0.6612213903591565
    },
    {
      "train_step": 6436,
      "sup_loss": 0.635495146551723
    },
    {
      "train_step": 6437,
      "sup_loss": 0.7078319878564915
    },
    {
      "train_step": 6438,
      "sup_loss": 0.4855553066249486
    },
    {
      "train_step": 6439,
      "sup_loss": 0.774091946632696
    },
    {
      "train_step": 6440,
      "sup_loss": 0.6194022218858177
    },
    {
      "train_step": 6441,
      "sup_loss": 0.4411178575087776
    },
    {
      "train_step": 6442,
      "sup_loss": 0.6455368075116577
    },
    {
      "train_step": 6443,
      "sup_loss": 0.49803597151162343
    },
    {
      "train_step": 6444,
      "sup_loss": 0.3280669553567238
    },
    {
      "train_step": 6445,
      "sup_loss": 0.7343316879099008
    },
    {
      "train_step": 6446,
      "sup_loss": 0.5639734956617151
    },
    {
      "train_step": 6447,
      "sup_loss": 0.62204403604898
    },
    {
      "train_step": 6448,
      "sup_loss": 0.4746096350646061
    },
    {
      "train_step": 6449,
      "sup_loss": 0.4191687173130505
    },
    {
      "train_step": 6450,
      "sup_loss": 0.6180268280676311
    },
    {
      "train_step": 6451,
      "sup_loss": 0.6876786927179388
    },
    {
      "train_step": 6452,
      "sup_loss": 0.6089681025028767
    },
    {
      "train_step": 6453,
      "sup_loss": 0.4871840472870621
    },
    {
      "train_step": 6454,
      "sup_loss": 0.4182756808391438
    },
    {
      "train_step": 6455,
      "sup_loss": 0.5727469168400336
    },
    {
      "train_step": 6456,
      "sup_loss": 0.7231842944558695
    },
    {
      "train_step": 6457,
      "sup_loss": 0.6029795862441366
    },
    {
      "train_step": 6458,
      "sup_loss": 0.6412199230957212
    },
    {
      "train_step": 6459,
      "sup_loss": 0.31996410799608416
    },
    {
      "train_step": 6460,
      "sup_loss": 0.6231501170337789
    },
    {
      "train_step": 6461,
      "sup_loss": 0.38732213853687697
    },
    {
      "train_step": 6462,
      "sup_loss": 0.5711802955286926
    },
    {
      "train_step": 6463,
      "sup_loss": 0.4653613661181287
    },
    {
      "train_step": 6464,
      "sup_loss": 0.6438269717831906
    },
    {
      "train_step": 6465,
      "sup_loss": 0.3654108696302071
    },
    {
      "train_step": 6466,
      "sup_loss": 0.8407621399442045
    },
    {
      "train_step": 6467,
      "sup_loss": 0.7516122472259168
    },
    {
      "train_step": 6468,
      "sup_loss": 0.5766651455950562
    },
    {
      "train_step": 6469,
      "sup_loss": 0.5211884571589579
    },
    {
      "train_step": 6470,
      "sup_loss": 0.49083316758620094
    },
    {
      "train_step": 6471,
      "sup_loss": 0.4071931350752637
    },
    {
      "train_step": 6472,
      "sup_loss": 0.5655922611324121
    },
    {
      "train_step": 6473,
      "sup_loss": 0.589551407041895
    },
    {
      "train_step": 6474,
      "sup_loss": 0.7486112915913289
    },
    {
      "train_step": 6475,
      "sup_loss": 0.5508220476029966
    },
    {
      "train_step": 6476,
      "sup_loss": 0.47783302361068003
    },
    {
      "train_step": 6477,
      "sup_loss": 0.20858412850173194
    },
    {
      "train_step": 6478,
      "sup_loss": 0.3514444217110625
    },
    {
      "train_step": 6479,
      "sup_loss": 0.5975450559114219
    },
    {
      "train_step": 6480,
      "sup_loss": 0.3607242083105536
    },
    {
      "train_step": 6481,
      "sup_loss": 0.45175159259931014
    },
    {
      "train_step": 6482,
      "sup_loss": 0.3888868608240111
    },
    {
      "train_step": 6483,
      "sup_loss": 0.38152616430874403
    },
    {
      "train_step": 6484,
      "sup_loss": 0.3432227491173236
    },
    {
      "train_step": 6485,
      "sup_loss": 0.611033580159188
    },
    {
      "train_step": 6486,
      "sup_loss": 0.5019358260217257
    },
    {
      "train_step": 6487,
      "sup_loss": 0.581406876116554
    },
    {
      "train_step": 6488,
      "sup_loss": 0.5162631534543546
    },
    {
      "train_step": 6489,
      "sup_loss": 0.4089030044164436
    },
    {
      "train_step": 6490,
      "sup_loss": 0.6739030753770806
    },
    {
      "train_step": 6491,
      "sup_loss": 0.6174478885532759
    },
    {
      "train_step": 6492,
      "sup_loss": 0.6597941083013449
    },
    {
      "train_step": 6493,
      "sup_loss": 0.565796095930038
    },
    {
      "train_step": 6494,
      "sup_loss": 0.5711799263382449
    },
    {
      "train_step": 6495,
      "sup_loss": 0.7268363605556281
    },
    {
      "train_step": 6496,
      "sup_loss": 0.7786248486333978
    },
    {
      "train_step": 6497,
      "sup_loss": 0.45992446318325597
    },
    {
      "train_step": 6498,
      "sup_loss": 0.5817945636672125
    },
    {
      "train_step": 6499,
      "sup_loss": 0.46647402112792186
    },
    {
      "train_step": 6500,
      "sup_loss": 0.6680507433246045
    },
    {
      "train_step": 6501,
      "sup_loss": 0.6190888130680204
    },
    {
      "train_step": 6502,
      "sup_loss": 0.5136257834712225
    },
    {
      "train_step": 6503,
      "sup_loss": 0.46736906105503845
    },
    {
      "train_step": 6504,
      "sup_loss": 0.454724295945636
    },
    {
      "train_step": 6505,
      "sup_loss": 0.7981995006444162
    },
    {
      "train_step": 6506,
      "sup_loss": 0.6090818316838145
    },
    {
      "train_step": 6507,
      "sup_loss": 0.5322592295234273
    },
    {
      "train_step": 6508,
      "sup_loss": 0.30200202280438343
    },
    {
      "train_step": 6509,
      "sup_loss": 0.4836850600194098
    },
    {
      "train_step": 6510,
      "sup_loss": 0.39394559243755817
    },
    {
      "train_step": 6511,
      "sup_loss": 0.6035728022851875
    },
    {
      "train_step": 6512,
      "sup_loss": 0.6920140619677608
    },
    {
      "train_step": 6513,
      "sup_loss": 0.5247771603173179
    },
    {
      "train_step": 6514,
      "sup_loss": 0.4233094491949964
    },
    {
      "train_step": 6515,
      "sup_loss": 0.7576081111555044
    },
    {
      "train_step": 6516,
      "sup_loss": 0.6502386676968577
    },
    {
      "train_step": 6517,
      "sup_loss": 0.5567359468929286
    },
    {
      "train_step": 6518,
      "sup_loss": 0.5117216211518397
    },
    {
      "train_step": 6519,
      "sup_loss": 0.4981287698931042
    },
    {
      "train_step": 6520,
      "sup_loss": 0.7099338935186287
    },
    {
      "train_step": 6521,
      "sup_loss": 0.44167140898415896
    },
    {
      "train_step": 6522,
      "sup_loss": 0.7260733545089966
    },
    {
      "train_step": 6523,
      "sup_loss": 0.31419463498953326
    },
    {
      "train_step": 6524,
      "sup_loss": 0.7271309598098192
    },
    {
      "train_step": 6525,
      "sup_loss": 0.6605125676996112
    },
    {
      "train_step": 6526,
      "sup_loss": 0.6571548122626867
    },
    {
      "train_step": 6527,
      "sup_loss": 0.49628031770601494
    },
    {
      "train_step": 6528,
      "sup_loss": 0.5577517280095605
    },
    {
      "train_step": 6529,
      "sup_loss": 0.7972071829020467
    },
    {
      "train_step": 6530,
      "sup_loss": 0.5342654190818349
    },
    {
      "train_step": 6531,
      "sup_loss": 0.4546286309226658
    },
    {
      "train_step": 6532,
      "sup_loss": 0.6437370959375357
    },
    {
      "train_step": 6533,
      "sup_loss": 0.7486629353231509
    },
    {
      "train_step": 6534,
      "sup_loss": 0.5254208570209616
    },
    {
      "train_step": 6535,
      "sup_loss": 0.6596692206164029
    },
    {
      "train_step": 6536,
      "sup_loss": 0.6530923770742174
    },
    {
      "train_step": 6537,
      "sup_loss": 0.510118184645427
    },
    {
      "train_step": 6538,
      "sup_loss": 0.4015017227381503
    },
    {
      "train_step": 6539,
      "sup_loss": 0.45712627815643925
    },
    {
      "train_step": 6540,
      "sup_loss": 0.6248079635855697
    },
    {
      "train_step": 6541,
      "sup_loss": 0.35969454841626675
    },
    {
      "train_step": 6542,
      "sup_loss": 0.6161262364481067
    },
    {
      "train_step": 6543,
      "sup_loss": 0.5792864746782102
    },
    {
      "train_step": 6544,
      "sup_loss": 0.5707175639802611
    },
    {
      "train_step": 6545,
      "sup_loss": 0.46176064940237455
    },
    {
      "train_step": 6546,
      "sup_loss": 0.6538411314010067
    },
    {
      "train_step": 6547,
      "sup_loss": 0.6098426477854811
    },
    {
      "train_step": 6548,
      "sup_loss": 0.6191665236471575
    },
    {
      "train_step": 6549,
      "sup_loss": 0.8165319880926449
    },
    {
      "train_step": 6550,
      "sup_loss": 0.5572340140533643
    },
    {
      "train_step": 6551,
      "sup_loss": 0.7143221923127472
    },
    {
      "train_step": 6552,
      "sup_loss": 0.3115080380814309
    },
    {
      "train_step": 6553,
      "sup_loss": 0.4071116484330558
    },
    {
      "train_step": 6554,
      "sup_loss": 0.7698942720253964
    },
    {
      "train_step": 6555,
      "sup_loss": 0.6560538915677095
    },
    {
      "train_step": 6556,
      "sup_loss": 0.429157288269503
    },
    {
      "train_step": 6557,
      "sup_loss": 0.45238028712777434
    },
    {
      "train_step": 6558,
      "sup_loss": 0.5594584246666914
    },
    {
      "train_step": 6559,
      "sup_loss": 0.7154585747966679
    },
    {
      "train_step": 6560,
      "sup_loss": 0.6667070047779905
    },
    {
      "train_step": 6561,
      "sup_loss": 0.5158097070300176
    },
    {
      "train_step": 6562,
      "sup_loss": 0.3807972875627942
    },
    {
      "train_step": 6563,
      "sup_loss": 0.5988831842946118
    },
    {
      "train_step": 6564,
      "sup_loss": 0.6499244053874038
    },
    {
      "train_step": 6565,
      "sup_loss": 0.7179582404540168
    },
    {
      "train_step": 6566,
      "sup_loss": 0.7123006845952811
    },
    {
      "train_step": 6567,
      "sup_loss": 0.5819359115514402
    },
    {
      "train_step": 6568,
      "sup_loss": 0.33180585198726337
    },
    {
      "train_step": 6569,
      "sup_loss": 0.5003783913259353
    },
    {
      "train_step": 6570,
      "sup_loss": 0.5070915334220816
    },
    {
      "train_step": 6571,
      "sup_loss": 0.5836713585005623
    },
    {
      "train_step": 6572,
      "sup_loss": 0.7546649920437348
    },
    {
      "train_step": 6573,
      "sup_loss": 0.6313172803816267
    },
    {
      "train_step": 6574,
      "sup_loss": 0.5826328709012107
    },
    {
      "train_step": 6575,
      "sup_loss": 0.4965691726794863
    },
    {
      "train_step": 6576,
      "sup_loss": 0.3056686250077014
    },
    {
      "train_step": 6577,
      "sup_loss": 0.7954825219309534
    },
    {
      "train_step": 6578,
      "sup_loss": 0.4990616749232041
    },
    {
      "train_step": 6579,
      "sup_loss": 0.4324767698405228
    },
    {
      "train_step": 6580,
      "sup_loss": 0.6697368384010228
    },
    {
      "train_step": 6581,
      "sup_loss": 0.4791425601820103
    },
    {
      "train_step": 6582,
      "sup_loss": 0.5472066308528964
    },
    {
      "train_step": 6583,
      "sup_loss": 0.6822079650941341
    },
    {
      "train_step": 6584,
      "sup_loss": 0.635823661108534
    },
    {
      "train_step": 6585,
      "sup_loss": 0.6970182221605529
    },
    {
      "train_step": 6586,
      "sup_loss": 0.5317304489140176
    },
    {
      "train_step": 6587,
      "sup_loss": 0.44511196345831444
    },
    {
      "train_step": 6588,
      "sup_loss": 0.8968838473903441
    },
    {
      "train_step": 6589,
      "sup_loss": 0.5570100765682977
    },
    {
      "train_step": 6590,
      "sup_loss": 0.540125984910723
    },
    {
      "train_step": 6591,
      "sup_loss": 0.6137053974802371
    },
    {
      "train_step": 6592,
      "sup_loss": 0.5459370892275301
    },
    {
      "train_step": 6593,
      "sup_loss": 0.42091486379697896
    },
    {
      "train_step": 6594,
      "sup_loss": 0.47010161270902445
    },
    {
      "train_step": 6595,
      "sup_loss": 0.6192469793783058
    },
    {
      "train_step": 6596,
      "sup_loss": 0.5676550233968135
    },
    {
      "train_step": 6597,
      "sup_loss": 0.7579533971780208
    },
    {
      "train_step": 6598,
      "sup_loss": 0.8182980041780294
    },
    {
      "train_step": 6599,
      "sup_loss": 0.636127868155197
    },
    {
      "train_step": 6600,
      "sup_loss": 0.5792247704174804
    },
    {
      "train_step": 6601,
      "sup_loss": 0.3773670808828411
    },
    {
      "train_step": 6602,
      "sup_loss": 0.5615406651479989
    },
    {
      "train_step": 6603,
      "sup_loss": 0.5379878124229478
    },
    {
      "train_step": 6604,
      "sup_loss": 0.7158524008810955
    },
    {
      "train_step": 6605,
      "sup_loss": 0.4899937867684547
    },
    {
      "train_step": 6606,
      "sup_loss": 0.3681243964982285
    },
    {
      "train_step": 6607,
      "sup_loss": 0.5817376359099663
    },
    {
      "train_step": 6608,
      "sup_loss": 0.774129414343843
    },
    {
      "train_step": 6609,
      "sup_loss": 0.4233113655320592
    },
    {
      "train_step": 6610,
      "sup_loss": 0.5771746446832453
    },
    {
      "train_step": 6611,
      "sup_loss": 0.7369879032987863
    },
    {
      "train_step": 6612,
      "sup_loss": 0.6379523980653382
    },
    {
      "train_step": 6613,
      "sup_loss": 0.5174053134510337
    },
    {
      "train_step": 6614,
      "sup_loss": 0.5336713714584596
    },
    {
      "train_step": 6615,
      "sup_loss": 0.5280675022244774
    },
    {
      "train_step": 6616,
      "sup_loss": 0.4890018270909745
    },
    {
      "train_step": 6617,
      "sup_loss": 0.38895408478158144
    },
    {
      "train_step": 6618,
      "sup_loss": 0.4730085656069719
    },
    {
      "train_step": 6619,
      "sup_loss": 0.6943359705217941
    },
    {
      "train_step": 6620,
      "sup_loss": 0.3367836838804548
    },
    {
      "train_step": 6621,
      "sup_loss": 0.40262000445679735
    },
    {
      "train_step": 6622,
      "sup_loss": 0.6401016489599469
    },
    {
      "train_step": 6623,
      "sup_loss": 0.7122005280663168
    },
    {
      "train_step": 6624,
      "sup_loss": 0.6707868923559713
    },
    {
      "train_step": 6625,
      "sup_loss": 0.7287166200827773
    },
    {
      "train_step": 6626,
      "sup_loss": 0.6816560080272653
    },
    {
      "train_step": 6627,
      "sup_loss": 0.5050001374042107
    },
    {
      "train_step": 6628,
      "sup_loss": 0.5787718404580264
    },
    {
      "train_step": 6629,
      "sup_loss": 0.777834456339689
    },
    {
      "train_step": 6630,
      "sup_loss": 0.746164103332374
    },
    {
      "train_step": 6631,
      "sup_loss": 0.5497612167256254
    },
    {
      "train_step": 6632,
      "sup_loss": 0.5102053737748959
    },
    {
      "train_step": 6633,
      "sup_loss": 0.34942278872700117
    },
    {
      "train_step": 6634,
      "sup_loss": 0.62235650330859
    },
    {
      "train_step": 6635,
      "sup_loss": 0.432288063291791
    },
    {
      "train_step": 6636,
      "sup_loss": 0.46403970281565293
    },
    {
      "train_step": 6637,
      "sup_loss": 0.37709023161519517
    },
    {
      "train_step": 6638,
      "sup_loss": 0.6292004356522772
    },
    {
      "train_step": 6639,
      "sup_loss": 0.72104529912257
    },
    {
      "train_step": 6640,
      "sup_loss": 0.4343165767606882
    },
    {
      "train_step": 6641,
      "sup_loss": 0.5956724950092684
    },
    {
      "train_step": 6642,
      "sup_loss": 0.31002773115758414
    },
    {
      "train_step": 6643,
      "sup_loss": 0.4181076586361327
    },
    {
      "train_step": 6644,
      "sup_loss": 0.4871930395464634
    },
    {
      "train_step": 6645,
      "sup_loss": 0.5494983863738453
    },
    {
      "train_step": 6646,
      "sup_loss": 0.5958235108445351
    },
    {
      "train_step": 6647,
      "sup_loss": 0.5379915081518372
    },
    {
      "train_step": 6648,
      "sup_loss": 0.6033815056393584
    },
    {
      "train_step": 6649,
      "sup_loss": 0.5554651797751448
    },
    {
      "train_step": 6650,
      "sup_loss": 0.3989400900486964
    },
    {
      "train_step": 6651,
      "sup_loss": 0.47473822325713116
    },
    {
      "train_step": 6652,
      "sup_loss": 0.6102245759412507
    },
    {
      "train_step": 6653,
      "sup_loss": 0.506833163330257
    },
    {
      "train_step": 6654,
      "sup_loss": 0.613201012951579
    },
    {
      "train_step": 6655,
      "sup_loss": 0.7088464619239481
    },
    {
      "train_step": 6656,
      "sup_loss": 0.5994708689851099
    },
    {
      "train_step": 6657,
      "sup_loss": 0.2702865228447731
    },
    {
      "train_step": 6658,
      "sup_loss": 0.6158614359374006
    },
    {
      "train_step": 6659,
      "sup_loss": 0.6772767544009664
    },
    {
      "train_step": 6660,
      "sup_loss": 0.45851108037536004
    },
    {
      "train_step": 6661,
      "sup_loss": 0.27956261009201827
    },
    {
      "train_step": 6662,
      "sup_loss": 0.6066756686203382
    },
    {
      "train_step": 6663,
      "sup_loss": 0.5455427139695325
    },
    {
      "train_step": 6664,
      "sup_loss": 0.39800573352197555
    },
    {
      "train_step": 6665,
      "sup_loss": 0.759681588889084
    },
    {
      "train_step": 6666,
      "sup_loss": 0.36490877056178256
    },
    {
      "train_step": 6667,
      "sup_loss": 0.7779528350255018
    },
    {
      "train_step": 6668,
      "sup_loss": 0.4976535850042123
    },
    {
      "train_step": 6669,
      "sup_loss": 0.6346028408358559
    },
    {
      "train_step": 6670,
      "sup_loss": 0.484328439924254
    },
    {
      "train_step": 6671,
      "sup_loss": 0.42790736965118975
    },
    {
      "train_step": 6672,
      "sup_loss": 0.6966749956960574
    },
    {
      "train_step": 6673,
      "sup_loss": 0.7518411154214213
    },
    {
      "train_step": 6674,
      "sup_loss": 0.6342262212557997
    },
    {
      "train_step": 6675,
      "sup_loss": 0.8366056359499207
    },
    {
      "train_step": 6676,
      "sup_loss": 0.47534973662280827
    },
    {
      "train_step": 6677,
      "sup_loss": 0.5572701113257831
    },
    {
      "train_step": 6678,
      "sup_loss": 0.5616151366130389
    },
    {
      "train_step": 6679,
      "sup_loss": 0.5671284742551383
    },
    {
      "train_step": 6680,
      "sup_loss": 0.40392798827912924
    },
    {
      "train_step": 6681,
      "sup_loss": 0.4884520433089679
    },
    {
      "train_step": 6682,
      "sup_loss": 0.6912935569564428
    },
    {
      "train_step": 6683,
      "sup_loss": 0.6849462392261384
    },
    {
      "train_step": 6684,
      "sup_loss": 0.6002748023856943
    },
    {
      "train_step": 6685,
      "sup_loss": 0.47442638339743504
    },
    {
      "train_step": 6686,
      "sup_loss": 0.6501911765857075
    },
    {
      "train_step": 6687,
      "sup_loss": 0.42458069147075983
    },
    {
      "train_step": 6688,
      "sup_loss": 0.49145073948009005
    },
    {
      "train_step": 6689,
      "sup_loss": 0.5883626915820265
    },
    {
      "train_step": 6690,
      "sup_loss": 0.2532760389980728
    },
    {
      "train_step": 6691,
      "sup_loss": 0.4133386682157324
    },
    {
      "train_step": 6692,
      "sup_loss": 0.473249813223429
    },
    {
      "train_step": 6693,
      "sup_loss": 0.4973976505946609
    },
    {
      "train_step": 6694,
      "sup_loss": 0.461084142171042
    },
    {
      "train_step": 6695,
      "sup_loss": 0.608335941580626
    },
    {
      "train_step": 6696,
      "sup_loss": 0.5194168136725431
    },
    {
      "train_step": 6697,
      "sup_loss": 0.5009885516253609
    },
    {
      "train_step": 6698,
      "sup_loss": 0.5522208630199025
    },
    {
      "train_step": 6699,
      "sup_loss": 0.6180576303964783
    },
    {
      "train_step": 6700,
      "sup_loss": 0.5703854601836568
    },
    {
      "train_step": 6701,
      "sup_loss": 0.5127210024919858
    },
    {
      "train_step": 6702,
      "sup_loss": 0.38413174544034917
    },
    {
      "train_step": 6703,
      "sup_loss": 0.46054261785948686
    },
    {
      "train_step": 6704,
      "sup_loss": 0.6968662202977705
    },
    {
      "train_step": 6705,
      "sup_loss": 0.7241037584335009
    },
    {
      "train_step": 6706,
      "sup_loss": 0.5770499950278885
    },
    {
      "train_step": 6707,
      "sup_loss": 0.4966085652013622
    },
    {
      "train_step": 6708,
      "sup_loss": 0.5002157889719023
    },
    {
      "train_step": 6709,
      "sup_loss": 0.6165092016378917
    },
    {
      "train_step": 6710,
      "sup_loss": 0.7141463807139121
    },
    {
      "train_step": 6711,
      "sup_loss": 0.6375207855923164
    },
    {
      "train_step": 6712,
      "sup_loss": 0.37146748027604576
    },
    {
      "train_step": 6713,
      "sup_loss": 0.7014096150246646
    },
    {
      "train_step": 6714,
      "sup_loss": 0.6978941038682355
    },
    {
      "train_step": 6715,
      "sup_loss": 0.4555743063588486
    },
    {
      "train_step": 6716,
      "sup_loss": 0.49128672968340753
    },
    {
      "train_step": 6717,
      "sup_loss": 0.583189209640828
    },
    {
      "train_step": 6718,
      "sup_loss": 0.7220893893500462
    },
    {
      "train_step": 6719,
      "sup_loss": 0.7538508238866937
    },
    {
      "train_step": 6720,
      "sup_loss": 0.8996644907936237
    },
    {
      "train_step": 6721,
      "sup_loss": 0.6387782370877627
    },
    {
      "train_step": 6722,
      "sup_loss": 0.524362747931132
    },
    {
      "train_step": 6723,
      "sup_loss": 0.24972856582600902
    },
    {
      "train_step": 6724,
      "sup_loss": 0.6012292972220589
    },
    {
      "train_step": 6725,
      "sup_loss": 0.6015842001749991
    },
    {
      "train_step": 6726,
      "sup_loss": 0.5770535260865641
    },
    {
      "train_step": 6727,
      "sup_loss": 0.4165649751755803
    },
    {
      "train_step": 6728,
      "sup_loss": 0.3361694126527073
    },
    {
      "train_step": 6729,
      "sup_loss": 0.6345939493820381
    },
    {
      "train_step": 6730,
      "sup_loss": 0.8110769586649207
    },
    {
      "train_step": 6731,
      "sup_loss": 0.7209454158414967
    },
    {
      "train_step": 6732,
      "sup_loss": 0.612492695252874
    },
    {
      "train_step": 6733,
      "sup_loss": 0.5479009562849816
    },
    {
      "train_step": 6734,
      "sup_loss": 0.6309924628746334
    },
    {
      "train_step": 6735,
      "sup_loss": 0.7009544062557684
    },
    {
      "train_step": 6736,
      "sup_loss": 0.46855403623157676
    },
    {
      "train_step": 6737,
      "sup_loss": 0.38195727862260276
    },
    {
      "train_step": 6738,
      "sup_loss": 0.5086962234318303
    },
    {
      "train_step": 6739,
      "sup_loss": 0.597375222725019
    },
    {
      "train_step": 6740,
      "sup_loss": 0.5791384711531391
    },
    {
      "train_step": 6741,
      "sup_loss": 0.4507248821724277
    },
    {
      "train_step": 6742,
      "sup_loss": 0.6839898660121235
    },
    {
      "train_step": 6743,
      "sup_loss": 0.699194591104425
    },
    {
      "train_step": 6744,
      "sup_loss": 0.7294981827050918
    },
    {
      "train_step": 6745,
      "sup_loss": 0.5867172818337818
    },
    {
      "train_step": 6746,
      "sup_loss": 0.6293788503875878
    },
    {
      "train_step": 6747,
      "sup_loss": 0.565287657608046
    },
    {
      "train_step": 6748,
      "sup_loss": 0.5362325187233032
    },
    {
      "train_step": 6749,
      "sup_loss": 0.6994962769097854
    },
    {
      "train_step": 6750,
      "sup_loss": 0.5213401934142923
    },
    {
      "train_step": 6751,
      "sup_loss": 0.40971616387804555
    },
    {
      "train_step": 6752,
      "sup_loss": 0.4799352171689192
    },
    {
      "train_step": 6753,
      "sup_loss": 0.5847788979435447
    },
    {
      "train_step": 6754,
      "sup_loss": 0.6420010341691081
    },
    {
      "train_step": 6755,
      "sup_loss": 0.42681649096661495
    },
    {
      "train_step": 6756,
      "sup_loss": 0.6735817051312215
    },
    {
      "train_step": 6757,
      "sup_loss": 0.5833921524910671
    },
    {
      "train_step": 6758,
      "sup_loss": 0.6090339944898299
    },
    {
      "train_step": 6759,
      "sup_loss": 0.7148967570294325
    },
    {
      "train_step": 6760,
      "sup_loss": 0.6083054161421931
    },
    {
      "train_step": 6761,
      "sup_loss": 0.6162649154906803
    },
    {
      "train_step": 6762,
      "sup_loss": 0.5131022208058383
    },
    {
      "train_step": 6763,
      "sup_loss": 0.5130403121233715
    },
    {
      "train_step": 6764,
      "sup_loss": 0.38884405935356764
    },
    {
      "train_step": 6765,
      "sup_loss": 0.44121170090329637
    },
    {
      "train_step": 6766,
      "sup_loss": 0.6161481842625065
    },
    {
      "train_step": 6767,
      "sup_loss": 0.7508680879210616
    },
    {
      "train_step": 6768,
      "sup_loss": 0.7344507419805939
    },
    {
      "train_step": 6769,
      "sup_loss": 0.6943283044340954
    },
    {
      "train_step": 6770,
      "sup_loss": 0.514668280984446
    },
    {
      "train_step": 6771,
      "sup_loss": 0.782763359468965
    },
    {
      "train_step": 6772,
      "sup_loss": 0.561502791095373
    },
    {
      "train_step": 6773,
      "sup_loss": 0.6483490228288498
    },
    {
      "train_step": 6774,
      "sup_loss": 0.604349501200956
    },
    {
      "train_step": 6775,
      "sup_loss": 0.5986669693307345
    },
    {
      "train_step": 6776,
      "sup_loss": 0.6822050009889337
    },
    {
      "train_step": 6777,
      "sup_loss": 0.6262067659967145
    },
    {
      "train_step": 6778,
      "sup_loss": 0.5219801089528135
    },
    {
      "train_step": 6779,
      "sup_loss": 0.38978726372635913
    },
    {
      "train_step": 6780,
      "sup_loss": 0.5816215213507288
    },
    {
      "train_step": 6781,
      "sup_loss": 0.5341134097783243
    },
    {
      "train_step": 6782,
      "sup_loss": 0.707022334221928
    },
    {
      "train_step": 6783,
      "sup_loss": 0.5355352234514595
    },
    {
      "train_step": 6784,
      "sup_loss": 0.7144729405992386
    },
    {
      "train_step": 6785,
      "sup_loss": 0.6826880581266108
    },
    {
      "train_step": 6786,
      "sup_loss": 0.4484233459418371
    },
    {
      "train_step": 6787,
      "sup_loss": 0.7266164244258992
    },
    {
      "train_step": 6788,
      "sup_loss": 0.6335888754106628
    },
    {
      "train_step": 6789,
      "sup_loss": 0.5280117653966401
    },
    {
      "train_step": 6790,
      "sup_loss": 0.46986777800366897
    },
    {
      "train_step": 6791,
      "sup_loss": 0.4648434277632398
    },
    {
      "train_step": 6792,
      "sup_loss": 0.46667302810240313
    },
    {
      "train_step": 6793,
      "sup_loss": 0.4391396827924793
    },
    {
      "train_step": 6794,
      "sup_loss": 0.6243599144425372
    },
    {
      "train_step": 6795,
      "sup_loss": 0.7607176383457411
    },
    {
      "train_step": 6796,
      "sup_loss": 0.7204017491561464
    },
    {
      "train_step": 6797,
      "sup_loss": 0.5544855547679106
    },
    {
      "train_step": 6798,
      "sup_loss": 0.3143563218444534
    },
    {
      "train_step": 6799,
      "sup_loss": 0.5574330164432386
    },
    {
      "train_step": 6800,
      "sup_loss": 0.5291063770478786
    },
    {
      "train_step": 6801,
      "sup_loss": 0.569589097066264
    },
    {
      "train_step": 6802,
      "sup_loss": 0.5835491333590316
    },
    {
      "train_step": 6803,
      "sup_loss": 0.5733530564129117
    },
    {
      "train_step": 6804,
      "sup_loss": 0.5319369082579586
    },
    {
      "train_step": 6805,
      "sup_loss": 0.7032622868949744
    },
    {
      "train_step": 6806,
      "sup_loss": 0.7315815592661351
    },
    {
      "train_step": 6807,
      "sup_loss": 0.5265860346826555
    },
    {
      "train_step": 6808,
      "sup_loss": 0.708891870467091
    },
    {
      "train_step": 6809,
      "sup_loss": 0.4607503155778991
    },
    {
      "train_step": 6810,
      "sup_loss": 0.593873338763388
    },
    {
      "train_step": 6811,
      "sup_loss": 0.6664165042792191
    },
    {
      "train_step": 6812,
      "sup_loss": 0.6181370420374003
    },
    {
      "train_step": 6813,
      "sup_loss": 0.8022670321286445
    },
    {
      "train_step": 6814,
      "sup_loss": 0.6103702749066536
    },
    {
      "train_step": 6815,
      "sup_loss": 0.5584507856105339
    },
    {
      "train_step": 6816,
      "sup_loss": 0.7793025167072742
    },
    {
      "train_step": 6817,
      "sup_loss": 0.35298084200232177
    },
    {
      "train_step": 6818,
      "sup_loss": 0.5827191961491687
    },
    {
      "train_step": 6819,
      "sup_loss": 0.5827530467813542
    },
    {
      "train_step": 6820,
      "sup_loss": 0.598057742498784
    },
    {
      "train_step": 6821,
      "sup_loss": 0.4224933596392118
    },
    {
      "train_step": 6822,
      "sup_loss": 0.5247072071683152
    },
    {
      "train_step": 6823,
      "sup_loss": 0.48684068718024587
    },
    {
      "train_step": 6824,
      "sup_loss": 0.35446681712346606
    },
    {
      "train_step": 6825,
      "sup_loss": 0.6727009095873212
    },
    {
      "train_step": 6826,
      "sup_loss": 0.5847304176937124
    },
    {
      "train_step": 6827,
      "sup_loss": 0.6030037382148788
    },
    {
      "train_step": 6828,
      "sup_loss": 0.693218393124643
    },
    {
      "train_step": 6829,
      "sup_loss": 0.5771688291424788
    },
    {
      "train_step": 6830,
      "sup_loss": 0.44269192178020245
    },
    {
      "train_step": 6831,
      "sup_loss": 0.4887551602366285
    },
    {
      "train_step": 6832,
      "sup_loss": 0.7037814012397778
    },
    {
      "train_step": 6833,
      "sup_loss": 0.5453758813219828
    },
    {
      "train_step": 6834,
      "sup_loss": 0.7150871209757471
    },
    {
      "train_step": 6835,
      "sup_loss": 0.47044076412947583
    },
    {
      "train_step": 6836,
      "sup_loss": 0.29545726776402426
    },
    {
      "train_step": 6837,
      "sup_loss": 0.7397708133670173
    },
    {
      "train_step": 6838,
      "sup_loss": 0.49861973092592987
    },
    {
      "train_step": 6839,
      "sup_loss": 0.5884294310671254
    },
    {
      "train_step": 6840,
      "sup_loss": 0.5449765564645241
    },
    {
      "train_step": 6841,
      "sup_loss": 0.598588115307881
    },
    {
      "train_step": 6842,
      "sup_loss": 0.519523304227834
    },
    {
      "train_step": 6843,
      "sup_loss": 0.6103151435423657
    },
    {
      "train_step": 6844,
      "sup_loss": 0.8513299745249848
    },
    {
      "train_step": 6845,
      "sup_loss": 0.4292467542406843
    },
    {
      "train_step": 6846,
      "sup_loss": 0.34052869264108127
    },
    {
      "train_step": 6847,
      "sup_loss": 0.5911377257413529
    },
    {
      "train_step": 6848,
      "sup_loss": 0.5421916319476631
    },
    {
      "train_step": 6849,
      "sup_loss": 0.8913427707979626
    },
    {
      "train_step": 6850,
      "sup_loss": 0.5825608807190038
    },
    {
      "train_step": 6851,
      "sup_loss": 0.5735302050039698
    },
    {
      "train_step": 6852,
      "sup_loss": 0.4379112253068069
    },
    {
      "train_step": 6853,
      "sup_loss": 0.5113559452648884
    },
    {
      "train_step": 6854,
      "sup_loss": 0.5018990540307351
    },
    {
      "train_step": 6855,
      "sup_loss": 0.41705193528112144
    },
    {
      "train_step": 6856,
      "sup_loss": 0.7463756784677322
    },
    {
      "train_step": 6857,
      "sup_loss": 0.6177400422102746
    },
    {
      "train_step": 6858,
      "sup_loss": 0.6206455721165467
    },
    {
      "train_step": 6859,
      "sup_loss": 0.6548152660240053
    },
    {
      "train_step": 6860,
      "sup_loss": 0.6520132090432642
    },
    {
      "train_step": 6861,
      "sup_loss": 0.5166994166546532
    },
    {
      "train_step": 6862,
      "sup_loss": 0.5994189629346462
    },
    {
      "train_step": 6863,
      "sup_loss": 0.6613435794309245
    },
    {
      "train_step": 6864,
      "sup_loss": 0.7356638116364579
    },
    {
      "train_step": 6865,
      "sup_loss": 0.672512273950697
    },
    {
      "train_step": 6866,
      "sup_loss": 0.5995274813531088
    },
    {
      "train_step": 6867,
      "sup_loss": 0.5871652778850335
    },
    {
      "train_step": 6868,
      "sup_loss": 0.5354523293304706
    },
    {
      "train_step": 6869,
      "sup_loss": 0.5403140384042875
    },
    {
      "train_step": 6870,
      "sup_loss": 0.5627687104584196
    },
    {
      "train_step": 6871,
      "sup_loss": 0.503947048357678
    },
    {
      "train_step": 6872,
      "sup_loss": 0.5725419897939101
    },
    {
      "train_step": 6873,
      "sup_loss": 0.7584241997758152
    },
    {
      "train_step": 6874,
      "sup_loss": 0.4889115699063089
    },
    {
      "train_step": 6875,
      "sup_loss": 0.7105062554355478
    },
    {
      "train_step": 6876,
      "sup_loss": 0.6026333202132866
    },
    {
      "train_step": 6877,
      "sup_loss": 0.7268634755605592
    },
    {
      "train_step": 6878,
      "sup_loss": 0.5244435450187869
    },
    {
      "train_step": 6879,
      "sup_loss": 0.6376154792773661
    },
    {
      "train_step": 6880,
      "sup_loss": 0.8181122551674308
    },
    {
      "train_step": 6881,
      "sup_loss": 0.6853322505177352
    },
    {
      "train_step": 6882,
      "sup_loss": 0.8296988359548542
    },
    {
      "train_step": 6883,
      "sup_loss": 0.4201324760498876
    },
    {
      "train_step": 6884,
      "sup_loss": 0.5150835254958628
    },
    {
      "train_step": 6885,
      "sup_loss": 0.5765239344434686
    },
    {
      "train_step": 6886,
      "sup_loss": 0.49049290677489854
    },
    {
      "train_step": 6887,
      "sup_loss": 0.27443513532473984
    },
    {
      "train_step": 6888,
      "sup_loss": 0.570016101255688
    },
    {
      "train_step": 6889,
      "sup_loss": 0.6832983166615432
    },
    {
      "train_step": 6890,
      "sup_loss": 0.5803118954509645
    },
    {
      "train_step": 6891,
      "sup_loss": 0.5606004535943316
    },
    {
      "train_step": 6892,
      "sup_loss": 0.4918073172702593
    },
    {
      "train_step": 6893,
      "sup_loss": 0.283833547401077
    },
    {
      "train_step": 6894,
      "sup_loss": 0.6135304896815859
    },
    {
      "train_step": 6895,
      "sup_loss": 0.5340185096568072
    },
    {
      "train_step": 6896,
      "sup_loss": 0.7555096981430945
    },
    {
      "train_step": 6897,
      "sup_loss": 0.46046206965238595
    },
    {
      "train_step": 6898,
      "sup_loss": 0.500811069799976
    },
    {
      "train_step": 6899,
      "sup_loss": 0.43860884237130465
    },
    {
      "train_step": 6900,
      "sup_loss": 0.6674852174120637
    },
    {
      "train_step": 6901,
      "sup_loss": 0.551863726803579
    },
    {
      "train_step": 6902,
      "sup_loss": 0.6322058023459043
    },
    {
      "train_step": 6903,
      "sup_loss": 0.5797201636443854
    },
    {
      "train_step": 6904,
      "sup_loss": 0.6950336610756784
    },
    {
      "train_step": 6905,
      "sup_loss": 0.6644350727652271
    },
    {
      "train_step": 6906,
      "sup_loss": 0.6128108227108566
    },
    {
      "train_step": 6907,
      "sup_loss": 0.7476635292413322
    },
    {
      "train_step": 6908,
      "sup_loss": 0.7644309038839757
    },
    {
      "train_step": 6909,
      "sup_loss": 0.7226437136780808
    },
    {
      "train_step": 6910,
      "sup_loss": 0.6302086313074893
    },
    {
      "train_step": 6911,
      "sup_loss": 0.5748101055041617
    },
    {
      "train_step": 6912,
      "sup_loss": 0.6742422197217475
    },
    {
      "train_step": 6913,
      "sup_loss": 0.5928339214790651
    },
    {
      "train_step": 6914,
      "sup_loss": 0.5077687599406382
    },
    {
      "train_step": 6915,
      "sup_loss": 0.5671647491456449
    },
    {
      "train_step": 6916,
      "sup_loss": 0.4828723222238098
    },
    {
      "train_step": 6917,
      "sup_loss": 0.5537961285260633
    },
    {
      "train_step": 6918,
      "sup_loss": 0.5714798900338707
    },
    {
      "train_step": 6919,
      "sup_loss": 0.7654900436419027
    },
    {
      "train_step": 6920,
      "sup_loss": 0.5567753763610979
    },
    {
      "train_step": 6921,
      "sup_loss": 0.6055929445460604
    },
    {
      "train_step": 6922,
      "sup_loss": 0.35887506966912086
    },
    {
      "train_step": 6923,
      "sup_loss": 0.7096122939968675
    },
    {
      "train_step": 6924,
      "sup_loss": 0.30044878510480283
    },
    {
      "train_step": 6925,
      "sup_loss": 0.47997743183322694
    },
    {
      "train_step": 6926,
      "sup_loss": 0.5608159406993484
    },
    {
      "train_step": 6927,
      "sup_loss": 0.6296877847388497
    },
    {
      "train_step": 6928,
      "sup_loss": 0.5164869485161148
    },
    {
      "train_step": 6929,
      "sup_loss": 0.3714622092595596
    },
    {
      "train_step": 6930,
      "sup_loss": 0.27737477753611095
    },
    {
      "train_step": 6931,
      "sup_loss": 0.6682195810798718
    },
    {
      "train_step": 6932,
      "sup_loss": 0.6954309830702741
    },
    {
      "train_step": 6933,
      "sup_loss": 0.599933533120333
    },
    {
      "train_step": 6934,
      "sup_loss": 0.560381225707504
    },
    {
      "train_step": 6935,
      "sup_loss": 0.5260150642716714
    },
    {
      "train_step": 6936,
      "sup_loss": 0.8220083595116673
    },
    {
      "train_step": 6937,
      "sup_loss": 0.6196806614322267
    },
    {
      "train_step": 6938,
      "sup_loss": 0.41395168540785743
    },
    {
      "train_step": 6939,
      "sup_loss": 0.4625225554925352
    },
    {
      "train_step": 6940,
      "sup_loss": 0.7212318364626232
    },
    {
      "train_step": 6941,
      "sup_loss": 0.35567144677253554
    },
    {
      "train_step": 6942,
      "sup_loss": 0.6019319818092373
    },
    {
      "train_step": 6943,
      "sup_loss": 0.44474865749541614
    },
    {
      "train_step": 6944,
      "sup_loss": 0.6481340559175452
    },
    {
      "train_step": 6945,
      "sup_loss": 0.6691630802476423
    },
    {
      "train_step": 6946,
      "sup_loss": 0.509535285392894
    },
    {
      "train_step": 6947,
      "sup_loss": 0.5184698994668148
    },
    {
      "train_step": 6948,
      "sup_loss": 0.5087911158762578
    },
    {
      "train_step": 6949,
      "sup_loss": 0.6660557331544088
    },
    {
      "train_step": 6950,
      "sup_loss": 0.5326243317131848
    },
    {
      "train_step": 6951,
      "sup_loss": 0.6320132442460544
    },
    {
      "train_step": 6952,
      "sup_loss": 0.5385637664522812
    },
    {
      "train_step": 6953,
      "sup_loss": 0.36863250485902693
    },
    {
      "train_step": 6954,
      "sup_loss": 0.5544236551106648
    },
    {
      "train_step": 6955,
      "sup_loss": 0.6281885517235467
    },
    {
      "train_step": 6956,
      "sup_loss": 0.6454152810309475
    },
    {
      "train_step": 6957,
      "sup_loss": 0.612224322851566
    },
    {
      "train_step": 6958,
      "sup_loss": 0.8194523540655266
    },
    {
      "train_step": 6959,
      "sup_loss": 0.5566414228780687
    },
    {
      "train_step": 6960,
      "sup_loss": 0.47432712090494444
    },
    {
      "train_step": 6961,
      "sup_loss": 0.4709572321592438
    },
    {
      "train_step": 6962,
      "sup_loss": 0.6053617512685198
    },
    {
      "train_step": 6963,
      "sup_loss": 0.6578205456526822
    },
    {
      "train_step": 6964,
      "sup_loss": 0.5718689664447677
    },
    {
      "train_step": 6965,
      "sup_loss": 0.6152394422068405
    },
    {
      "train_step": 6966,
      "sup_loss": 0.5313322573318809
    },
    {
      "train_step": 6967,
      "sup_loss": 0.8383471401221699
    },
    {
      "train_step": 6968,
      "sup_loss": 0.6965315204270002
    },
    {
      "train_step": 6969,
      "sup_loss": 0.717119773947771
    },
    {
      "train_step": 6970,
      "sup_loss": 0.6616993952681682
    },
    {
      "train_step": 6971,
      "sup_loss": 0.36539009229907393
    },
    {
      "train_step": 6972,
      "sup_loss": 0.5785691742155724
    },
    {
      "train_step": 6973,
      "sup_loss": 0.7503072098342056
    },
    {
      "train_step": 6974,
      "sup_loss": 0.6197023851562876
    },
    {
      "train_step": 6975,
      "sup_loss": 0.47239254872769004
    },
    {
      "train_step": 6976,
      "sup_loss": 0.32615313679197955
    },
    {
      "train_step": 6977,
      "sup_loss": 0.4741010148603556
    },
    {
      "train_step": 6978,
      "sup_loss": 0.5175718594939369
    },
    {
      "train_step": 6979,
      "sup_loss": 0.4966406693710407
    },
    {
      "train_step": 6980,
      "sup_loss": 0.29907314941815816
    },
    {
      "train_step": 6981,
      "sup_loss": 0.6609874152427111
    },
    {
      "train_step": 6982,
      "sup_loss": 0.593407277526936
    },
    {
      "train_step": 6983,
      "sup_loss": 0.40530660764853976
    },
    {
      "train_step": 6984,
      "sup_loss": 0.38889757391926455
    },
    {
      "train_step": 6985,
      "sup_loss": 0.6529946012047314
    },
    {
      "train_step": 6986,
      "sup_loss": 0.4464405118384073
    },
    {
      "train_step": 6987,
      "sup_loss": 0.6169933934753484
    },
    {
      "train_step": 6988,
      "sup_loss": 0.5595400837542353
    },
    {
      "train_step": 6989,
      "sup_loss": 0.678611206625479
    },
    {
      "train_step": 6990,
      "sup_loss": 0.3466548354711382
    },
    {
      "train_step": 6991,
      "sup_loss": 0.46634958510610536
    },
    {
      "train_step": 6992,
      "sup_loss": 0.3430921052649611
    },
    {
      "train_step": 6993,
      "sup_loss": 0.47791800247103683
    },
    {
      "train_step": 6994,
      "sup_loss": 0.4836252846551657
    },
    {
      "train_step": 6995,
      "sup_loss": 0.6058779324315109
    },
    {
      "train_step": 6996,
      "sup_loss": 0.6891101553905454
    },
    {
      "train_step": 6997,
      "sup_loss": 0.5591289009927922
    },
    {
      "train_step": 6998,
      "sup_loss": 0.528813012397705
    },
    {
      "train_step": 6999,
      "sup_loss": 0.7016945956657327
    },
    {
      "train_step": 7000,
      "sup_loss": 0.4759200313228644
    },
    {
      "train_step": 7001,
      "sup_loss": 0.5670724518354107
    },
    {
      "train_step": 7002,
      "sup_loss": 0.5906213120927991
    },
    {
      "train_step": 7003,
      "sup_loss": 0.5958546322339521
    },
    {
      "train_step": 7004,
      "sup_loss": 0.3697196014673676
    },
    {
      "train_step": 7005,
      "sup_loss": 0.7439918369687398
    },
    {
      "train_step": 7006,
      "sup_loss": 0.6229292175188257
    },
    {
      "train_step": 7007,
      "sup_loss": 0.7250710106584022
    },
    {
      "train_step": 7008,
      "sup_loss": 0.5591141917609228
    },
    {
      "train_step": 7009,
      "sup_loss": 0.8315452232999779
    },
    {
      "train_step": 7010,
      "sup_loss": 0.36102049667451963
    },
    {
      "train_step": 7011,
      "sup_loss": 0.7662284419544922
    },
    {
      "train_step": 7012,
      "sup_loss": 0.3530429147490347
    },
    {
      "train_step": 7013,
      "sup_loss": 0.5564617506405882
    },
    {
      "train_step": 7014,
      "sup_loss": 0.3880470492241201
    },
    {
      "train_step": 7015,
      "sup_loss": 0.6352042332085478
    },
    {
      "train_step": 7016,
      "sup_loss": 0.4908071208021889
    },
    {
      "train_step": 7017,
      "sup_loss": 0.6514718118133205
    },
    {
      "train_step": 7018,
      "sup_loss": 0.7858446392877885
    },
    {
      "train_step": 7019,
      "sup_loss": 0.4320707675797765
    },
    {
      "train_step": 7020,
      "sup_loss": 0.5483186504258772
    },
    {
      "train_step": 7021,
      "sup_loss": 0.6525558061485052
    },
    {
      "train_step": 7022,
      "sup_loss": 0.4775272000654629
    },
    {
      "train_step": 7023,
      "sup_loss": 0.5289744202105952
    },
    {
      "train_step": 7024,
      "sup_loss": 0.5647276807899926
    },
    {
      "train_step": 7025,
      "sup_loss": 0.7213093914411158
    },
    {
      "train_step": 7026,
      "sup_loss": 0.5974701609610165
    },
    {
      "train_step": 7027,
      "sup_loss": 0.500538782627736
    },
    {
      "train_step": 7028,
      "sup_loss": 0.8015448112018668
    },
    {
      "train_step": 7029,
      "sup_loss": 0.5249909020466338
    },
    {
      "train_step": 7030,
      "sup_loss": 0.5372951641764243
    },
    {
      "train_step": 7031,
      "sup_loss": 0.7196267632104502
    },
    {
      "train_step": 7032,
      "sup_loss": 0.6299746613325461
    },
    {
      "train_step": 7033,
      "sup_loss": 0.6311549689221448
    },
    {
      "train_step": 7034,
      "sup_loss": 0.5685822943973542
    },
    {
      "train_step": 7035,
      "sup_loss": 0.6132258352691452
    },
    {
      "train_step": 7036,
      "sup_loss": 0.41970673648351176
    },
    {
      "train_step": 7037,
      "sup_loss": 0.42248448253662774
    },
    {
      "train_step": 7038,
      "sup_loss": 0.6508223455186342
    },
    {
      "train_step": 7039,
      "sup_loss": 0.5001022523710349
    },
    {
      "train_step": 7040,
      "sup_loss": 0.6516530695123878
    },
    {
      "train_step": 7041,
      "sup_loss": 0.7544376011082754
    },
    {
      "train_step": 7042,
      "sup_loss": 0.5849141048324321
    },
    {
      "train_step": 7043,
      "sup_loss": 0.5710165956371005
    },
    {
      "train_step": 7044,
      "sup_loss": 0.7128926686711585
    },
    {
      "train_step": 7045,
      "sup_loss": 0.7301604154637825
    },
    {
      "train_step": 7046,
      "sup_loss": 0.6645469771842777
    },
    {
      "train_step": 7047,
      "sup_loss": 0.6165318326454795
    },
    {
      "train_step": 7048,
      "sup_loss": 0.6418971476975979
    },
    {
      "train_step": 7049,
      "sup_loss": 0.4723498266039318
    },
    {
      "train_step": 7050,
      "sup_loss": 0.7076838805879794
    },
    {
      "train_step": 7051,
      "sup_loss": 0.7215941244157803
    },
    {
      "train_step": 7052,
      "sup_loss": 0.647593607994353
    },
    {
      "train_step": 7053,
      "sup_loss": 0.5800828572639259
    },
    {
      "train_step": 7054,
      "sup_loss": 0.6817889426175491
    },
    {
      "train_step": 7055,
      "sup_loss": 0.7201630142756171
    },
    {
      "train_step": 7056,
      "sup_loss": 0.7522977243426827
    },
    {
      "train_step": 7057,
      "sup_loss": 0.8678678831084796
    },
    {
      "train_step": 7058,
      "sup_loss": 0.5450965797455001
    },
    {
      "train_step": 7059,
      "sup_loss": 0.6168718619719699
    },
    {
      "train_step": 7060,
      "sup_loss": 0.6682983557678625
    },
    {
      "train_step": 7061,
      "sup_loss": 0.685734340991804
    },
    {
      "train_step": 7062,
      "sup_loss": 0.6339033554491197
    },
    {
      "train_step": 7063,
      "sup_loss": 0.7419253704956202
    },
    {
      "train_step": 7064,
      "sup_loss": 0.35092971648134247
    },
    {
      "train_step": 7065,
      "sup_loss": 0.5355609287941693
    },
    {
      "train_step": 7066,
      "sup_loss": 0.5167439426978088
    },
    {
      "train_step": 7067,
      "sup_loss": 0.5361617600303716
    },
    {
      "train_step": 7068,
      "sup_loss": 0.6376008308154013
    },
    {
      "train_step": 7069,
      "sup_loss": 0.5685544212458797
    },
    {
      "train_step": 7070,
      "sup_loss": 0.7397316226037359
    },
    {
      "train_step": 7071,
      "sup_loss": 0.6536169734963119
    },
    {
      "train_step": 7072,
      "sup_loss": 0.6571711568158889
    },
    {
      "train_step": 7073,
      "sup_loss": 0.6108295320574889
    },
    {
      "train_step": 7074,
      "sup_loss": 0.6185707706276135
    },
    {
      "train_step": 7075,
      "sup_loss": 0.6569330122760055
    },
    {
      "train_step": 7076,
      "sup_loss": 0.7214279096074399
    },
    {
      "train_step": 7077,
      "sup_loss": 0.33041073878351224
    },
    {
      "train_step": 7078,
      "sup_loss": 0.45292234978007884
    },
    {
      "train_step": 7079,
      "sup_loss": 0.6479576334305718
    },
    {
      "train_step": 7080,
      "sup_loss": 0.7646751326539427
    },
    {
      "train_step": 7081,
      "sup_loss": 0.44764500692890574
    },
    {
      "train_step": 7082,
      "sup_loss": 0.41985700349081845
    },
    {
      "train_step": 7083,
      "sup_loss": 0.4296569832777526
    },
    {
      "train_step": 7084,
      "sup_loss": 0.679989692260384
    },
    {
      "train_step": 7085,
      "sup_loss": 0.6358774460862409
    },
    {
      "train_step": 7086,
      "sup_loss": 0.4244183462453303
    },
    {
      "train_step": 7087,
      "sup_loss": 0.5193844033722566
    },
    {
      "train_step": 7088,
      "sup_loss": 0.6925532697157811
    },
    {
      "train_step": 7089,
      "sup_loss": 0.6606525259051381
    },
    {
      "train_step": 7090,
      "sup_loss": 0.3278620172344795
    },
    {
      "train_step": 7091,
      "sup_loss": 0.3767737814412065
    },
    {
      "train_step": 7092,
      "sup_loss": 0.5942212473884466
    },
    {
      "train_step": 7093,
      "sup_loss": 0.6344764925393994
    },
    {
      "train_step": 7094,
      "sup_loss": 0.7008111731424952
    },
    {
      "train_step": 7095,
      "sup_loss": 0.5657057733682745
    },
    {
      "train_step": 7096,
      "sup_loss": 0.37694291988753864
    },
    {
      "train_step": 7097,
      "sup_loss": 0.3996111597029225
    },
    {
      "train_step": 7098,
      "sup_loss": 0.7800121219678781
    },
    {
      "train_step": 7099,
      "sup_loss": 0.5781361752025991
    },
    {
      "train_step": 7100,
      "sup_loss": 0.5311558821309594
    },
    {
      "train_step": 7101,
      "sup_loss": 0.4367106535479643
    },
    {
      "train_step": 7102,
      "sup_loss": 0.7195479991000671
    },
    {
      "train_step": 7103,
      "sup_loss": 0.6395953581256679
    },
    {
      "train_step": 7104,
      "sup_loss": 0.583925795334505
    },
    {
      "train_step": 7105,
      "sup_loss": 0.45595857202084444
    },
    {
      "train_step": 7106,
      "sup_loss": 0.6601437806605859
    },
    {
      "train_step": 7107,
      "sup_loss": 0.5780575203838739
    },
    {
      "train_step": 7108,
      "sup_loss": 0.6700127510191767
    },
    {
      "train_step": 7109,
      "sup_loss": 0.6773709287930285
    },
    {
      "train_step": 7110,
      "sup_loss": 0.6107333288297868
    },
    {
      "train_step": 7111,
      "sup_loss": 0.6967188898122647
    },
    {
      "train_step": 7112,
      "sup_loss": 0.5004943213224026
    },
    {
      "train_step": 7113,
      "sup_loss": 0.5880655233879492
    },
    {
      "train_step": 7114,
      "sup_loss": 0.3841872115910116
    },
    {
      "train_step": 7115,
      "sup_loss": 0.6315719877661827
    },
    {
      "train_step": 7116,
      "sup_loss": 0.675591027801237
    },
    {
      "train_step": 7117,
      "sup_loss": 0.39785469651114314
    },
    {
      "train_step": 7118,
      "sup_loss": 0.6767614837229361
    },
    {
      "train_step": 7119,
      "sup_loss": 0.5949779958306269
    },
    {
      "train_step": 7120,
      "sup_loss": 0.6257001080004075
    },
    {
      "train_step": 7121,
      "sup_loss": 0.7334612889510209
    },
    {
      "train_step": 7122,
      "sup_loss": 0.599473311888672
    },
    {
      "train_step": 7123,
      "sup_loss": 0.536992230726248
    },
    {
      "train_step": 7124,
      "sup_loss": 0.4448552400064609
    },
    {
      "train_step": 7125,
      "sup_loss": 0.6612735943603912
    },
    {
      "train_step": 7126,
      "sup_loss": 0.5235980247041292
    },
    {
      "train_step": 7127,
      "sup_loss": 0.31495532045874847
    },
    {
      "train_step": 7128,
      "sup_loss": 0.4214995400694256
    },
    {
      "train_step": 7129,
      "sup_loss": 0.3864212139369423
    },
    {
      "train_step": 7130,
      "sup_loss": 0.5714249810189054
    },
    {
      "train_step": 7131,
      "sup_loss": 0.46786865661027005
    },
    {
      "train_step": 7132,
      "sup_loss": 0.5249826725531492
    },
    {
      "train_step": 7133,
      "sup_loss": 0.3904366510526539
    },
    {
      "train_step": 7134,
      "sup_loss": 0.5959489135311955
    },
    {
      "train_step": 7135,
      "sup_loss": 0.7056191314360318
    },
    {
      "train_step": 7136,
      "sup_loss": 0.4932535984955243
    },
    {
      "train_step": 7137,
      "sup_loss": 0.5756473581368722
    },
    {
      "train_step": 7138,
      "sup_loss": 0.685373383545134
    },
    {
      "train_step": 7139,
      "sup_loss": 0.3897860535245119
    },
    {
      "train_step": 7140,
      "sup_loss": 0.8308117752404567
    },
    {
      "train_step": 7141,
      "sup_loss": 0.5074156774018855
    },
    {
      "train_step": 7142,
      "sup_loss": 0.3343643201297491
    },
    {
      "train_step": 7143,
      "sup_loss": 0.5383913634417328
    },
    {
      "train_step": 7144,
      "sup_loss": 0.6453516727284733
    },
    {
      "train_step": 7145,
      "sup_loss": 0.6974405846657928
    },
    {
      "train_step": 7146,
      "sup_loss": 0.6880687369881873
    },
    {
      "train_step": 7147,
      "sup_loss": 0.5514970633635147
    },
    {
      "train_step": 7148,
      "sup_loss": 0.5339552166057736
    },
    {
      "train_step": 7149,
      "sup_loss": 0.6888326666929265
    },
    {
      "train_step": 7150,
      "sup_loss": 0.7753303803230812
    },
    {
      "train_step": 7151,
      "sup_loss": 0.7591803402754855
    },
    {
      "train_step": 7152,
      "sup_loss": 0.5711586429132148
    },
    {
      "train_step": 7153,
      "sup_loss": 0.4615967435335672
    },
    {
      "train_step": 7154,
      "sup_loss": 0.4160510926290304
    },
    {
      "train_step": 7155,
      "sup_loss": 0.6924588508023587
    },
    {
      "train_step": 7156,
      "sup_loss": 0.29086227705215967
    },
    {
      "train_step": 7157,
      "sup_loss": 0.6228153502667522
    },
    {
      "train_step": 7158,
      "sup_loss": 0.41891158594294897
    },
    {
      "train_step": 7159,
      "sup_loss": 0.5643431833342314
    },
    {
      "train_step": 7160,
      "sup_loss": 0.5939164381801039
    },
    {
      "train_step": 7161,
      "sup_loss": 0.3661969966937735
    },
    {
      "train_step": 7162,
      "sup_loss": 0.3321822933004199
    },
    {
      "train_step": 7163,
      "sup_loss": 0.7298928834251761
    },
    {
      "train_step": 7164,
      "sup_loss": 0.5142965346992017
    },
    {
      "train_step": 7165,
      "sup_loss": 0.580608654347732
    },
    {
      "train_step": 7166,
      "sup_loss": 0.5977554295141659
    },
    {
      "train_step": 7167,
      "sup_loss": 0.3961232454012038
    },
    {
      "train_step": 7168,
      "sup_loss": 0.5405043306780274
    },
    {
      "train_step": 7169,
      "sup_loss": 0.8318285637958956
    },
    {
      "train_step": 7170,
      "sup_loss": 0.6196437486966996
    },
    {
      "train_step": 7171,
      "sup_loss": 0.699058835210085
    },
    {
      "train_step": 7172,
      "sup_loss": 0.6322343488852945
    },
    {
      "train_step": 7173,
      "sup_loss": 0.6076019691883214
    },
    {
      "train_step": 7174,
      "sup_loss": 0.6219153643377836
    },
    {
      "train_step": 7175,
      "sup_loss": 0.6914725484298446
    },
    {
      "train_step": 7176,
      "sup_loss": 0.7402756214493708
    },
    {
      "train_step": 7177,
      "sup_loss": 0.48801665712733777
    },
    {
      "train_step": 7178,
      "sup_loss": 0.5699975124347936
    },
    {
      "train_step": 7179,
      "sup_loss": 0.6372106584133862
    },
    {
      "train_step": 7180,
      "sup_loss": 0.512882678195805
    },
    {
      "train_step": 7181,
      "sup_loss": 0.5423396349069771
    },
    {
      "train_step": 7182,
      "sup_loss": 0.5167552400222264
    },
    {
      "train_step": 7183,
      "sup_loss": 0.32273310398964106
    },
    {
      "train_step": 7184,
      "sup_loss": 0.4514581134128636
    },
    {
      "train_step": 7185,
      "sup_loss": 0.4093191355161503
    },
    {
      "train_step": 7186,
      "sup_loss": 1.336629257725957
    },
    {
      "train_step": 7187,
      "sup_loss": 0.5622230305131487
    },
    {
      "train_step": 7188,
      "sup_loss": 0.5279435974586292
    },
    {
      "train_step": 7189,
      "sup_loss": 0.5073263408044507
    },
    {
      "train_step": 7190,
      "sup_loss": 0.42728221583950926
    },
    {
      "train_step": 7191,
      "sup_loss": 0.5432207364603561
    },
    {
      "train_step": 7192,
      "sup_loss": 0.6865188058325609
    },
    {
      "train_step": 7193,
      "sup_loss": 0.6269897560209167
    },
    {
      "train_step": 7194,
      "sup_loss": 0.6092239174930892
    },
    {
      "train_step": 7195,
      "sup_loss": 0.49517845997938825
    },
    {
      "train_step": 7196,
      "sup_loss": 0.5430833087281347
    },
    {
      "train_step": 7197,
      "sup_loss": 0.5347957443073919
    },
    {
      "train_step": 7198,
      "sup_loss": 0.6160972052502978
    },
    {
      "train_step": 7199,
      "sup_loss": 0.6651055657055135
    },
    {
      "train_step": 7200,
      "sup_loss": 0.42215392812983515
    },
    {
      "train_step": 7201,
      "sup_loss": 0.6411655582341182
    },
    {
      "train_step": 7202,
      "sup_loss": 0.4523101148756299
    },
    {
      "train_step": 7203,
      "sup_loss": 0.8081026731403318
    },
    {
      "train_step": 7204,
      "sup_loss": 0.6324769918971741
    },
    {
      "train_step": 7205,
      "sup_loss": 0.7517252033627337
    },
    {
      "train_step": 7206,
      "sup_loss": 0.32426892747168745
    },
    {
      "train_step": 7207,
      "sup_loss": 0.7054776782120595
    },
    {
      "train_step": 7208,
      "sup_loss": 0.5646614629392395
    },
    {
      "train_step": 7209,
      "sup_loss": 0.514410995740829
    },
    {
      "train_step": 7210,
      "sup_loss": 0.7047515438173021
    },
    {
      "train_step": 7211,
      "sup_loss": 0.5574722203277085
    },
    {
      "train_step": 7212,
      "sup_loss": 0.661056103552603
    },
    {
      "train_step": 7213,
      "sup_loss": 0.5651874335042371
    },
    {
      "train_step": 7214,
      "sup_loss": 0.6668026391757584
    },
    {
      "train_step": 7215,
      "sup_loss": 0.605572154673281
    },
    {
      "train_step": 7216,
      "sup_loss": 0.6336771263233479
    },
    {
      "train_step": 7217,
      "sup_loss": 0.5886447229407564
    },
    {
      "train_step": 7218,
      "sup_loss": 0.7462812312941514
    },
    {
      "train_step": 7219,
      "sup_loss": 0.5562815896658003
    },
    {
      "train_step": 7220,
      "sup_loss": 0.6558315016506405
    },
    {
      "train_step": 7221,
      "sup_loss": 0.4924168277801242
    },
    {
      "train_step": 7222,
      "sup_loss": 0.6017259791955905
    },
    {
      "train_step": 7223,
      "sup_loss": 0.4032211331901655
    },
    {
      "train_step": 7224,
      "sup_loss": 0.574182863131536
    },
    {
      "train_step": 7225,
      "sup_loss": 0.6158191425395563
    },
    {
      "train_step": 7226,
      "sup_loss": 0.6593416866781596
    },
    {
      "train_step": 7227,
      "sup_loss": 0.5725021538401984
    },
    {
      "train_step": 7228,
      "sup_loss": 0.5828218010423598
    },
    {
      "train_step": 7229,
      "sup_loss": 0.4364559370640472
    },
    {
      "train_step": 7230,
      "sup_loss": 0.5811952518850122
    },
    {
      "train_step": 7231,
      "sup_loss": 0.5740114693962124
    },
    {
      "train_step": 7232,
      "sup_loss": 0.4512786679009904
    },
    {
      "train_step": 7233,
      "sup_loss": 0.5364904179508032
    },
    {
      "train_step": 7234,
      "sup_loss": 0.5565143854732406
    },
    {
      "train_step": 7235,
      "sup_loss": 0.8291487779204143
    },
    {
      "train_step": 7236,
      "sup_loss": 0.8020274533092445
    },
    {
      "train_step": 7237,
      "sup_loss": 0.6419000465906752
    },
    {
      "train_step": 7238,
      "sup_loss": 0.7201749092129528
    },
    {
      "train_step": 7239,
      "sup_loss": 0.6123100626610328
    },
    {
      "train_step": 7240,
      "sup_loss": 0.5527204704139359
    },
    {
      "train_step": 7241,
      "sup_loss": 0.6490085419417818
    },
    {
      "train_step": 7242,
      "sup_loss": 0.8318441992268032
    },
    {
      "train_step": 7243,
      "sup_loss": 0.4273715537116318
    },
    {
      "train_step": 7244,
      "sup_loss": 0.5035093609393112
    },
    {
      "train_step": 7245,
      "sup_loss": 0.6372800574590927
    },
    {
      "train_step": 7246,
      "sup_loss": 0.5518689958701115
    },
    {
      "train_step": 7247,
      "sup_loss": 0.3560868731880269
    },
    {
      "train_step": 7248,
      "sup_loss": 0.71633339009672
    },
    {
      "train_step": 7249,
      "sup_loss": 0.7221673181770474
    },
    {
      "train_step": 7250,
      "sup_loss": 0.4357680628527442
    },
    {
      "train_step": 7251,
      "sup_loss": 0.5315556937524192
    },
    {
      "train_step": 7252,
      "sup_loss": 0.5208264072970236
    },
    {
      "train_step": 7253,
      "sup_loss": 0.712005833495402
    },
    {
      "train_step": 7254,
      "sup_loss": 0.7768048051718295
    },
    {
      "train_step": 7255,
      "sup_loss": 0.549578712209467
    },
    {
      "train_step": 7256,
      "sup_loss": 0.3433249229267783
    },
    {
      "train_step": 7257,
      "sup_loss": 0.42541065376119813
    },
    {
      "train_step": 7258,
      "sup_loss": 0.4573893891314398
    },
    {
      "train_step": 7259,
      "sup_loss": 0.5062222705997519
    },
    {
      "train_step": 7260,
      "sup_loss": 0.6391594333825306
    },
    {
      "train_step": 7261,
      "sup_loss": 0.4870482040748256
    },
    {
      "train_step": 7262,
      "sup_loss": 0.5854850533551305
    },
    {
      "train_step": 7263,
      "sup_loss": 0.3855419107608873
    },
    {
      "train_step": 7264,
      "sup_loss": 0.8385475255260446
    },
    {
      "train_step": 7265,
      "sup_loss": 0.6943233433440192
    },
    {
      "train_step": 7266,
      "sup_loss": 0.4603004765538629
    },
    {
      "train_step": 7267,
      "sup_loss": 0.5541576036742225
    },
    {
      "train_step": 7268,
      "sup_loss": 0.37529764059704235
    },
    {
      "train_step": 7269,
      "sup_loss": 0.674578670872759
    },
    {
      "train_step": 7270,
      "sup_loss": 0.5711689684537892
    },
    {
      "train_step": 7271,
      "sup_loss": 0.4547704777295524
    },
    {
      "train_step": 7272,
      "sup_loss": 0.5432791044643668
    },
    {
      "train_step": 7273,
      "sup_loss": 0.5708388914034462
    },
    {
      "train_step": 7274,
      "sup_loss": 0.3547417870430297
    },
    {
      "train_step": 7275,
      "sup_loss": 0.8402441932674867
    },
    {
      "train_step": 7276,
      "sup_loss": 0.5302293375287652
    },
    {
      "train_step": 7277,
      "sup_loss": 0.5320273863812525
    },
    {
      "train_step": 7278,
      "sup_loss": 0.5985031734056184
    },
    {
      "train_step": 7279,
      "sup_loss": 0.5428857597915543
    },
    {
      "train_step": 7280,
      "sup_loss": 0.5859506278231663
    },
    {
      "train_step": 7281,
      "sup_loss": 0.2634180559944007
    },
    {
      "train_step": 7282,
      "sup_loss": 0.4550305283553972
    },
    {
      "train_step": 7283,
      "sup_loss": 0.3990013172466974
    },
    {
      "train_step": 7284,
      "sup_loss": 0.7527652291408227
    },
    {
      "train_step": 7285,
      "sup_loss": 0.4952296798338516
    },
    {
      "train_step": 7286,
      "sup_loss": 0.5109676123650659
    },
    {
      "train_step": 7287,
      "sup_loss": 0.6988793419451436
    },
    {
      "train_step": 7288,
      "sup_loss": 0.5758735910001302
    },
    {
      "train_step": 7289,
      "sup_loss": 0.5239938002421585
    },
    {
      "train_step": 7290,
      "sup_loss": 0.7574247013523244
    },
    {
      "train_step": 7291,
      "sup_loss": 0.5364187572327617
    },
    {
      "train_step": 7292,
      "sup_loss": 0.539206708313649
    },
    {
      "train_step": 7293,
      "sup_loss": 0.5829936611168981
    },
    {
      "train_step": 7294,
      "sup_loss": 0.6597649734209344
    },
    {
      "train_step": 7295,
      "sup_loss": 0.3700471865620842
    },
    {
      "train_step": 7296,
      "sup_loss": 0.5925825949616602
    },
    {
      "train_step": 7297,
      "sup_loss": 0.5067336293502855
    },
    {
      "train_step": 7298,
      "sup_loss": 0.38233032220598195
    },
    {
      "train_step": 7299,
      "sup_loss": 0.7107368316628198
    },
    {
      "train_step": 7300,
      "sup_loss": 0.5429600620165577
    },
    {
      "train_step": 7301,
      "sup_loss": 0.591068523206475
    },
    {
      "train_step": 7302,
      "sup_loss": 0.5412228194316339
    },
    {
      "train_step": 7303,
      "sup_loss": 0.5231613058353755
    },
    {
      "train_step": 7304,
      "sup_loss": 0.5487932101521789
    },
    {
      "train_step": 7305,
      "sup_loss": 0.39320719905857293
    },
    {
      "train_step": 7306,
      "sup_loss": 0.584913590349933
    },
    {
      "train_step": 7307,
      "sup_loss": 0.5068383021964997
    },
    {
      "train_step": 7308,
      "sup_loss": 0.6791887346947015
    },
    {
      "train_step": 7309,
      "sup_loss": 0.6057667908515372
    },
    {
      "train_step": 7310,
      "sup_loss": 0.7492488547162156
    },
    {
      "train_step": 7311,
      "sup_loss": 0.5209922241791136
    },
    {
      "train_step": 7312,
      "sup_loss": 0.8570880396498813
    },
    {
      "train_step": 7313,
      "sup_loss": 0.44105931448317237
    },
    {
      "train_step": 7314,
      "sup_loss": 0.5715606737814454
    },
    {
      "train_step": 7315,
      "sup_loss": 0.6498775395315491
    },
    {
      "train_step": 7316,
      "sup_loss": 0.7603516820709084
    },
    {
      "train_step": 7317,
      "sup_loss": 0.6741235894242336
    },
    {
      "train_step": 7318,
      "sup_loss": 0.43620260020272256
    },
    {
      "train_step": 7319,
      "sup_loss": 0.5270782054794212
    },
    {
      "train_step": 7320,
      "sup_loss": 0.5524546177857388
    },
    {
      "train_step": 7321,
      "sup_loss": 0.49966629376457283
    },
    {
      "train_step": 7322,
      "sup_loss": 0.680446350486624
    },
    {
      "train_step": 7323,
      "sup_loss": 0.32278153077178867
    },
    {
      "train_step": 7324,
      "sup_loss": 0.7052813730074994
    },
    {
      "train_step": 7325,
      "sup_loss": 0.3264951442840616
    },
    {
      "train_step": 7326,
      "sup_loss": 0.5684161031401851
    },
    {
      "train_step": 7327,
      "sup_loss": 0.5349392617381988
    },
    {
      "train_step": 7328,
      "sup_loss": 0.37959823343179233
    },
    {
      "train_step": 7329,
      "sup_loss": 0.802148461284442
    },
    {
      "train_step": 7330,
      "sup_loss": 0.525352207212853
    },
    {
      "train_step": 7331,
      "sup_loss": 0.426357190799201
    },
    {
      "train_step": 7332,
      "sup_loss": 0.4067252193852294
    },
    {
      "train_step": 7333,
      "sup_loss": 0.7606918666802549
    },
    {
      "train_step": 7334,
      "sup_loss": 0.4372093209186962
    },
    {
      "train_step": 7335,
      "sup_loss": 0.4878978831478845
    },
    {
      "train_step": 7336,
      "sup_loss": 0.39004759696870356
    },
    {
      "train_step": 7337,
      "sup_loss": 0.5641601138018061
    },
    {
      "train_step": 7338,
      "sup_loss": 0.7217338802241576
    },
    {
      "train_step": 7339,
      "sup_loss": 0.34349646716264753
    },
    {
      "train_step": 7340,
      "sup_loss": 0.35746876658607146
    },
    {
      "train_step": 7341,
      "sup_loss": 0.7421700018897454
    },
    {
      "train_step": 7342,
      "sup_loss": 0.41431658778559666
    },
    {
      "train_step": 7343,
      "sup_loss": 0.6895623223017118
    },
    {
      "train_step": 7344,
      "sup_loss": 0.3802701316952512
    },
    {
      "train_step": 7345,
      "sup_loss": 0.5180405128390559
    },
    {
      "train_step": 7346,
      "sup_loss": 0.5091718217830581
    },
    {
      "train_step": 7347,
      "sup_loss": 0.6238230167682678
    },
    {
      "train_step": 7348,
      "sup_loss": 0.6996947212255612
    },
    {
      "train_step": 7349,
      "sup_loss": 0.507682610751801
    },
    {
      "train_step": 7350,
      "sup_loss": 0.5128902207764157
    },
    {
      "train_step": 7351,
      "sup_loss": 0.6737921520945522
    },
    {
      "train_step": 7352,
      "sup_loss": 0.6158500987792805
    },
    {
      "train_step": 7353,
      "sup_loss": 0.44488988310156985
    },
    {
      "train_step": 7354,
      "sup_loss": 0.5039702556346498
    },
    {
      "train_step": 7355,
      "sup_loss": 0.5871665287097971
    },
    {
      "train_step": 7356,
      "sup_loss": 0.6266570040505344
    },
    {
      "train_step": 7357,
      "sup_loss": 0.5414916914207583
    },
    {
      "train_step": 7358,
      "sup_loss": 0.4131931325212884
    },
    {
      "train_step": 7359,
      "sup_loss": 0.5581648529186858
    },
    {
      "train_step": 7360,
      "sup_loss": 0.4678668908607482
    },
    {
      "train_step": 7361,
      "sup_loss": 0.5498769895983303
    },
    {
      "train_step": 7362,
      "sup_loss": 0.5051159110842454
    },
    {
      "train_step": 7363,
      "sup_loss": 0.32274252892729705
    },
    {
      "train_step": 7364,
      "sup_loss": 0.558124757741378
    },
    {
      "train_step": 7365,
      "sup_loss": 0.5775030409025138
    },
    {
      "train_step": 7366,
      "sup_loss": 0.7542042674956564
    },
    {
      "train_step": 7367,
      "sup_loss": 0.4868080425183483
    },
    {
      "train_step": 7368,
      "sup_loss": 0.6718138596443209
    },
    {
      "train_step": 7369,
      "sup_loss": 0.348070037695816
    },
    {
      "train_step": 7370,
      "sup_loss": 0.6089878378358878
    },
    {
      "train_step": 7371,
      "sup_loss": 0.3570157701916869
    },
    {
      "train_step": 7372,
      "sup_loss": 0.6299974081380724
    },
    {
      "train_step": 7373,
      "sup_loss": 0.5842855542334189
    },
    {
      "train_step": 7374,
      "sup_loss": 0.6533727252548794
    },
    {
      "train_step": 7375,
      "sup_loss": 0.4580777946973034
    },
    {
      "train_step": 7376,
      "sup_loss": 0.37384172868717586
    },
    {
      "train_step": 7377,
      "sup_loss": 0.5327408755641991
    },
    {
      "train_step": 7378,
      "sup_loss": 0.6352550936639201
    },
    {
      "train_step": 7379,
      "sup_loss": 0.6816601345261145
    },
    {
      "train_step": 7380,
      "sup_loss": 0.7391916619202311
    },
    {
      "train_step": 7381,
      "sup_loss": 0.6565963654007926
    },
    {
      "train_step": 7382,
      "sup_loss": 0.5164608133512081
    },
    {
      "train_step": 7383,
      "sup_loss": 0.5271732254242292
    },
    {
      "train_step": 7384,
      "sup_loss": 0.6214901756622784
    },
    {
      "train_step": 7385,
      "sup_loss": 0.6308009144414481
    },
    {
      "train_step": 7386,
      "sup_loss": 0.6295262661335007
    },
    {
      "train_step": 7387,
      "sup_loss": 0.3589943054901105
    },
    {
      "train_step": 7388,
      "sup_loss": 0.7026772752146705
    },
    {
      "train_step": 7389,
      "sup_loss": 0.34924053685598133
    },
    {
      "train_step": 7390,
      "sup_loss": 0.47683093034792723
    },
    {
      "train_step": 7391,
      "sup_loss": 0.6349067760438412
    },
    {
      "train_step": 7392,
      "sup_loss": 0.5558758372151826
    },
    {
      "train_step": 7393,
      "sup_loss": 0.5786717545309719
    },
    {
      "train_step": 7394,
      "sup_loss": 0.3500984293368121
    },
    {
      "train_step": 7395,
      "sup_loss": 0.5838156563437401
    },
    {
      "train_step": 7396,
      "sup_loss": 0.36565495782688406
    },
    {
      "train_step": 7397,
      "sup_loss": 0.6564176253803871
    },
    {
      "train_step": 7398,
      "sup_loss": 0.7609335585258141
    },
    {
      "train_step": 7399,
      "sup_loss": 0.6889666739308212
    },
    {
      "train_step": 7400,
      "sup_loss": 0.5730063918893842
    },
    {
      "train_step": 7401,
      "sup_loss": 0.6255786407782575
    },
    {
      "train_step": 7402,
      "sup_loss": 0.5722689216327063
    },
    {
      "train_step": 7403,
      "sup_loss": 0.5551237233731009
    },
    {
      "train_step": 7404,
      "sup_loss": 0.3930148261313927
    },
    {
      "train_step": 7405,
      "sup_loss": 0.599495911799019
    },
    {
      "train_step": 7406,
      "sup_loss": 0.515420147229367
    },
    {
      "train_step": 7407,
      "sup_loss": 0.4327497180031555
    },
    {
      "train_step": 7408,
      "sup_loss": 0.734143107509815
    },
    {
      "train_step": 7409,
      "sup_loss": 0.5755394940287801
    },
    {
      "train_step": 7410,
      "sup_loss": 0.4455234233336656
    },
    {
      "train_step": 7411,
      "sup_loss": 0.6033356809769104
    },
    {
      "train_step": 7412,
      "sup_loss": 0.587378313914503
    },
    {
      "train_step": 7413,
      "sup_loss": 0.5151882652814529
    },
    {
      "train_step": 7414,
      "sup_loss": 0.503759116853151
    },
    {
      "train_step": 7415,
      "sup_loss": 0.574019992113588
    },
    {
      "train_step": 7416,
      "sup_loss": 0.7275593221413004
    },
    {
      "train_step": 7417,
      "sup_loss": 0.42104203027866316
    },
    {
      "train_step": 7418,
      "sup_loss": 0.5555081353486357
    },
    {
      "train_step": 7419,
      "sup_loss": 0.5066850636704601
    },
    {
      "train_step": 7420,
      "sup_loss": 0.6266833328902925
    },
    {
      "train_step": 7421,
      "sup_loss": 0.504241386793024
    },
    {
      "train_step": 7422,
      "sup_loss": 0.5011504109368303
    },
    {
      "train_step": 7423,
      "sup_loss": 0.5550619097791345
    },
    {
      "train_step": 7424,
      "sup_loss": 0.5702368556665841
    },
    {
      "train_step": 7425,
      "sup_loss": 0.5690854798506639
    },
    {
      "train_step": 7426,
      "sup_loss": 0.48789306328506304
    },
    {
      "train_step": 7427,
      "sup_loss": 0.4882184767671382
    },
    {
      "train_step": 7428,
      "sup_loss": 0.5310106398931027
    },
    {
      "train_step": 7429,
      "sup_loss": 0.5545162211850541
    },
    {
      "train_step": 7430,
      "sup_loss": 0.46395361427574144
    },
    {
      "train_step": 7431,
      "sup_loss": 0.2615444374951284
    },
    {
      "train_step": 7432,
      "sup_loss": 0.686957634854854
    },
    {
      "train_step": 7433,
      "sup_loss": 0.2538218829562488
    },
    {
      "train_step": 7434,
      "sup_loss": 0.6137266917855343
    },
    {
      "train_step": 7435,
      "sup_loss": 0.7107321501416609
    },
    {
      "train_step": 7436,
      "sup_loss": 0.35094275643103184
    },
    {
      "train_step": 7437,
      "sup_loss": 0.7394996644036554
    },
    {
      "train_step": 7438,
      "sup_loss": 0.40608412880731637
    },
    {
      "train_step": 7439,
      "sup_loss": 0.535854780839721
    },
    {
      "train_step": 7440,
      "sup_loss": 0.5878719546095853
    },
    {
      "train_step": 7441,
      "sup_loss": 0.3725479758504462
    },
    {
      "train_step": 7442,
      "sup_loss": 0.6178852495195216
    },
    {
      "train_step": 7443,
      "sup_loss": 0.6158243567906552
    },
    {
      "train_step": 7444,
      "sup_loss": 0.7814631383431675
    },
    {
      "train_step": 7445,
      "sup_loss": 0.6615751352705048
    },
    {
      "train_step": 7446,
      "sup_loss": 0.6463345605267313
    },
    {
      "train_step": 7447,
      "sup_loss": 0.24215142887324642
    },
    {
      "train_step": 7448,
      "sup_loss": 0.4236004794223492
    },
    {
      "train_step": 7449,
      "sup_loss": 0.34095406470527156
    },
    {
      "train_step": 7450,
      "sup_loss": 0.6950528978869757
    },
    {
      "train_step": 7451,
      "sup_loss": 0.4278700585721044
    },
    {
      "train_step": 7452,
      "sup_loss": 0.5728567987632011
    },
    {
      "train_step": 7453,
      "sup_loss": 0.7483661594343538
    },
    {
      "train_step": 7454,
      "sup_loss": 0.6124026317744116
    },
    {
      "train_step": 7455,
      "sup_loss": 0.7040762103508721
    },
    {
      "train_step": 7456,
      "sup_loss": 0.7266184883040097
    },
    {
      "train_step": 7457,
      "sup_loss": 0.6271036657645196
    },
    {
      "train_step": 7458,
      "sup_loss": 0.5817841403817863
    },
    {
      "train_step": 7459,
      "sup_loss": 0.6478726203139988
    },
    {
      "train_step": 7460,
      "sup_loss": 0.3849665969677206
    },
    {
      "train_step": 7461,
      "sup_loss": 0.6288021468969602
    },
    {
      "train_step": 7462,
      "sup_loss": 0.6918015651851699
    },
    {
      "train_step": 7463,
      "sup_loss": 0.8833006675127805
    },
    {
      "train_step": 7464,
      "sup_loss": 0.5412008520597957
    },
    {
      "train_step": 7465,
      "sup_loss": 0.5985314205373016
    },
    {
      "train_step": 7466,
      "sup_loss": 0.8201653798416672
    },
    {
      "train_step": 7467,
      "sup_loss": 0.6747107421497117
    },
    {
      "train_step": 7468,
      "sup_loss": 0.6962998237240695
    },
    {
      "train_step": 7469,
      "sup_loss": 0.4298355682136442
    },
    {
      "train_step": 7470,
      "sup_loss": 0.4566964601663342
    },
    {
      "train_step": 7471,
      "sup_loss": 0.7623815274622395
    },
    {
      "train_step": 7472,
      "sup_loss": 0.5785132250368483
    },
    {
      "train_step": 7473,
      "sup_loss": 0.7849569874079467
    },
    {
      "train_step": 7474,
      "sup_loss": 0.6757277305427064
    },
    {
      "train_step": 7475,
      "sup_loss": 0.6911924163489688
    },
    {
      "train_step": 7476,
      "sup_loss": 0.6054533087379509
    },
    {
      "train_step": 7477,
      "sup_loss": 0.6082123750839411
    },
    {
      "train_step": 7478,
      "sup_loss": 0.4141373592340651
    },
    {
      "train_step": 7479,
      "sup_loss": 0.4405503060479099
    },
    {
      "train_step": 7480,
      "sup_loss": 0.5005081459560455
    },
    {
      "train_step": 7481,
      "sup_loss": 0.5042153740392421
    },
    {
      "train_step": 7482,
      "sup_loss": 0.539353429926876
    },
    {
      "train_step": 7483,
      "sup_loss": 0.5929775261387772
    },
    {
      "train_step": 7484,
      "sup_loss": 0.4505856532525805
    },
    {
      "train_step": 7485,
      "sup_loss": 0.41510877868515966
    },
    {
      "train_step": 7486,
      "sup_loss": 0.5139810220147322
    },
    {
      "train_step": 7487,
      "sup_loss": 0.6933004528152716
    },
    {
      "train_step": 7488,
      "sup_loss": 0.5603771723748016
    },
    {
      "train_step": 7489,
      "sup_loss": 0.7278536040318072
    },
    {
      "train_step": 7490,
      "sup_loss": 0.8065047909421453
    },
    {
      "train_step": 7491,
      "sup_loss": 0.6188390132113707
    },
    {
      "train_step": 7492,
      "sup_loss": 0.6237135742380873
    },
    {
      "train_step": 7493,
      "sup_loss": 0.6304017992855881
    },
    {
      "train_step": 7494,
      "sup_loss": 0.5078351405330142
    },
    {
      "train_step": 7495,
      "sup_loss": 0.7320386141021905
    },
    {
      "train_step": 7496,
      "sup_loss": 0.5837369529292747
    },
    {
      "train_step": 7497,
      "sup_loss": 0.5873042431691632
    },
    {
      "train_step": 7498,
      "sup_loss": 0.6548788579609227
    },
    {
      "train_step": 7499,
      "sup_loss": 0.4385232898648574
    },
    {
      "train_step": 7500,
      "sup_loss": 0.6162052398010771
    },
    {
      "train_step": 7501,
      "sup_loss": 0.5940834996570544
    },
    {
      "train_step": 7502,
      "sup_loss": 0.5850007390445604
    },
    {
      "train_step": 7503,
      "sup_loss": 0.311969876534159
    },
    {
      "train_step": 7504,
      "sup_loss": 0.47806899050014445
    },
    {
      "train_step": 7505,
      "sup_loss": 0.6054965276084298
    },
    {
      "train_step": 7506,
      "sup_loss": 0.6360743363386112
    },
    {
      "train_step": 7507,
      "sup_loss": 0.7009386925423823
    },
    {
      "train_step": 7508,
      "sup_loss": 0.6610792754930083
    },
    {
      "train_step": 7509,
      "sup_loss": 0.5544560595485705
    },
    {
      "train_step": 7510,
      "sup_loss": 0.5496628971683402
    },
    {
      "train_step": 7511,
      "sup_loss": 0.4709898854595222
    },
    {
      "train_step": 7512,
      "sup_loss": 0.3908119285482041
    },
    {
      "train_step": 7513,
      "sup_loss": 0.674064705372602
    },
    {
      "train_step": 7514,
      "sup_loss": 0.547724449453798
    },
    {
      "train_step": 7515,
      "sup_loss": 0.6539476647135233
    },
    {
      "train_step": 7516,
      "sup_loss": 0.36872654949248174
    },
    {
      "train_step": 7517,
      "sup_loss": 0.6287980866683732
    },
    {
      "train_step": 7518,
      "sup_loss": 0.7436686755558737
    },
    {
      "train_step": 7519,
      "sup_loss": 0.5775139413151993
    },
    {
      "train_step": 7520,
      "sup_loss": 0.5389425221185824
    },
    {
      "train_step": 7521,
      "sup_loss": 0.6338930290021194
    },
    {
      "train_step": 7522,
      "sup_loss": 0.5127975808947832
    },
    {
      "train_step": 7523,
      "sup_loss": 0.5964023937525599
    },
    {
      "train_step": 7524,
      "sup_loss": 0.55431781925114
    },
    {
      "train_step": 7525,
      "sup_loss": 0.46691397193942963
    },
    {
      "train_step": 7526,
      "sup_loss": 0.5830805177865723
    },
    {
      "train_step": 7527,
      "sup_loss": 0.6305942784919532
    },
    {
      "train_step": 7528,
      "sup_loss": 0.4324002492306943
    },
    {
      "train_step": 7529,
      "sup_loss": 0.6441482137652003
    },
    {
      "train_step": 7530,
      "sup_loss": 0.6263147927229407
    },
    {
      "train_step": 7531,
      "sup_loss": 0.5452662269848088
    },
    {
      "train_step": 7532,
      "sup_loss": 0.5895173062077619
    },
    {
      "train_step": 7533,
      "sup_loss": 0.6299025714965023
    },
    {
      "train_step": 7534,
      "sup_loss": 0.5022313643634078
    },
    {
      "train_step": 7535,
      "sup_loss": 0.49897440818584576
    },
    {
      "train_step": 7536,
      "sup_loss": 0.576176028984599
    },
    {
      "train_step": 7537,
      "sup_loss": 0.7235796130537177
    },
    {
      "train_step": 7538,
      "sup_loss": 0.669343222653462
    },
    {
      "train_step": 7539,
      "sup_loss": 0.6134575614808655
    },
    {
      "train_step": 7540,
      "sup_loss": 0.6572864305293754
    },
    {
      "train_step": 7541,
      "sup_loss": 0.6247407279092786
    },
    {
      "train_step": 7542,
      "sup_loss": 0.6034858803269717
    },
    {
      "train_step": 7543,
      "sup_loss": 0.48809878433927506
    },
    {
      "train_step": 7544,
      "sup_loss": 0.6682560736735712
    },
    {
      "train_step": 7545,
      "sup_loss": 0.44575924076162887
    },
    {
      "train_step": 7546,
      "sup_loss": 0.4868076802345572
    },
    {
      "train_step": 7547,
      "sup_loss": 0.5092509104687786
    },
    {
      "train_step": 7548,
      "sup_loss": 0.5384854305348608
    },
    {
      "train_step": 7549,
      "sup_loss": 0.4708213759212518
    },
    {
      "train_step": 7550,
      "sup_loss": 0.42557192818057765
    },
    {
      "train_step": 7551,
      "sup_loss": 0.4226537049439445
    },
    {
      "train_step": 7552,
      "sup_loss": 0.47677198129274834
    },
    {
      "train_step": 7553,
      "sup_loss": 0.3325507639860989
    },
    {
      "train_step": 7554,
      "sup_loss": 0.5889040636447734
    },
    {
      "train_step": 7555,
      "sup_loss": 0.6086651368554087
    },
    {
      "train_step": 7556,
      "sup_loss": 0.5051204939749316
    },
    {
      "train_step": 7557,
      "sup_loss": 0.5833587807701595
    },
    {
      "train_step": 7558,
      "sup_loss": 0.5966057275362333
    },
    {
      "train_step": 7559,
      "sup_loss": 0.4619756799198092
    },
    {
      "train_step": 7560,
      "sup_loss": 0.46073952545644303
    },
    {
      "train_step": 7561,
      "sup_loss": 0.30393486193876323
    },
    {
      "train_step": 7562,
      "sup_loss": 0.6607345903596672
    },
    {
      "train_step": 7563,
      "sup_loss": 0.39448011896560786
    },
    {
      "train_step": 7564,
      "sup_loss": 0.8733968201979109
    },
    {
      "train_step": 7565,
      "sup_loss": 0.5773456721619036
    },
    {
      "train_step": 7566,
      "sup_loss": 0.38592336116234777
    },
    {
      "train_step": 7567,
      "sup_loss": 0.45608813972283163
    },
    {
      "train_step": 7568,
      "sup_loss": 0.31464969864551384
    },
    {
      "train_step": 7569,
      "sup_loss": 0.45713840249694937
    },
    {
      "train_step": 7570,
      "sup_loss": 0.7030892651944463
    },
    {
      "train_step": 7571,
      "sup_loss": 0.5645915048057684
    },
    {
      "train_step": 7572,
      "sup_loss": 0.5468655134476016
    },
    {
      "train_step": 7573,
      "sup_loss": 1.2823752097911836
    },
    {
      "train_step": 7574,
      "sup_loss": 0.7354608699599712
    },
    {
      "train_step": 7575,
      "sup_loss": 0.5160352510470569
    },
    {
      "train_step": 7576,
      "sup_loss": 0.6417568667064439
    },
    {
      "train_step": 7577,
      "sup_loss": 0.6657083042321785
    },
    {
      "train_step": 7578,
      "sup_loss": 0.5689301316496544
    },
    {
      "train_step": 7579,
      "sup_loss": 0.6436363089603981
    },
    {
      "train_step": 7580,
      "sup_loss": 0.522736669101372
    },
    {
      "train_step": 7581,
      "sup_loss": 0.5454149857961628
    },
    {
      "train_step": 7582,
      "sup_loss": 0.47233670145826734
    },
    {
      "train_step": 7583,
      "sup_loss": 0.604122827292211
    },
    {
      "train_step": 7584,
      "sup_loss": 0.4548253876293653
    },
    {
      "train_step": 7585,
      "sup_loss": 0.49897416614766815
    },
    {
      "train_step": 7586,
      "sup_loss": 0.4750213346248923
    },
    {
      "train_step": 7587,
      "sup_loss": 0.6585414289045536
    },
    {
      "train_step": 7588,
      "sup_loss": 0.6455197131751266
    },
    {
      "train_step": 7589,
      "sup_loss": 0.6510747808069084
    },
    {
      "train_step": 7590,
      "sup_loss": 0.4509099557887819
    },
    {
      "train_step": 7591,
      "sup_loss": 0.6652269607558698
    },
    {
      "train_step": 7592,
      "sup_loss": 0.7986696964435365
    },
    {
      "train_step": 7593,
      "sup_loss": 0.4617484188683729
    },
    {
      "train_step": 7594,
      "sup_loss": 0.4385727583895388
    },
    {
      "train_step": 7595,
      "sup_loss": 0.8130800309962807
    },
    {
      "train_step": 7596,
      "sup_loss": 0.553496158379901
    },
    {
      "train_step": 7597,
      "sup_loss": 0.3604803937915747
    },
    {
      "train_step": 7598,
      "sup_loss": 0.5695912815793491
    },
    {
      "train_step": 7599,
      "sup_loss": 0.556715390831775
    },
    {
      "train_step": 7600,
      "sup_loss": 0.4745746899079088
    },
    {
      "train_step": 7601,
      "sup_loss": 0.7288158417396194
    },
    {
      "train_step": 7602,
      "sup_loss": 0.7167958580432973
    },
    {
      "train_step": 7603,
      "sup_loss": 0.6079777340773513
    },
    {
      "train_step": 7604,
      "sup_loss": 0.7469452969714574
    },
    {
      "train_step": 7605,
      "sup_loss": 0.6203003274384109
    },
    {
      "train_step": 7606,
      "sup_loss": 0.5767728597043683
    },
    {
      "train_step": 7607,
      "sup_loss": 0.7413003788595927
    },
    {
      "train_step": 7608,
      "sup_loss": 0.568856264688523
    },
    {
      "train_step": 7609,
      "sup_loss": 0.4128503831904273
    },
    {
      "train_step": 7610,
      "sup_loss": 0.6699448129880505
    },
    {
      "train_step": 7611,
      "sup_loss": 0.4129062668495788
    },
    {
      "train_step": 7612,
      "sup_loss": 0.2843840728180822
    },
    {
      "train_step": 7613,
      "sup_loss": 0.6717074231437711
    },
    {
      "train_step": 7614,
      "sup_loss": 0.6085900441041208
    },
    {
      "train_step": 7615,
      "sup_loss": 0.5272453713861852
    },
    {
      "train_step": 7616,
      "sup_loss": 0.5874861437090578
    },
    {
      "train_step": 7617,
      "sup_loss": 0.598846395489699
    },
    {
      "train_step": 7618,
      "sup_loss": 0.6733953581016547
    },
    {
      "train_step": 7619,
      "sup_loss": 0.4883095440465675
    },
    {
      "train_step": 7620,
      "sup_loss": 0.42244631889267614
    },
    {
      "train_step": 7621,
      "sup_loss": 0.7929492111554671
    },
    {
      "train_step": 7622,
      "sup_loss": 0.3510257512806595
    },
    {
      "train_step": 7623,
      "sup_loss": 0.6129644150510534
    },
    {
      "train_step": 7624,
      "sup_loss": 0.29282237181877263
    },
    {
      "train_step": 7625,
      "sup_loss": 0.5833783970235276
    },
    {
      "train_step": 7626,
      "sup_loss": 0.7321555395202118
    },
    {
      "train_step": 7627,
      "sup_loss": 0.3680915453745707
    },
    {
      "train_step": 7628,
      "sup_loss": 0.45936077406117726
    },
    {
      "train_step": 7629,
      "sup_loss": 0.44556255534452693
    },
    {
      "train_step": 7630,
      "sup_loss": 0.5915738313110964
    },
    {
      "train_step": 7631,
      "sup_loss": 0.3723209432816774
    },
    {
      "train_step": 7632,
      "sup_loss": 0.6379343785260108
    },
    {
      "train_step": 7633,
      "sup_loss": 0.5804019738014473
    },
    {
      "train_step": 7634,
      "sup_loss": 0.5224670519474427
    },
    {
      "train_step": 7635,
      "sup_loss": 0.2326122731115277
    },
    {
      "train_step": 7636,
      "sup_loss": 0.6541133394530504
    },
    {
      "train_step": 7637,
      "sup_loss": 0.8771231597280941
    },
    {
      "train_step": 7638,
      "sup_loss": 0.6294601063168428
    },
    {
      "train_step": 7639,
      "sup_loss": 0.6635538960427677
    },
    {
      "train_step": 7640,
      "sup_loss": 0.4108915217140045
    },
    {
      "train_step": 7641,
      "sup_loss": 0.561406919295425
    },
    {
      "train_step": 7642,
      "sup_loss": 0.3085116940004766
    },
    {
      "train_step": 7643,
      "sup_loss": 0.7058483934234174
    },
    {
      "train_step": 7644,
      "sup_loss": 0.5947256046828102
    },
    {
      "train_step": 7645,
      "sup_loss": 0.5519525679088182
    },
    {
      "train_step": 7646,
      "sup_loss": 0.583697221976911
    },
    {
      "train_step": 7647,
      "sup_loss": 0.7009230702051658
    },
    {
      "train_step": 7648,
      "sup_loss": 0.6128370705429581
    },
    {
      "train_step": 7649,
      "sup_loss": 0.37638957381984717
    },
    {
      "train_step": 7650,
      "sup_loss": 0.43360922933297724
    },
    {
      "train_step": 7651,
      "sup_loss": 0.6574894745355707
    },
    {
      "train_step": 7652,
      "sup_loss": 0.7933532474225929
    },
    {
      "train_step": 7653,
      "sup_loss": 0.6836949850608357
    },
    {
      "train_step": 7654,
      "sup_loss": 0.5642633244926759
    },
    {
      "train_step": 7655,
      "sup_loss": 0.6126964154708663
    },
    {
      "train_step": 7656,
      "sup_loss": 0.728918520420839
    },
    {
      "train_step": 7657,
      "sup_loss": 0.5552071165666612
    },
    {
      "train_step": 7658,
      "sup_loss": 0.5192178142785608
    },
    {
      "train_step": 7659,
      "sup_loss": 0.41092006251148927
    },
    {
      "train_step": 7660,
      "sup_loss": 0.40164028209088565
    },
    {
      "train_step": 7661,
      "sup_loss": 0.4876334793741024
    },
    {
      "train_step": 7662,
      "sup_loss": 0.4794927738288973
    },
    {
      "train_step": 7663,
      "sup_loss": 0.671148027127297
    },
    {
      "train_step": 7664,
      "sup_loss": 0.6285912910543794
    },
    {
      "train_step": 7665,
      "sup_loss": 0.5540492488183639
    },
    {
      "train_step": 7666,
      "sup_loss": 0.5331657496870805
    },
    {
      "train_step": 7667,
      "sup_loss": 0.876593808851251
    },
    {
      "train_step": 7668,
      "sup_loss": 0.5268886846541051
    },
    {
      "train_step": 7669,
      "sup_loss": 0.5361241367611035
    },
    {
      "train_step": 7670,
      "sup_loss": 0.5619424705644247
    },
    {
      "train_step": 7671,
      "sup_loss": 0.42821460837420666
    },
    {
      "train_step": 7672,
      "sup_loss": 0.5479635816593558
    },
    {
      "train_step": 7673,
      "sup_loss": 0.5212889342280794
    },
    {
      "train_step": 7674,
      "sup_loss": 0.6490334496244753
    },
    {
      "train_step": 7675,
      "sup_loss": 0.7448165132949639
    },
    {
      "train_step": 7676,
      "sup_loss": 0.5756355218010755
    },
    {
      "train_step": 7677,
      "sup_loss": 0.7019955868003888
    },
    {
      "train_step": 7678,
      "sup_loss": 0.461483996957257
    },
    {
      "train_step": 7679,
      "sup_loss": 0.698305349277818
    },
    {
      "train_step": 7680,
      "sup_loss": 0.6048254978964125
    },
    {
      "train_step": 7681,
      "sup_loss": 0.6940481077270908
    },
    {
      "train_step": 7682,
      "sup_loss": 0.7710336099651758
    },
    {
      "train_step": 7683,
      "sup_loss": 0.40258544939916047
    },
    {
      "train_step": 7684,
      "sup_loss": 0.6009148692241896
    },
    {
      "train_step": 7685,
      "sup_loss": 0.5992788389512013
    },
    {
      "train_step": 7686,
      "sup_loss": 0.6821113442274576
    },
    {
      "train_step": 7687,
      "sup_loss": 0.3987710421323911
    },
    {
      "train_step": 7688,
      "sup_loss": 0.3276167618597182
    },
    {
      "train_step": 7689,
      "sup_loss": 0.6445533388220576
    },
    {
      "train_step": 7690,
      "sup_loss": 0.6563640381739283
    },
    {
      "train_step": 7691,
      "sup_loss": 0.5375175898402512
    },
    {
      "train_step": 7692,
      "sup_loss": 0.3530549999164063
    },
    {
      "train_step": 7693,
      "sup_loss": 0.8305795960448316
    },
    {
      "train_step": 7694,
      "sup_loss": 0.6767719677896917
    },
    {
      "train_step": 7695,
      "sup_loss": 0.4845273209489489
    },
    {
      "train_step": 7696,
      "sup_loss": 0.9750855804306935
    },
    {
      "train_step": 7697,
      "sup_loss": 0.643397473423486
    },
    {
      "train_step": 7698,
      "sup_loss": 0.5564541636535406
    },
    {
      "train_step": 7699,
      "sup_loss": 0.7785739109549733
    },
    {
      "train_step": 7700,
      "sup_loss": 0.7398670527022336
    },
    {
      "train_step": 7701,
      "sup_loss": 0.6268165933601277
    },
    {
      "train_step": 7702,
      "sup_loss": 0.7993910799640943
    },
    {
      "train_step": 7703,
      "sup_loss": 0.6758382537002878
    },
    {
      "train_step": 7704,
      "sup_loss": 0.46174631202456873
    },
    {
      "train_step": 7705,
      "sup_loss": 0.42706550285860073
    },
    {
      "train_step": 7706,
      "sup_loss": 0.27168180578354767
    },
    {
      "train_step": 7707,
      "sup_loss": 0.3010535570797583
    },
    {
      "train_step": 7708,
      "sup_loss": 0.6480192026716198
    },
    {
      "train_step": 7709,
      "sup_loss": 0.6014794916321955
    },
    {
      "train_step": 7710,
      "sup_loss": 0.6582398548252996
    },
    {
      "train_step": 7711,
      "sup_loss": 0.5651045624437627
    },
    {
      "train_step": 7712,
      "sup_loss": 0.6953298092127569
    },
    {
      "train_step": 7713,
      "sup_loss": 0.4000029797841588
    },
    {
      "train_step": 7714,
      "sup_loss": 0.4017386834000512
    },
    {
      "train_step": 7715,
      "sup_loss": 0.41329678727527425
    },
    {
      "train_step": 7716,
      "sup_loss": 0.44295593771824887
    },
    {
      "train_step": 7717,
      "sup_loss": 0.5554020146071921
    },
    {
      "train_step": 7718,
      "sup_loss": 0.47816570770402916
    },
    {
      "train_step": 7719,
      "sup_loss": 0.6341634735848415
    },
    {
      "train_step": 7720,
      "sup_loss": 0.48094282943234934
    },
    {
      "train_step": 7721,
      "sup_loss": 0.5130302862408307
    },
    {
      "train_step": 7722,
      "sup_loss": 0.5958195844749274
    },
    {
      "train_step": 7723,
      "sup_loss": 0.5927574801453785
    },
    {
      "train_step": 7724,
      "sup_loss": 0.5644886287580765
    },
    {
      "train_step": 7725,
      "sup_loss": 0.7573321856798574
    },
    {
      "train_step": 7726,
      "sup_loss": 0.6952131327086849
    },
    {
      "train_step": 7727,
      "sup_loss": 0.7098324894939874
    },
    {
      "train_step": 7728,
      "sup_loss": 0.7590661526236234
    },
    {
      "train_step": 7729,
      "sup_loss": 0.4819413545832245
    },
    {
      "train_step": 7730,
      "sup_loss": 0.4652879904509588
    },
    {
      "train_step": 7731,
      "sup_loss": 0.3508451080942583
    },
    {
      "train_step": 7732,
      "sup_loss": 0.5489332571733766
    },
    {
      "train_step": 7733,
      "sup_loss": 0.6588679237480397
    },
    {
      "train_step": 7734,
      "sup_loss": 0.5085962085848047
    },
    {
      "train_step": 7735,
      "sup_loss": 0.538937454227582
    },
    {
      "train_step": 7736,
      "sup_loss": 0.44065638149994135
    },
    {
      "train_step": 7737,
      "sup_loss": 0.6117695083909822
    },
    {
      "train_step": 7738,
      "sup_loss": 0.573876384408522
    },
    {
      "train_step": 7739,
      "sup_loss": 0.6363667447507604
    },
    {
      "train_step": 7740,
      "sup_loss": 0.7096764655893328
    },
    {
      "train_step": 7741,
      "sup_loss": 0.7455527482495499
    },
    {
      "train_step": 7742,
      "sup_loss": 0.5801347610028682
    },
    {
      "train_step": 7743,
      "sup_loss": 0.5715073273994434
    },
    {
      "train_step": 7744,
      "sup_loss": 0.5880626067660376
    },
    {
      "train_step": 7745,
      "sup_loss": 0.5502541378535521
    },
    {
      "train_step": 7746,
      "sup_loss": 0.34879001800965465
    },
    {
      "train_step": 7747,
      "sup_loss": 0.46167611573272876
    },
    {
      "train_step": 7748,
      "sup_loss": 0.5026288054613759
    },
    {
      "train_step": 7749,
      "sup_loss": 0.5500712409055454
    },
    {
      "train_step": 7750,
      "sup_loss": 0.5632454868349148
    },
    {
      "train_step": 7751,
      "sup_loss": 0.7417981710718757
    },
    {
      "train_step": 7752,
      "sup_loss": 0.6248626046138397
    },
    {
      "train_step": 7753,
      "sup_loss": 0.5177942689096924
    },
    {
      "train_step": 7754,
      "sup_loss": 0.5561634651832508
    },
    {
      "train_step": 7755,
      "sup_loss": 0.3153643152697539
    },
    {
      "train_step": 7756,
      "sup_loss": 0.514810317822567
    },
    {
      "train_step": 7757,
      "sup_loss": 0.5925901985151913
    },
    {
      "train_step": 7758,
      "sup_loss": 0.6315739797562296
    },
    {
      "train_step": 7759,
      "sup_loss": 0.4287312615387242
    },
    {
      "train_step": 7760,
      "sup_loss": 0.7892931078602565
    },
    {
      "train_step": 7761,
      "sup_loss": 0.5441625571580613
    },
    {
      "train_step": 7762,
      "sup_loss": 0.6614187496115234
    },
    {
      "train_step": 7763,
      "sup_loss": 0.17415981454363383
    },
    {
      "train_step": 7764,
      "sup_loss": 0.5092618456983355
    },
    {
      "train_step": 7765,
      "sup_loss": 0.4502760512181883
    },
    {
      "train_step": 7766,
      "sup_loss": 0.36272681448977484
    },
    {
      "train_step": 7767,
      "sup_loss": 0.5384903346197202
    },
    {
      "train_step": 7768,
      "sup_loss": 0.5209571396935413
    },
    {
      "train_step": 7769,
      "sup_loss": 0.5487023013655321
    },
    {
      "train_step": 7770,
      "sup_loss": 0.49089400325437416
    },
    {
      "train_step": 7771,
      "sup_loss": 0.6839977027320054
    },
    {
      "train_step": 7772,
      "sup_loss": 0.48534424179177926
    },
    {
      "train_step": 7773,
      "sup_loss": 0.5963371197837747
    },
    {
      "train_step": 7774,
      "sup_loss": 0.7487343477147681
    },
    {
      "train_step": 7775,
      "sup_loss": 0.3636760625550693
    },
    {
      "train_step": 7776,
      "sup_loss": 0.6494168704897678
    },
    {
      "train_step": 7777,
      "sup_loss": 0.6035000614333871
    },
    {
      "train_step": 7778,
      "sup_loss": 0.5890169436327439
    },
    {
      "train_step": 7779,
      "sup_loss": 0.3357125273211949
    },
    {
      "train_step": 7780,
      "sup_loss": 0.7457752354759555
    },
    {
      "train_step": 7781,
      "sup_loss": 0.5114890032723766
    },
    {
      "train_step": 7782,
      "sup_loss": 0.7406198465008137
    },
    {
      "train_step": 7783,
      "sup_loss": 0.7206851411958072
    },
    {
      "train_step": 7784,
      "sup_loss": 0.5830263031217295
    },
    {
      "train_step": 7785,
      "sup_loss": 0.531905037114837
    },
    {
      "train_step": 7786,
      "sup_loss": 0.48962461170871774
    },
    {
      "train_step": 7787,
      "sup_loss": 0.5862781695149513
    },
    {
      "train_step": 7788,
      "sup_loss": 0.4491935987913073
    },
    {
      "train_step": 7789,
      "sup_loss": 0.4119105525690478
    },
    {
      "train_step": 7790,
      "sup_loss": 0.6302207611088197
    },
    {
      "train_step": 7791,
      "sup_loss": 0.5724819047884351
    },
    {
      "train_step": 7792,
      "sup_loss": 0.4766862836347797
    },
    {
      "train_step": 7793,
      "sup_loss": 0.4584749459332953
    },
    {
      "train_step": 7794,
      "sup_loss": 0.37786510132845863
    },
    {
      "train_step": 7795,
      "sup_loss": 0.41786709913133147
    },
    {
      "train_step": 7796,
      "sup_loss": 0.43577569729344606
    },
    {
      "train_step": 7797,
      "sup_loss": 0.49108312154002925
    },
    {
      "train_step": 7798,
      "sup_loss": 0.2780970862758237
    },
    {
      "train_step": 7799,
      "sup_loss": 0.5674142995474724
    },
    {
      "train_step": 7800,
      "sup_loss": 0.289645657722806
    },
    {
      "train_step": 7801,
      "sup_loss": 0.6642682260006798
    },
    {
      "train_step": 7802,
      "sup_loss": 0.5074681158307931
    },
    {
      "train_step": 7803,
      "sup_loss": 0.5208083260010528
    },
    {
      "train_step": 7804,
      "sup_loss": 0.48902671619776583
    },
    {
      "train_step": 7805,
      "sup_loss": 0.39387814318202535
    },
    {
      "train_step": 7806,
      "sup_loss": 0.5956524119507444
    },
    {
      "train_step": 7807,
      "sup_loss": 0.4115161512900088
    },
    {
      "train_step": 7808,
      "sup_loss": 0.6478653060212632
    },
    {
      "train_step": 7809,
      "sup_loss": 0.5869876709141151
    },
    {
      "train_step": 7810,
      "sup_loss": 0.4991076052850731
    },
    {
      "train_step": 7811,
      "sup_loss": 0.687850947923622
    },
    {
      "train_step": 7812,
      "sup_loss": 0.7729188727173351
    },
    {
      "train_step": 7813,
      "sup_loss": 0.5105757329947256
    },
    {
      "train_step": 7814,
      "sup_loss": 0.6904517739344394
    },
    {
      "train_step": 7815,
      "sup_loss": 0.6389002228413454
    },
    {
      "train_step": 7816,
      "sup_loss": 0.5923344327445234
    },
    {
      "train_step": 7817,
      "sup_loss": 0.5866727426485853
    },
    {
      "train_step": 7818,
      "sup_loss": 0.5896691700628212
    },
    {
      "train_step": 7819,
      "sup_loss": 0.7222915686069785
    },
    {
      "train_step": 7820,
      "sup_loss": 0.3854980096213361
    },
    {
      "train_step": 7821,
      "sup_loss": 0.6265569895806427
    },
    {
      "train_step": 7822,
      "sup_loss": 0.723926645203092
    },
    {
      "train_step": 7823,
      "sup_loss": 0.6613785135978074
    },
    {
      "train_step": 7824,
      "sup_loss": 0.4922393876288244
    },
    {
      "train_step": 7825,
      "sup_loss": 0.528815654050516
    },
    {
      "train_step": 7826,
      "sup_loss": 0.640307798524546
    },
    {
      "train_step": 7827,
      "sup_loss": 0.464942364294199
    },
    {
      "train_step": 7828,
      "sup_loss": 0.49085021301639337
    },
    {
      "train_step": 7829,
      "sup_loss": 0.7434229736960302
    },
    {
      "train_step": 7830,
      "sup_loss": 0.42164813162903436
    },
    {
      "train_step": 7831,
      "sup_loss": 0.5461478106114241
    },
    {
      "train_step": 7832,
      "sup_loss": 0.3606272449789602
    },
    {
      "train_step": 7833,
      "sup_loss": 0.6163794497688225
    },
    {
      "train_step": 7834,
      "sup_loss": 0.5570008541358893
    },
    {
      "train_step": 7835,
      "sup_loss": 0.6689756583913885
    },
    {
      "train_step": 7836,
      "sup_loss": 0.4623454553037214
    },
    {
      "train_step": 7837,
      "sup_loss": 0.5586693747342711
    },
    {
      "train_step": 7838,
      "sup_loss": 0.4951369310002145
    },
    {
      "train_step": 7839,
      "sup_loss": 0.5628347964379828
    },
    {
      "train_step": 7840,
      "sup_loss": 0.3995555925065084
    },
    {
      "train_step": 7841,
      "sup_loss": 0.6567311018503171
    },
    {
      "train_step": 7842,
      "sup_loss": 0.550756326194607
    },
    {
      "train_step": 7843,
      "sup_loss": 0.6819585881182088
    },
    {
      "train_step": 7844,
      "sup_loss": 0.32108045574105093
    },
    {
      "train_step": 7845,
      "sup_loss": 0.7225248222081454
    },
    {
      "train_step": 7846,
      "sup_loss": 0.6451877208073673
    },
    {
      "train_step": 7847,
      "sup_loss": 0.46305248999933535
    },
    {
      "train_step": 7848,
      "sup_loss": 0.7235549298841187
    },
    {
      "train_step": 7849,
      "sup_loss": 0.4680031989802985
    },
    {
      "train_step": 7850,
      "sup_loss": 0.5675912492338823
    },
    {
      "train_step": 7851,
      "sup_loss": 0.6269569532085774
    },
    {
      "train_step": 7852,
      "sup_loss": 0.31985244265600454
    },
    {
      "train_step": 7853,
      "sup_loss": 0.6815334774771852
    },
    {
      "train_step": 7854,
      "sup_loss": 0.5191682155188003
    },
    {
      "train_step": 7855,
      "sup_loss": 0.5788230927372244
    },
    {
      "train_step": 7856,
      "sup_loss": 0.4797500683592434
    },
    {
      "train_step": 7857,
      "sup_loss": 0.5237866723201101
    },
    {
      "train_step": 7858,
      "sup_loss": 0.5952493477908973
    },
    {
      "train_step": 7859,
      "sup_loss": 0.4731904479673715
    },
    {
      "train_step": 7860,
      "sup_loss": 0.6022649721818202
    },
    {
      "train_step": 7861,
      "sup_loss": 0.32613115010642446
    },
    {
      "train_step": 7862,
      "sup_loss": 0.6856356465247222
    },
    {
      "train_step": 7863,
      "sup_loss": 0.4854248268896745
    },
    {
      "train_step": 7864,
      "sup_loss": 0.4619777783188631
    },
    {
      "train_step": 7865,
      "sup_loss": 0.6141079952779709
    },
    {
      "train_step": 7866,
      "sup_loss": 0.47773817850378825
    },
    {
      "train_step": 7867,
      "sup_loss": 0.40848280415263183
    },
    {
      "train_step": 7868,
      "sup_loss": 0.6242897366900824
    },
    {
      "train_step": 7869,
      "sup_loss": 0.5973670257957635
    },
    {
      "train_step": 7870,
      "sup_loss": 0.610549163242656
    },
    {
      "train_step": 7871,
      "sup_loss": 0.7756588544664473
    },
    {
      "train_step": 7872,
      "sup_loss": 0.4975764862631783
    },
    {
      "train_step": 7873,
      "sup_loss": 0.6963851342172099
    },
    {
      "train_step": 7874,
      "sup_loss": 0.46664509241579094
    },
    {
      "train_step": 7875,
      "sup_loss": 0.6828772463250549
    },
    {
      "train_step": 7876,
      "sup_loss": 0.6312654725085398
    },
    {
      "train_step": 7877,
      "sup_loss": 0.4974534181460069
    },
    {
      "train_step": 7878,
      "sup_loss": 0.47385748164488517
    },
    {
      "train_step": 7879,
      "sup_loss": 0.6340499651580637
    },
    {
      "train_step": 7880,
      "sup_loss": 0.42324239141549613
    },
    {
      "train_step": 7881,
      "sup_loss": 0.6989778238894944
    },
    {
      "train_step": 7882,
      "sup_loss": 0.4587314305949173
    },
    {
      "train_step": 7883,
      "sup_loss": 0.46019366550517027
    },
    {
      "train_step": 7884,
      "sup_loss": 0.6586398688666029
    },
    {
      "train_step": 7885,
      "sup_loss": 0.6470653301293746
    },
    {
      "train_step": 7886,
      "sup_loss": 0.5531186984791815
    },
    {
      "train_step": 7887,
      "sup_loss": 0.7307793420915168
    },
    {
      "train_step": 7888,
      "sup_loss": 0.6413462498399922
    },
    {
      "train_step": 7889,
      "sup_loss": 0.5419495974625949
    },
    {
      "train_step": 7890,
      "sup_loss": 0.6014678460437552
    },
    {
      "train_step": 7891,
      "sup_loss": 0.7181164718087446
    },
    {
      "train_step": 7892,
      "sup_loss": 0.44678011360619546
    },
    {
      "train_step": 7893,
      "sup_loss": 0.7145701021861355
    },
    {
      "train_step": 7894,
      "sup_loss": 0.5839550975132671
    },
    {
      "train_step": 7895,
      "sup_loss": 0.574952411642374
    },
    {
      "train_step": 7896,
      "sup_loss": 0.4371603613508771
    },
    {
      "train_step": 7897,
      "sup_loss": 0.4549614714899685
    },
    {
      "train_step": 7898,
      "sup_loss": 0.7934436596445151
    },
    {
      "train_step": 7899,
      "sup_loss": 0.5407979265375378
    },
    {
      "train_step": 7900,
      "sup_loss": 0.684910713688322
    },
    {
      "train_step": 7901,
      "sup_loss": 0.5388491555534499
    },
    {
      "train_step": 7902,
      "sup_loss": 0.43186705212785226
    },
    {
      "train_step": 7903,
      "sup_loss": 0.632210888721872
    },
    {
      "train_step": 7904,
      "sup_loss": 0.7444752704134321
    },
    {
      "train_step": 7905,
      "sup_loss": 0.6675878603198945
    },
    {
      "train_step": 7906,
      "sup_loss": 0.6426326838608019
    },
    {
      "train_step": 7907,
      "sup_loss": 0.5436803048887164
    },
    {
      "train_step": 7908,
      "sup_loss": 0.5300642657764221
    },
    {
      "train_step": 7909,
      "sup_loss": 0.5609628659580589
    },
    {
      "train_step": 7910,
      "sup_loss": 0.5402549883710877
    },
    {
      "train_step": 7911,
      "sup_loss": 0.5411059095868059
    },
    {
      "train_step": 7912,
      "sup_loss": 0.48827897387912766
    },
    {
      "train_step": 7913,
      "sup_loss": 0.7122420053791912
    },
    {
      "train_step": 7914,
      "sup_loss": 0.55095190665656
    },
    {
      "train_step": 7915,
      "sup_loss": 0.5523392893574208
    },
    {
      "train_step": 7916,
      "sup_loss": 0.3300631742703449
    },
    {
      "train_step": 7917,
      "sup_loss": 0.38151807345377864
    },
    {
      "train_step": 7918,
      "sup_loss": 0.32098921944546005
    },
    {
      "train_step": 7919,
      "sup_loss": 0.6292626712082315
    },
    {
      "train_step": 7920,
      "sup_loss": 0.7915799848835363
    },
    {
      "train_step": 7921,
      "sup_loss": 0.7356747004405297
    },
    {
      "train_step": 7922,
      "sup_loss": 0.6627713421970584
    },
    {
      "train_step": 7923,
      "sup_loss": 0.5280650263861935
    },
    {
      "train_step": 7924,
      "sup_loss": 0.44659158669869947
    },
    {
      "train_step": 7925,
      "sup_loss": 0.6059923210618378
    },
    {
      "train_step": 7926,
      "sup_loss": 0.6174469455991042
    },
    {
      "train_step": 7927,
      "sup_loss": 0.5876677148178054
    },
    {
      "train_step": 7928,
      "sup_loss": 0.5687036248847717
    },
    {
      "train_step": 7929,
      "sup_loss": 0.6765408918848116
    },
    {
      "train_step": 7930,
      "sup_loss": 0.7605280210549876
    },
    {
      "train_step": 7931,
      "sup_loss": 0.24340114752690403
    },
    {
      "train_step": 7932,
      "sup_loss": 0.6754878847265906
    },
    {
      "train_step": 7933,
      "sup_loss": 0.5405553581667535
    },
    {
      "train_step": 7934,
      "sup_loss": 0.5419377131087423
    },
    {
      "train_step": 7935,
      "sup_loss": 0.5437161831451661
    },
    {
      "train_step": 7936,
      "sup_loss": 0.4971703313413309
    },
    {
      "train_step": 7937,
      "sup_loss": 0.6527628972119405
    },
    {
      "train_step": 7938,
      "sup_loss": 0.7707261002086764
    },
    {
      "train_step": 7939,
      "sup_loss": 0.5330782828653416
    },
    {
      "train_step": 7940,
      "sup_loss": 0.6224741981632975
    },
    {
      "train_step": 7941,
      "sup_loss": 0.6863331857031242
    },
    {
      "train_step": 7942,
      "sup_loss": 0.7514168200764852
    },
    {
      "train_step": 7943,
      "sup_loss": 0.5597512702972669
    },
    {
      "train_step": 7944,
      "sup_loss": 0.4117320753975926
    },
    {
      "train_step": 7945,
      "sup_loss": 0.4798721878389399
    },
    {
      "train_step": 7946,
      "sup_loss": 0.5600374548613349
    },
    {
      "train_step": 7947,
      "sup_loss": 0.7564691129803918
    },
    {
      "train_step": 7948,
      "sup_loss": 0.5787269456785219
    },
    {
      "train_step": 7949,
      "sup_loss": 0.7261821094490161
    },
    {
      "train_step": 7950,
      "sup_loss": 0.5696428396652725
    },
    {
      "train_step": 7951,
      "sup_loss": 0.6173816285476514
    },
    {
      "train_step": 7952,
      "sup_loss": 0.6594278741052385
    },
    {
      "train_step": 7953,
      "sup_loss": 0.45807768640149804
    },
    {
      "train_step": 7954,
      "sup_loss": 0.43562429135099007
    },
    {
      "train_step": 7955,
      "sup_loss": 0.43899822744510936
    },
    {
      "train_step": 7956,
      "sup_loss": 0.5261201626069999
    },
    {
      "train_step": 7957,
      "sup_loss": 0.48085792143504363
    },
    {
      "train_step": 7958,
      "sup_loss": 0.6470630881734873
    },
    {
      "train_step": 7959,
      "sup_loss": 0.5635447890641336
    },
    {
      "train_step": 7960,
      "sup_loss": 0.48840599113991334
    },
    {
      "train_step": 7961,
      "sup_loss": 0.5803245075097615
    },
    {
      "train_step": 7962,
      "sup_loss": 0.44060611361573593
    },
    {
      "train_step": 7963,
      "sup_loss": 0.7509534084938376
    },
    {
      "train_step": 7964,
      "sup_loss": 0.5689521860304753
    },
    {
      "train_step": 7965,
      "sup_loss": 0.34811145499537377
    },
    {
      "train_step": 7966,
      "sup_loss": 0.5477338714156778
    },
    {
      "train_step": 7967,
      "sup_loss": 0.47919843972799914
    },
    {
      "train_step": 7968,
      "sup_loss": 0.687450185459945
    },
    {
      "train_step": 7969,
      "sup_loss": 0.6439399758766046
    },
    {
      "train_step": 7970,
      "sup_loss": 0.3339942296342729
    },
    {
      "train_step": 7971,
      "sup_loss": 0.8031573286302378
    },
    {
      "train_step": 7972,
      "sup_loss": 0.5114933854171638
    },
    {
      "train_step": 7973,
      "sup_loss": 0.48379632299264164
    },
    {
      "train_step": 7974,
      "sup_loss": 0.3655299246500539
    },
    {
      "train_step": 7975,
      "sup_loss": 0.7050321285894496
    },
    {
      "train_step": 7976,
      "sup_loss": 0.6080998993369464
    },
    {
      "train_step": 7977,
      "sup_loss": 0.769528020330123
    },
    {
      "train_step": 7978,
      "sup_loss": 0.6073404865669544
    },
    {
      "train_step": 7979,
      "sup_loss": 0.6093002336907577
    },
    {
      "train_step": 7980,
      "sup_loss": 0.4694243333006985
    },
    {
      "train_step": 7981,
      "sup_loss": 0.43222234026500167
    },
    {
      "train_step": 7982,
      "sup_loss": 0.6265349368304991
    },
    {
      "train_step": 7983,
      "sup_loss": 0.5553602893918939
    },
    {
      "train_step": 7984,
      "sup_loss": 0.4937616747444801
    },
    {
      "train_step": 7985,
      "sup_loss": 0.3754699314943506
    },
    {
      "train_step": 7986,
      "sup_loss": 0.41627716804501935
    },
    {
      "train_step": 7987,
      "sup_loss": 0.4072030038699378
    },
    {
      "train_step": 7988,
      "sup_loss": 0.5297325914100922
    },
    {
      "train_step": 7989,
      "sup_loss": 0.22861956645650747
    },
    {
      "train_step": 7990,
      "sup_loss": 0.5798275333610686
    },
    {
      "train_step": 7991,
      "sup_loss": 0.7643269778261098
    },
    {
      "train_step": 7992,
      "sup_loss": 0.5369825420974434
    },
    {
      "train_step": 7993,
      "sup_loss": 0.5766872537530695
    },
    {
      "train_step": 7994,
      "sup_loss": 0.4746605674602441
    },
    {
      "train_step": 7995,
      "sup_loss": 0.5517590829888
    },
    {
      "train_step": 7996,
      "sup_loss": 0.765251230705021
    },
    {
      "train_step": 7997,
      "sup_loss": 0.5729283945226507
    },
    {
      "train_step": 7998,
      "sup_loss": 0.5751715668030746
    },
    {
      "train_step": 7999,
      "sup_loss": 0.5487401777335906
    },
    {
      "train_step": 8000,
      "sup_loss": 0.5895432572308799
    },
    {
      "train_step": 8001,
      "sup_loss": 0.6392326388030051
    },
    {
      "train_step": 8002,
      "sup_loss": 0.7930966541732315
    },
    {
      "train_step": 8003,
      "sup_loss": 0.4251192911212727
    },
    {
      "train_step": 8004,
      "sup_loss": 0.6140131604096783
    },
    {
      "train_step": 8005,
      "sup_loss": 0.39248429415065794
    },
    {
      "train_step": 8006,
      "sup_loss": 0.595711241247059
    },
    {
      "train_step": 8007,
      "sup_loss": 0.3450733001038072
    },
    {
      "train_step": 8008,
      "sup_loss": 0.5008250077950207
    },
    {
      "train_step": 8009,
      "sup_loss": 0.4857822136043106
    },
    {
      "train_step": 8010,
      "sup_loss": 0.6654769852072466
    },
    {
      "train_step": 8011,
      "sup_loss": 0.5697058604481767
    },
    {
      "train_step": 8012,
      "sup_loss": 0.714396544446984
    },
    {
      "train_step": 8013,
      "sup_loss": 0.6162780642784109
    },
    {
      "train_step": 8014,
      "sup_loss": 0.3282781908966876
    },
    {
      "train_step": 8015,
      "sup_loss": 0.5954126502512472
    },
    {
      "train_step": 8016,
      "sup_loss": 0.6407175948853272
    },
    {
      "train_step": 8017,
      "sup_loss": 0.38094556776165134
    },
    {
      "train_step": 8018,
      "sup_loss": 0.5977516160552663
    },
    {
      "train_step": 8019,
      "sup_loss": 0.7468597227834931
    },
    {
      "train_step": 8020,
      "sup_loss": 0.6120076022106009
    },
    {
      "train_step": 8021,
      "sup_loss": 0.7292409903475888
    },
    {
      "train_step": 8022,
      "sup_loss": 0.6597478968620587
    },
    {
      "train_step": 8023,
      "sup_loss": 0.6238144843862865
    },
    {
      "train_step": 8024,
      "sup_loss": 0.5123615783018843
    },
    {
      "train_step": 8025,
      "sup_loss": 0.300647138618286
    },
    {
      "train_step": 8026,
      "sup_loss": 0.5461486254892354
    },
    {
      "train_step": 8027,
      "sup_loss": 0.6095673493288961
    },
    {
      "train_step": 8028,
      "sup_loss": 0.5023694219347667
    },
    {
      "train_step": 8029,
      "sup_loss": 0.5853242964024259
    },
    {
      "train_step": 8030,
      "sup_loss": 0.5642009615611141
    },
    {
      "train_step": 8031,
      "sup_loss": 0.49413983193598615
    },
    {
      "train_step": 8032,
      "sup_loss": 0.6890333371514518
    },
    {
      "train_step": 8033,
      "sup_loss": 0.5389236516143096
    },
    {
      "train_step": 8034,
      "sup_loss": 0.4153985093329982
    },
    {
      "train_step": 8035,
      "sup_loss": 0.5847291510393283
    },
    {
      "train_step": 8036,
      "sup_loss": 0.6403809062626555
    },
    {
      "train_step": 8037,
      "sup_loss": 0.7455424488360729
    },
    {
      "train_step": 8038,
      "sup_loss": 0.3434253421646546
    },
    {
      "train_step": 8039,
      "sup_loss": 0.6487086987655416
    },
    {
      "train_step": 8040,
      "sup_loss": 0.481448271840656
    },
    {
      "train_step": 8041,
      "sup_loss": 0.6798292387470882
    },
    {
      "train_step": 8042,
      "sup_loss": 0.4986984859805174
    },
    {
      "train_step": 8043,
      "sup_loss": 0.604066139810507
    },
    {
      "train_step": 8044,
      "sup_loss": 0.5902122585899393
    },
    {
      "train_step": 8045,
      "sup_loss": 0.688089124090238
    },
    {
      "train_step": 8046,
      "sup_loss": 0.6615153530212059
    },
    {
      "train_step": 8047,
      "sup_loss": 0.4385389955179033
    },
    {
      "train_step": 8048,
      "sup_loss": 0.6663090565719922
    },
    {
      "train_step": 8049,
      "sup_loss": 0.4533133365118118
    },
    {
      "train_step": 8050,
      "sup_loss": 0.5910137619110003
    },
    {
      "train_step": 8051,
      "sup_loss": 0.47506780058928433
    },
    {
      "train_step": 8052,
      "sup_loss": 0.3164764785375409
    },
    {
      "train_step": 8053,
      "sup_loss": 0.5708381918052498
    },
    {
      "train_step": 8054,
      "sup_loss": 0.6837340013036609
    },
    {
      "train_step": 8055,
      "sup_loss": 0.42450765428622805
    },
    {
      "train_step": 8056,
      "sup_loss": 0.7823188086180505
    },
    {
      "train_step": 8057,
      "sup_loss": 0.5660443519938034
    },
    {
      "train_step": 8058,
      "sup_loss": 0.43858102273874905
    },
    {
      "train_step": 8059,
      "sup_loss": 0.30642161002822266
    },
    {
      "train_step": 8060,
      "sup_loss": 0.6200402808203929
    },
    {
      "train_step": 8061,
      "sup_loss": 0.6157960308197417
    },
    {
      "train_step": 8062,
      "sup_loss": 0.5521928983137108
    },
    {
      "train_step": 8063,
      "sup_loss": 0.6575641936319123
    },
    {
      "train_step": 8064,
      "sup_loss": 0.6689095654372189
    },
    {
      "train_step": 8065,
      "sup_loss": 0.6076127942723145
    },
    {
      "train_step": 8066,
      "sup_loss": 0.6046058046816125
    },
    {
      "train_step": 8067,
      "sup_loss": 0.48475237965521595
    },
    {
      "train_step": 8068,
      "sup_loss": 0.7308213870557198
    },
    {
      "train_step": 8069,
      "sup_loss": 0.3309008334806002
    },
    {
      "train_step": 8070,
      "sup_loss": 0.3681974663554666
    },
    {
      "train_step": 8071,
      "sup_loss": 0.6879500138081595
    },
    {
      "train_step": 8072,
      "sup_loss": 0.4947877602838252
    },
    {
      "train_step": 8073,
      "sup_loss": 0.5137763389936791
    },
    {
      "train_step": 8074,
      "sup_loss": 0.5974928055378491
    },
    {
      "train_step": 8075,
      "sup_loss": 0.5086029009473809
    },
    {
      "train_step": 8076,
      "sup_loss": 0.5353193202207601
    },
    {
      "train_step": 8077,
      "sup_loss": 0.43566998469139634
    },
    {
      "train_step": 8078,
      "sup_loss": 0.6335784430640621
    },
    {
      "train_step": 8079,
      "sup_loss": 0.40831805645941116
    },
    {
      "train_step": 8080,
      "sup_loss": 0.6733032213347676
    },
    {
      "train_step": 8081,
      "sup_loss": 0.5598340400632554
    },
    {
      "train_step": 8082,
      "sup_loss": 0.5501446464474641
    },
    {
      "train_step": 8083,
      "sup_loss": 0.6356696984339478
    },
    {
      "train_step": 8084,
      "sup_loss": 0.6472395820908357
    },
    {
      "train_step": 8085,
      "sup_loss": 0.6620734554953512
    },
    {
      "train_step": 8086,
      "sup_loss": 0.3491382347340962
    },
    {
      "train_step": 8087,
      "sup_loss": 0.7140582713938506
    },
    {
      "train_step": 8088,
      "sup_loss": 0.6475426288116186
    },
    {
      "train_step": 8089,
      "sup_loss": 0.6347531209205781
    },
    {
      "train_step": 8090,
      "sup_loss": 0.6156897818639137
    },
    {
      "train_step": 8091,
      "sup_loss": 0.42357886217066254
    },
    {
      "train_step": 8092,
      "sup_loss": 0.5469705921941534
    },
    {
      "train_step": 8093,
      "sup_loss": 0.5709901495423011
    },
    {
      "train_step": 8094,
      "sup_loss": 0.5040801818501582
    },
    {
      "train_step": 8095,
      "sup_loss": 0.5902310752401894
    },
    {
      "train_step": 8096,
      "sup_loss": 0.6388530224283995
    },
    {
      "train_step": 8097,
      "sup_loss": 0.740561970469857
    },
    {
      "train_step": 8098,
      "sup_loss": 0.3788475854650119
    },
    {
      "train_step": 8099,
      "sup_loss": 0.8140507717049897
    },
    {
      "train_step": 8100,
      "sup_loss": 0.32695455222099457
    },
    {
      "train_step": 8101,
      "sup_loss": 0.70280425518829
    },
    {
      "train_step": 8102,
      "sup_loss": 0.7746607972284937
    },
    {
      "train_step": 8103,
      "sup_loss": 0.5120996178744406
    },
    {
      "train_step": 8104,
      "sup_loss": 0.3612114820326972
    },
    {
      "train_step": 8105,
      "sup_loss": 0.3335679007388663
    },
    {
      "train_step": 8106,
      "sup_loss": 0.3857956546086786
    },
    {
      "train_step": 8107,
      "sup_loss": 0.5103796610679642
    },
    {
      "train_step": 8108,
      "sup_loss": 0.4799550112182459
    },
    {
      "train_step": 8109,
      "sup_loss": 0.5599869483683791
    },
    {
      "train_step": 8110,
      "sup_loss": 0.4964666356227954
    },
    {
      "train_step": 8111,
      "sup_loss": 0.477760396373655
    },
    {
      "train_step": 8112,
      "sup_loss": 0.9454473362966156
    },
    {
      "train_step": 8113,
      "sup_loss": 0.5151613795633243
    },
    {
      "train_step": 8114,
      "sup_loss": 0.4720773046703158
    },
    {
      "train_step": 8115,
      "sup_loss": 0.5392094585297404
    },
    {
      "train_step": 8116,
      "sup_loss": 0.42568770359577496
    },
    {
      "train_step": 8117,
      "sup_loss": 0.5588069461587699
    },
    {
      "train_step": 8118,
      "sup_loss": 0.5024854009526196
    },
    {
      "train_step": 8119,
      "sup_loss": 0.42389421853877235
    },
    {
      "train_step": 8120,
      "sup_loss": 0.6408962402612293
    },
    {
      "train_step": 8121,
      "sup_loss": 0.7428943605010458
    },
    {
      "train_step": 8122,
      "sup_loss": 0.34452720933008446
    },
    {
      "train_step": 8123,
      "sup_loss": 0.6864356691110158
    },
    {
      "train_step": 8124,
      "sup_loss": 0.574999859612727
    },
    {
      "train_step": 8125,
      "sup_loss": 0.6542475492297344
    },
    {
      "train_step": 8126,
      "sup_loss": 0.4841795467584918
    },
    {
      "train_step": 8127,
      "sup_loss": 0.469959304864175
    },
    {
      "train_step": 8128,
      "sup_loss": 0.6131612490095631
    },
    {
      "train_step": 8129,
      "sup_loss": 0.4699567167437876
    },
    {
      "train_step": 8130,
      "sup_loss": 0.5979384365478084
    },
    {
      "train_step": 8131,
      "sup_loss": 0.5694884923693705
    },
    {
      "train_step": 8132,
      "sup_loss": 0.4619876340570748
    },
    {
      "train_step": 8133,
      "sup_loss": 0.3476390878388886
    },
    {
      "train_step": 8134,
      "sup_loss": 0.5464836830387203
    },
    {
      "train_step": 8135,
      "sup_loss": 0.2990616396527566
    },
    {
      "train_step": 8136,
      "sup_loss": 0.4847713762370699
    },
    {
      "train_step": 8137,
      "sup_loss": 0.5550950638343716
    },
    {
      "train_step": 8138,
      "sup_loss": 0.578901143844976
    },
    {
      "train_step": 8139,
      "sup_loss": 0.4962468910478174
    },
    {
      "train_step": 8140,
      "sup_loss": 0.5092231221507894
    },
    {
      "train_step": 8141,
      "sup_loss": 0.6542822793684906
    },
    {
      "train_step": 8142,
      "sup_loss": 0.6945874573028502
    },
    {
      "train_step": 8143,
      "sup_loss": 0.6568527692740579
    },
    {
      "train_step": 8144,
      "sup_loss": 0.5545348191623691
    },
    {
      "train_step": 8145,
      "sup_loss": 0.31954752610604953
    },
    {
      "train_step": 8146,
      "sup_loss": 0.43042386650758896
    },
    {
      "train_step": 8147,
      "sup_loss": 0.7626626115914594
    },
    {
      "train_step": 8148,
      "sup_loss": 0.696167281307046
    },
    {
      "train_step": 8149,
      "sup_loss": 0.46835365951553015
    },
    {
      "train_step": 8150,
      "sup_loss": 0.5328053472595857
    },
    {
      "train_step": 8151,
      "sup_loss": 0.6571772344250582
    },
    {
      "train_step": 8152,
      "sup_loss": 0.5609359200367752
    },
    {
      "train_step": 8153,
      "sup_loss": 0.6745173762421436
    },
    {
      "train_step": 8154,
      "sup_loss": 0.5186230844597878
    },
    {
      "train_step": 8155,
      "sup_loss": 0.6356318407335241
    },
    {
      "train_step": 8156,
      "sup_loss": 0.5855735840935855
    },
    {
      "train_step": 8157,
      "sup_loss": 0.6607086241905902
    },
    {
      "train_step": 8158,
      "sup_loss": 0.5034322660368552
    },
    {
      "train_step": 8159,
      "sup_loss": 0.464802601548616
    },
    {
      "train_step": 8160,
      "sup_loss": 0.5830852305440519
    },
    {
      "train_step": 8161,
      "sup_loss": 0.6674122536338911
    },
    {
      "train_step": 8162,
      "sup_loss": 0.5476032836035551
    },
    {
      "train_step": 8163,
      "sup_loss": 0.6775768501998799
    },
    {
      "train_step": 8164,
      "sup_loss": 0.5561016158267269
    },
    {
      "train_step": 8165,
      "sup_loss": 0.5003131917070456
    },
    {
      "train_step": 8166,
      "sup_loss": 0.6558642493922703
    },
    {
      "train_step": 8167,
      "sup_loss": 0.2973972168732294
    },
    {
      "train_step": 8168,
      "sup_loss": 0.5477325553605384
    },
    {
      "train_step": 8169,
      "sup_loss": 0.6270805520432421
    },
    {
      "train_step": 8170,
      "sup_loss": 0.5625935427652067
    },
    {
      "train_step": 8171,
      "sup_loss": 0.6786737830874229
    },
    {
      "train_step": 8172,
      "sup_loss": 0.36432188370294716
    },
    {
      "train_step": 8173,
      "sup_loss": 0.570570829714028
    },
    {
      "train_step": 8174,
      "sup_loss": 0.3824850054527291
    },
    {
      "train_step": 8175,
      "sup_loss": 0.47004296028693593
    },
    {
      "train_step": 8176,
      "sup_loss": 0.6477652936763293
    },
    {
      "train_step": 8177,
      "sup_loss": 0.5144342060916878
    },
    {
      "train_step": 8178,
      "sup_loss": 0.7528307497862741
    },
    {
      "train_step": 8179,
      "sup_loss": 0.8813381680235035
    },
    {
      "train_step": 8180,
      "sup_loss": 0.6241491550522527
    },
    {
      "train_step": 8181,
      "sup_loss": 0.5897078064994083
    },
    {
      "train_step": 8182,
      "sup_loss": 0.48190098806368986
    },
    {
      "train_step": 8183,
      "sup_loss": 0.5613781614944979
    },
    {
      "train_step": 8184,
      "sup_loss": 0.40627191503138665
    },
    {
      "train_step": 8185,
      "sup_loss": 0.7206198732082684
    },
    {
      "train_step": 8186,
      "sup_loss": 0.4854316674188558
    },
    {
      "train_step": 8187,
      "sup_loss": 0.750016566183475
    },
    {
      "train_step": 8188,
      "sup_loss": 0.5229019235857062
    },
    {
      "train_step": 8189,
      "sup_loss": 0.7847311610564017
    },
    {
      "train_step": 8190,
      "sup_loss": 0.6751466555129967
    },
    {
      "train_step": 8191,
      "sup_loss": 0.5081232690447299
    },
    {
      "train_step": 8192,
      "sup_loss": 0.6660847298418122
    },
    {
      "train_step": 8193,
      "sup_loss": 0.5969995132349525
    },
    {
      "train_step": 8194,
      "sup_loss": 0.4533157119480019
    },
    {
      "train_step": 8195,
      "sup_loss": 0.3452457552926597
    },
    {
      "train_step": 8196,
      "sup_loss": 0.3710869373115407
    },
    {
      "train_step": 8197,
      "sup_loss": 0.5160018841339568
    },
    {
      "train_step": 8198,
      "sup_loss": 0.6789632703794561
    },
    {
      "train_step": 8199,
      "sup_loss": 0.5323153400284474
    },
    {
      "train_step": 8200,
      "sup_loss": 0.7386101665058764
    },
    {
      "train_step": 8201,
      "sup_loss": 0.5126714051273931
    },
    {
      "train_step": 8202,
      "sup_loss": 0.6669560930774387
    },
    {
      "train_step": 8203,
      "sup_loss": 0.5101912340860603
    },
    {
      "train_step": 8204,
      "sup_loss": 0.6564150885597817
    },
    {
      "train_step": 8205,
      "sup_loss": 0.6820461944609141
    },
    {
      "train_step": 8206,
      "sup_loss": 0.648045253634748
    },
    {
      "train_step": 8207,
      "sup_loss": 0.4946383290411554
    },
    {
      "train_step": 8208,
      "sup_loss": 0.4468898438186492
    },
    {
      "train_step": 8209,
      "sup_loss": 0.7078855074519025
    },
    {
      "train_step": 8210,
      "sup_loss": 0.5212353742572284
    },
    {
      "train_step": 8211,
      "sup_loss": 0.3901164054574628
    },
    {
      "train_step": 8212,
      "sup_loss": 0.5442991274627975
    },
    {
      "train_step": 8213,
      "sup_loss": 0.6005807695072117
    },
    {
      "train_step": 8214,
      "sup_loss": 0.7023180382353558
    },
    {
      "train_step": 8215,
      "sup_loss": 0.8919450932979188
    },
    {
      "train_step": 8216,
      "sup_loss": 0.5554270543323456
    },
    {
      "train_step": 8217,
      "sup_loss": 0.7759480516097557
    },
    {
      "train_step": 8218,
      "sup_loss": 0.5642233889094477
    },
    {
      "train_step": 8219,
      "sup_loss": 0.3906466166265861
    },
    {
      "train_step": 8220,
      "sup_loss": 0.5837611303756516
    },
    {
      "train_step": 8221,
      "sup_loss": 0.4074395037531821
    },
    {
      "train_step": 8222,
      "sup_loss": 0.5388033997211369
    },
    {
      "train_step": 8223,
      "sup_loss": 0.5091354618927972
    },
    {
      "train_step": 8224,
      "sup_loss": 0.5613401556125971
    },
    {
      "train_step": 8225,
      "sup_loss": 0.6719791938464572
    },
    {
      "train_step": 8226,
      "sup_loss": 0.5888758855247999
    },
    {
      "train_step": 8227,
      "sup_loss": 0.45012371787381056
    },
    {
      "train_step": 8228,
      "sup_loss": 0.53956766968447
    },
    {
      "train_step": 8229,
      "sup_loss": 0.597329834776582
    },
    {
      "train_step": 8230,
      "sup_loss": 0.6062341106077637
    },
    {
      "train_step": 8231,
      "sup_loss": 0.5118890988563876
    },
    {
      "train_step": 8232,
      "sup_loss": 0.6159300073043527
    },
    {
      "train_step": 8233,
      "sup_loss": 0.5365908603382407
    },
    {
      "train_step": 8234,
      "sup_loss": 0.6403744874145827
    },
    {
      "train_step": 8235,
      "sup_loss": 0.6173178632563127
    },
    {
      "train_step": 8236,
      "sup_loss": 0.7108543638965625
    },
    {
      "train_step": 8237,
      "sup_loss": 0.6328845871370862
    },
    {
      "train_step": 8238,
      "sup_loss": 0.6690686279406374
    },
    {
      "train_step": 8239,
      "sup_loss": 0.5363547163740422
    },
    {
      "train_step": 8240,
      "sup_loss": 0.19917712674082946
    },
    {
      "train_step": 8241,
      "sup_loss": 0.5173006697808581
    },
    {
      "train_step": 8242,
      "sup_loss": 0.5524367073931341
    },
    {
      "train_step": 8243,
      "sup_loss": 0.45603663255508486
    },
    {
      "train_step": 8244,
      "sup_loss": 0.5662121020582417
    },
    {
      "train_step": 8245,
      "sup_loss": 0.57210675897108
    },
    {
      "train_step": 8246,
      "sup_loss": 0.5637648975747378
    },
    {
      "train_step": 8247,
      "sup_loss": 0.363106716325931
    },
    {
      "train_step": 8248,
      "sup_loss": 0.365989038723766
    },
    {
      "train_step": 8249,
      "sup_loss": 0.5416478239342195
    },
    {
      "train_step": 8250,
      "sup_loss": 0.6892929838791441
    },
    {
      "train_step": 8251,
      "sup_loss": 0.559191855542741
    },
    {
      "train_step": 8252,
      "sup_loss": 0.34382495352286524
    },
    {
      "train_step": 8253,
      "sup_loss": 0.7105480009430027
    },
    {
      "train_step": 8254,
      "sup_loss": 0.6031919707773694
    },
    {
      "train_step": 8255,
      "sup_loss": 0.6774038598688213
    },
    {
      "train_step": 8256,
      "sup_loss": 0.512633555402585
    },
    {
      "train_step": 8257,
      "sup_loss": 0.6957493292327231
    },
    {
      "train_step": 8258,
      "sup_loss": 0.7121050235245128
    },
    {
      "train_step": 8259,
      "sup_loss": 0.6011079500716809
    },
    {
      "train_step": 8260,
      "sup_loss": 0.4444945190742007
    },
    {
      "train_step": 8261,
      "sup_loss": 0.5976011602382958
    },
    {
      "train_step": 8262,
      "sup_loss": 0.4819360675073736
    },
    {
      "train_step": 8263,
      "sup_loss": 0.6840213990552962
    },
    {
      "train_step": 8264,
      "sup_loss": 0.621483175522317
    },
    {
      "train_step": 8265,
      "sup_loss": 0.6916896634942061
    },
    {
      "train_step": 8266,
      "sup_loss": 0.44744626585212577
    },
    {
      "train_step": 8267,
      "sup_loss": 0.24924432292277282
    },
    {
      "train_step": 8268,
      "sup_loss": 0.5923041020867827
    },
    {
      "train_step": 8269,
      "sup_loss": 0.4458762445153512
    },
    {
      "train_step": 8270,
      "sup_loss": 0.28359336358855586
    },
    {
      "train_step": 8271,
      "sup_loss": 0.46957866831052214
    },
    {
      "train_step": 8272,
      "sup_loss": 0.6455098901801003
    },
    {
      "train_step": 8273,
      "sup_loss": 0.5991225210209834
    },
    {
      "train_step": 8274,
      "sup_loss": 0.5815661299339225
    },
    {
      "train_step": 8275,
      "sup_loss": 0.44181357331261517
    },
    {
      "train_step": 8276,
      "sup_loss": 0.5781241044423113
    },
    {
      "train_step": 8277,
      "sup_loss": 0.5819289956915511
    },
    {
      "train_step": 8278,
      "sup_loss": 0.6337467066588612
    },
    {
      "train_step": 8279,
      "sup_loss": 0.41204005910092933
    },
    {
      "train_step": 8280,
      "sup_loss": 0.6428283251757518
    },
    {
      "train_step": 8281,
      "sup_loss": 0.33536581110266633
    },
    {
      "train_step": 8282,
      "sup_loss": 0.5790506333192281
    },
    {
      "train_step": 8283,
      "sup_loss": 0.5496428132942895
    },
    {
      "train_step": 8284,
      "sup_loss": 0.7320141265448266
    },
    {
      "train_step": 8285,
      "sup_loss": 0.6332869486758596
    },
    {
      "train_step": 8286,
      "sup_loss": 0.43972737891640445
    },
    {
      "train_step": 8287,
      "sup_loss": 0.4866319891337471
    },
    {
      "train_step": 8288,
      "sup_loss": 0.5484755321287159
    },
    {
      "train_step": 8289,
      "sup_loss": 0.6633167164234247
    },
    {
      "train_step": 8290,
      "sup_loss": 0.5394046821379884
    },
    {
      "train_step": 8291,
      "sup_loss": 0.4837254660560131
    },
    {
      "train_step": 8292,
      "sup_loss": 0.6012211449262086
    },
    {
      "train_step": 8293,
      "sup_loss": 0.5127462242271841
    },
    {
      "train_step": 8294,
      "sup_loss": 0.5481576994168716
    },
    {
      "train_step": 8295,
      "sup_loss": 0.5651370255508263
    },
    {
      "train_step": 8296,
      "sup_loss": 0.4896889087054951
    },
    {
      "train_step": 8297,
      "sup_loss": 0.629702202743576
    },
    {
      "train_step": 8298,
      "sup_loss": 0.6541228608691032
    },
    {
      "train_step": 8299,
      "sup_loss": 0.689912941631112
    },
    {
      "train_step": 8300,
      "sup_loss": 0.5351635429603869
    },
    {
      "train_step": 8301,
      "sup_loss": 0.7071705814239376
    },
    {
      "train_step": 8302,
      "sup_loss": 0.6805710409168747
    },
    {
      "train_step": 8303,
      "sup_loss": 0.6570471458439153
    },
    {
      "train_step": 8304,
      "sup_loss": 0.5523073940978461
    },
    {
      "train_step": 8305,
      "sup_loss": 0.7362182363684837
    },
    {
      "train_step": 8306,
      "sup_loss": 0.5715407884114048
    },
    {
      "train_step": 8307,
      "sup_loss": 0.6533971518369875
    },
    {
      "train_step": 8308,
      "sup_loss": 0.5719485592053278
    },
    {
      "train_step": 8309,
      "sup_loss": 0.49302117985054056
    },
    {
      "train_step": 8310,
      "sup_loss": 0.46903871236361266
    },
    {
      "train_step": 8311,
      "sup_loss": 0.513104173620786
    },
    {
      "train_step": 8312,
      "sup_loss": 0.7536821768715175
    },
    {
      "train_step": 8313,
      "sup_loss": 0.43332714833835584
    },
    {
      "train_step": 8314,
      "sup_loss": 0.5864643620939419
    },
    {
      "train_step": 8315,
      "sup_loss": 0.47872935647703796
    },
    {
      "train_step": 8316,
      "sup_loss": 0.6377572106104211
    },
    {
      "train_step": 8317,
      "sup_loss": 0.4602272072832261
    },
    {
      "train_step": 8318,
      "sup_loss": 0.8139952382709623
    },
    {
      "train_step": 8319,
      "sup_loss": 0.7211741755075487
    },
    {
      "train_step": 8320,
      "sup_loss": 0.501485781302831
    },
    {
      "train_step": 8321,
      "sup_loss": 0.42440584156220257
    },
    {
      "train_step": 8322,
      "sup_loss": 0.45782065041113795
    },
    {
      "train_step": 8323,
      "sup_loss": 0.681558174424261
    },
    {
      "train_step": 8324,
      "sup_loss": 0.24634042655036187
    },
    {
      "train_step": 8325,
      "sup_loss": 0.6293811677057088
    },
    {
      "train_step": 8326,
      "sup_loss": 0.575274119315573
    },
    {
      "train_step": 8327,
      "sup_loss": 0.5221163301057677
    },
    {
      "train_step": 8328,
      "sup_loss": 0.5701955939109129
    },
    {
      "train_step": 8329,
      "sup_loss": 0.4206773169581149
    },
    {
      "train_step": 8330,
      "sup_loss": 0.5571002759514834
    },
    {
      "train_step": 8331,
      "sup_loss": 0.43996263565317734
    },
    {
      "train_step": 8332,
      "sup_loss": 0.4303758373653558
    },
    {
      "train_step": 8333,
      "sup_loss": 0.652582934439752
    },
    {
      "train_step": 8334,
      "sup_loss": 0.5229143005337451
    },
    {
      "train_step": 8335,
      "sup_loss": 0.49945066660929316
    },
    {
      "train_step": 8336,
      "sup_loss": 0.5249142647909919
    },
    {
      "train_step": 8337,
      "sup_loss": 0.39680032010730354
    },
    {
      "train_step": 8338,
      "sup_loss": 0.42011809290114127
    },
    {
      "train_step": 8339,
      "sup_loss": 0.6142109453243184
    },
    {
      "train_step": 8340,
      "sup_loss": 0.6233607429956146
    },
    {
      "train_step": 8341,
      "sup_loss": 0.5692698635019149
    },
    {
      "train_step": 8342,
      "sup_loss": 0.43851818008682303
    },
    {
      "train_step": 8343,
      "sup_loss": 0.6422368400587082
    },
    {
      "train_step": 8344,
      "sup_loss": 0.6213766624782628
    },
    {
      "train_step": 8345,
      "sup_loss": 0.4668477315572711
    },
    {
      "train_step": 8346,
      "sup_loss": 0.38234416964847406
    },
    {
      "train_step": 8347,
      "sup_loss": 0.6251946969543669
    },
    {
      "train_step": 8348,
      "sup_loss": 0.37768768638648564
    },
    {
      "train_step": 8349,
      "sup_loss": 0.5052578997893441
    },
    {
      "train_step": 8350,
      "sup_loss": 0.33793013195041177
    },
    {
      "train_step": 8351,
      "sup_loss": 0.5536126024422873
    },
    {
      "train_step": 8352,
      "sup_loss": 0.4434212906450347
    },
    {
      "train_step": 8353,
      "sup_loss": 0.3599917269403676
    },
    {
      "train_step": 8354,
      "sup_loss": 0.7273735125603622
    },
    {
      "train_step": 8355,
      "sup_loss": 0.4515282162033975
    },
    {
      "train_step": 8356,
      "sup_loss": 0.4221628689233934
    },
    {
      "train_step": 8357,
      "sup_loss": 0.6403248387744311
    },
    {
      "train_step": 8358,
      "sup_loss": 0.5235320859546746
    },
    {
      "train_step": 8359,
      "sup_loss": 0.6613566677392795
    },
    {
      "train_step": 8360,
      "sup_loss": 0.6987270461468296
    },
    {
      "train_step": 8361,
      "sup_loss": 0.5537489842140854
    },
    {
      "train_step": 8362,
      "sup_loss": 0.5973097271827962
    },
    {
      "train_step": 8363,
      "sup_loss": 0.5060764168701821
    },
    {
      "train_step": 8364,
      "sup_loss": 0.5369071927352225
    },
    {
      "train_step": 8365,
      "sup_loss": 0.609179674306526
    },
    {
      "train_step": 8366,
      "sup_loss": 0.5119300238754768
    },
    {
      "train_step": 8367,
      "sup_loss": 0.6367301098282978
    },
    {
      "train_step": 8368,
      "sup_loss": 0.30782025541918867
    },
    {
      "train_step": 8369,
      "sup_loss": 0.5636640973398453
    },
    {
      "train_step": 8370,
      "sup_loss": 0.6365797916231472
    },
    {
      "train_step": 8371,
      "sup_loss": 0.5844493812359987
    },
    {
      "train_step": 8372,
      "sup_loss": 0.6379589782094386
    },
    {
      "train_step": 8373,
      "sup_loss": 0.6766755648224615
    },
    {
      "train_step": 8374,
      "sup_loss": 0.6296542306587197
    },
    {
      "train_step": 8375,
      "sup_loss": 0.5586257581944896
    },
    {
      "train_step": 8376,
      "sup_loss": 0.5446768445169445
    },
    {
      "train_step": 8377,
      "sup_loss": 0.49576569496441164
    },
    {
      "train_step": 8378,
      "sup_loss": 0.7446316156666883
    },
    {
      "train_step": 8379,
      "sup_loss": 0.47949238195607036
    },
    {
      "train_step": 8380,
      "sup_loss": 0.6098756048505172
    },
    {
      "train_step": 8381,
      "sup_loss": 0.6562997088119754
    },
    {
      "train_step": 8382,
      "sup_loss": 0.7421933642401665
    },
    {
      "train_step": 8383,
      "sup_loss": 0.6025963801900539
    },
    {
      "train_step": 8384,
      "sup_loss": 0.49093662957918277
    },
    {
      "train_step": 8385,
      "sup_loss": 0.5378556723268803
    },
    {
      "train_step": 8386,
      "sup_loss": 0.5685403704762475
    },
    {
      "train_step": 8387,
      "sup_loss": 0.6480858082612809
    },
    {
      "train_step": 8388,
      "sup_loss": 0.4280353747360991
    },
    {
      "train_step": 8389,
      "sup_loss": 0.6345751987949246
    },
    {
      "train_step": 8390,
      "sup_loss": 0.5076865936776761
    },
    {
      "train_step": 8391,
      "sup_loss": 0.4017240011968537
    },
    {
      "train_step": 8392,
      "sup_loss": 0.501847030494935
    },
    {
      "train_step": 8393,
      "sup_loss": 0.6424964538865826
    },
    {
      "train_step": 8394,
      "sup_loss": 0.5620439762483185
    },
    {
      "train_step": 8395,
      "sup_loss": 0.8367160937939409
    },
    {
      "train_step": 8396,
      "sup_loss": 0.5096114134487687
    },
    {
      "train_step": 8397,
      "sup_loss": 0.43371361771657946
    },
    {
      "train_step": 8398,
      "sup_loss": 0.44999226427616607
    },
    {
      "train_step": 8399,
      "sup_loss": 0.6626763799284856
    },
    {
      "train_step": 8400,
      "sup_loss": 0.4688750892014502
    },
    {
      "train_step": 8401,
      "sup_loss": 0.6127362977335981
    },
    {
      "train_step": 8402,
      "sup_loss": 0.43859531464195967
    },
    {
      "train_step": 8403,
      "sup_loss": 0.522632905072662
    },
    {
      "train_step": 8404,
      "sup_loss": 0.5203749239129413
    },
    {
      "train_step": 8405,
      "sup_loss": 0.5415588394080507
    },
    {
      "train_step": 8406,
      "sup_loss": 0.8009771218093976
    },
    {
      "train_step": 8407,
      "sup_loss": 0.6422387864974038
    },
    {
      "train_step": 8408,
      "sup_loss": 0.5603224031489281
    },
    {
      "train_step": 8409,
      "sup_loss": 0.6267433767655595
    },
    {
      "train_step": 8410,
      "sup_loss": 0.5556629796153923
    },
    {
      "train_step": 8411,
      "sup_loss": 0.6155149629965094
    },
    {
      "train_step": 8412,
      "sup_loss": 0.6305562299495311
    },
    {
      "train_step": 8413,
      "sup_loss": 0.6377267272594638
    },
    {
      "train_step": 8414,
      "sup_loss": 0.4925185443239376
    },
    {
      "train_step": 8415,
      "sup_loss": 0.6196030917111972
    },
    {
      "train_step": 8416,
      "sup_loss": 0.7576150049195703
    },
    {
      "train_step": 8417,
      "sup_loss": 0.6997972621680061
    },
    {
      "train_step": 8418,
      "sup_loss": 0.5170448180442264
    },
    {
      "train_step": 8419,
      "sup_loss": 0.5570529163081066
    },
    {
      "train_step": 8420,
      "sup_loss": 0.6180902119491231
    },
    {
      "train_step": 8421,
      "sup_loss": 0.5666191486222708
    },
    {
      "train_step": 8422,
      "sup_loss": 0.4846218115984966
    },
    {
      "train_step": 8423,
      "sup_loss": 0.3007747373320824
    },
    {
      "train_step": 8424,
      "sup_loss": 0.6931628903545974
    },
    {
      "train_step": 8425,
      "sup_loss": 0.5751996396409103
    },
    {
      "train_step": 8426,
      "sup_loss": 0.47251235981715983
    },
    {
      "train_step": 8427,
      "sup_loss": 0.6838405417870297
    },
    {
      "train_step": 8428,
      "sup_loss": 0.6574446997439769
    },
    {
      "train_step": 8429,
      "sup_loss": 0.5972372552617301
    },
    {
      "train_step": 8430,
      "sup_loss": 0.4773231247935872
    },
    {
      "train_step": 8431,
      "sup_loss": 0.5866645456684171
    },
    {
      "train_step": 8432,
      "sup_loss": 0.33179755064288813
    },
    {
      "train_step": 8433,
      "sup_loss": 0.38447624151845505
    },
    {
      "train_step": 8434,
      "sup_loss": 0.4931259469349481
    },
    {
      "train_step": 8435,
      "sup_loss": 0.5739664618293909
    },
    {
      "train_step": 8436,
      "sup_loss": 0.6104618969115367
    },
    {
      "train_step": 8437,
      "sup_loss": 0.559682314448493
    },
    {
      "train_step": 8438,
      "sup_loss": 0.35921658131797274
    },
    {
      "train_step": 8439,
      "sup_loss": 0.589420246005773
    },
    {
      "train_step": 8440,
      "sup_loss": 0.5457054192505107
    },
    {
      "train_step": 8441,
      "sup_loss": 0.5701767820016571
    },
    {
      "train_step": 8442,
      "sup_loss": 0.7580714311862166
    },
    {
      "train_step": 8443,
      "sup_loss": 0.5005270666071066
    },
    {
      "train_step": 8444,
      "sup_loss": 0.7599855518761667
    },
    {
      "train_step": 8445,
      "sup_loss": 0.6182838714627078
    },
    {
      "train_step": 8446,
      "sup_loss": 0.549524638748472
    },
    {
      "train_step": 8447,
      "sup_loss": 0.4505976413957614
    },
    {
      "train_step": 8448,
      "sup_loss": 0.5468922389006058
    },
    {
      "train_step": 8449,
      "sup_loss": 0.5044350567279784
    },
    {
      "train_step": 8450,
      "sup_loss": 0.5008097224640006
    },
    {
      "train_step": 8451,
      "sup_loss": 0.5838465287542786
    },
    {
      "train_step": 8452,
      "sup_loss": 0.4976980874853243
    },
    {
      "train_step": 8453,
      "sup_loss": 0.561359349015663
    },
    {
      "train_step": 8454,
      "sup_loss": 0.624577671083083
    },
    {
      "train_step": 8455,
      "sup_loss": 0.5682057606737091
    },
    {
      "train_step": 8456,
      "sup_loss": 0.4432536406316328
    },
    {
      "train_step": 8457,
      "sup_loss": 0.6801542496386853
    },
    {
      "train_step": 8458,
      "sup_loss": 0.6136962549169493
    },
    {
      "train_step": 8459,
      "sup_loss": 0.7839675849482117
    },
    {
      "train_step": 8460,
      "sup_loss": 0.6057451860024515
    },
    {
      "train_step": 8461,
      "sup_loss": 0.6164000545935473
    },
    {
      "train_step": 8462,
      "sup_loss": 0.724842057503461
    },
    {
      "train_step": 8463,
      "sup_loss": 0.674283474248937
    },
    {
      "train_step": 8464,
      "sup_loss": 0.4506586226871195
    },
    {
      "train_step": 8465,
      "sup_loss": 0.540454652126859
    },
    {
      "train_step": 8466,
      "sup_loss": 0.3315500303570115
    },
    {
      "train_step": 8467,
      "sup_loss": 0.47284289672828955
    },
    {
      "train_step": 8468,
      "sup_loss": 0.1750233167764244
    },
    {
      "train_step": 8469,
      "sup_loss": 0.47239820788419346
    },
    {
      "train_step": 8470,
      "sup_loss": 0.5975780542834785
    },
    {
      "train_step": 8471,
      "sup_loss": 0.5443984127891073
    },
    {
      "train_step": 8472,
      "sup_loss": 0.6193281625138674
    },
    {
      "train_step": 8473,
      "sup_loss": 0.6555745362454779
    },
    {
      "train_step": 8474,
      "sup_loss": 0.7242829624781046
    },
    {
      "train_step": 8475,
      "sup_loss": 0.62013420688459
    },
    {
      "train_step": 8476,
      "sup_loss": 0.5297833805574725
    },
    {
      "train_step": 8477,
      "sup_loss": 0.7206365660871746
    },
    {
      "train_step": 8478,
      "sup_loss": 0.4505304285703807
    },
    {
      "train_step": 8479,
      "sup_loss": 0.9037477114017369
    },
    {
      "train_step": 8480,
      "sup_loss": 0.5801615742568234
    },
    {
      "train_step": 8481,
      "sup_loss": 0.5085255311458643
    },
    {
      "train_step": 8482,
      "sup_loss": 0.6144524948832167
    },
    {
      "train_step": 8483,
      "sup_loss": 0.5717475872417931
    },
    {
      "train_step": 8484,
      "sup_loss": 0.7095931056372453
    },
    {
      "train_step": 8485,
      "sup_loss": 0.46410746121500784
    },
    {
      "train_step": 8486,
      "sup_loss": 0.2957744843887118
    },
    {
      "train_step": 8487,
      "sup_loss": 0.6700232478941518
    },
    {
      "train_step": 8488,
      "sup_loss": 0.4518430960678413
    },
    {
      "train_step": 8489,
      "sup_loss": 0.4730650756557079
    },
    {
      "train_step": 8490,
      "sup_loss": 0.5197619525995922
    },
    {
      "train_step": 8491,
      "sup_loss": 0.8066685162177827
    },
    {
      "train_step": 8492,
      "sup_loss": 0.49568737424361253
    },
    {
      "train_step": 8493,
      "sup_loss": 0.6206071499457539
    },
    {
      "train_step": 8494,
      "sup_loss": 0.6791027250923005
    },
    {
      "train_step": 8495,
      "sup_loss": 0.6982206681073846
    },
    {
      "train_step": 8496,
      "sup_loss": 0.6311426208981428
    },
    {
      "train_step": 8497,
      "sup_loss": 0.2614208615372804
    },
    {
      "train_step": 8498,
      "sup_loss": 0.7400925211619349
    },
    {
      "train_step": 8499,
      "sup_loss": 0.22282807260832727
    },
    {
      "train_step": 8500,
      "sup_loss": 0.7188741331626699
    },
    {
      "train_step": 8501,
      "sup_loss": 0.4539562467338103
    },
    {
      "train_step": 8502,
      "sup_loss": 0.43737684822270667
    },
    {
      "train_step": 8503,
      "sup_loss": 0.7002388323645774
    },
    {
      "train_step": 8504,
      "sup_loss": 0.49153024041529203
    },
    {
      "train_step": 8505,
      "sup_loss": 0.5452294214063511
    },
    {
      "train_step": 8506,
      "sup_loss": 0.4842039066901813
    },
    {
      "train_step": 8507,
      "sup_loss": 0.5981090362706316
    },
    {
      "train_step": 8508,
      "sup_loss": 0.778786618308181
    },
    {
      "train_step": 8509,
      "sup_loss": 0.5517239110321372
    },
    {
      "train_step": 8510,
      "sup_loss": 0.5060335973501279
    },
    {
      "train_step": 8511,
      "sup_loss": 0.6873283111807033
    },
    {
      "train_step": 8512,
      "sup_loss": 0.5365936148750854
    },
    {
      "train_step": 8513,
      "sup_loss": 0.24502385921174935
    },
    {
      "train_step": 8514,
      "sup_loss": 0.561682938968663
    },
    {
      "train_step": 8515,
      "sup_loss": 0.44291449494698915
    },
    {
      "train_step": 8516,
      "sup_loss": 0.6238718578868147
    },
    {
      "train_step": 8517,
      "sup_loss": 0.4105098778321668
    },
    {
      "train_step": 8518,
      "sup_loss": 0.6878766517497277
    },
    {
      "train_step": 8519,
      "sup_loss": 0.5890919455135653
    },
    {
      "train_step": 8520,
      "sup_loss": 0.6763872684140692
    },
    {
      "train_step": 8521,
      "sup_loss": 0.5827274436377365
    },
    {
      "train_step": 8522,
      "sup_loss": 0.6854202904414852
    },
    {
      "train_step": 8523,
      "sup_loss": 0.7373752400540081
    },
    {
      "train_step": 8524,
      "sup_loss": 0.6222722546691651
    },
    {
      "train_step": 8525,
      "sup_loss": 0.5627805409534674
    },
    {
      "train_step": 8526,
      "sup_loss": 0.7817359269489277
    },
    {
      "train_step": 8527,
      "sup_loss": 0.5910016737677667
    },
    {
      "train_step": 8528,
      "sup_loss": 0.6394751645379848
    },
    {
      "train_step": 8529,
      "sup_loss": 0.5113928825720325
    },
    {
      "train_step": 8530,
      "sup_loss": 0.5331282557084088
    },
    {
      "train_step": 8531,
      "sup_loss": 0.3551567672071439
    },
    {
      "train_step": 8532,
      "sup_loss": 0.5348370057506502
    },
    {
      "train_step": 8533,
      "sup_loss": 0.5536260730727034
    },
    {
      "train_step": 8534,
      "sup_loss": 0.5754309210074687
    },
    {
      "train_step": 8535,
      "sup_loss": 0.30120907982503653
    },
    {
      "train_step": 8536,
      "sup_loss": 0.6069982254515076
    },
    {
      "train_step": 8537,
      "sup_loss": 0.4142211816803035
    },
    {
      "train_step": 8538,
      "sup_loss": 0.6794057064565628
    },
    {
      "train_step": 8539,
      "sup_loss": 0.6446252436835515
    },
    {
      "train_step": 8540,
      "sup_loss": 0.6144466832642509
    },
    {
      "train_step": 8541,
      "sup_loss": 0.6100110110347682
    },
    {
      "train_step": 8542,
      "sup_loss": 0.43194677371175333
    },
    {
      "train_step": 8543,
      "sup_loss": 0.42441425190133697
    },
    {
      "train_step": 8544,
      "sup_loss": 0.7180282686335988
    },
    {
      "train_step": 8545,
      "sup_loss": 0.555433961524504
    },
    {
      "train_step": 8546,
      "sup_loss": 0.6397514438745541
    },
    {
      "train_step": 8547,
      "sup_loss": 0.46697492326762136
    },
    {
      "train_step": 8548,
      "sup_loss": 0.7394445656180055
    },
    {
      "train_step": 8549,
      "sup_loss": 0.5887497688519338
    },
    {
      "train_step": 8550,
      "sup_loss": 0.7415015310956699
    },
    {
      "train_step": 8551,
      "sup_loss": 0.5582110411319574
    },
    {
      "train_step": 8552,
      "sup_loss": 0.47203965322266017
    },
    {
      "train_step": 8553,
      "sup_loss": 0.5002415215216401
    },
    {
      "train_step": 8554,
      "sup_loss": 0.678716839575624
    },
    {
      "train_step": 8555,
      "sup_loss": 0.65056792913321
    },
    {
      "train_step": 8556,
      "sup_loss": 0.4519455438399882
    },
    {
      "train_step": 8557,
      "sup_loss": 0.39969095167692886
    },
    {
      "train_step": 8558,
      "sup_loss": 0.7348487804261026
    },
    {
      "train_step": 8559,
      "sup_loss": 0.42043000331782204
    },
    {
      "train_step": 8560,
      "sup_loss": 0.6582981134300888
    },
    {
      "train_step": 8561,
      "sup_loss": 0.6290608596432392
    },
    {
      "train_step": 8562,
      "sup_loss": 0.6302411307079169
    },
    {
      "train_step": 8563,
      "sup_loss": 0.24665641092443663
    },
    {
      "train_step": 8564,
      "sup_loss": 0.5368816815468489
    },
    {
      "train_step": 8565,
      "sup_loss": 0.5652369688141748
    },
    {
      "train_step": 8566,
      "sup_loss": 0.3842672396764534
    },
    {
      "train_step": 8567,
      "sup_loss": 0.6895733446561064
    },
    {
      "train_step": 8568,
      "sup_loss": 0.4485240582810916
    },
    {
      "train_step": 8569,
      "sup_loss": 0.392488282909261
    },
    {
      "train_step": 8570,
      "sup_loss": 0.6334690892788802
    },
    {
      "train_step": 8571,
      "sup_loss": 0.6820178129377459
    },
    {
      "train_step": 8572,
      "sup_loss": 0.33971421739849395
    },
    {
      "train_step": 8573,
      "sup_loss": 0.5208477250961185
    },
    {
      "train_step": 8574,
      "sup_loss": 0.6385917706225677
    },
    {
      "train_step": 8575,
      "sup_loss": 0.5960215951865624
    },
    {
      "train_step": 8576,
      "sup_loss": 0.40870839733235215
    },
    {
      "train_step": 8577,
      "sup_loss": 0.6674507340703455
    },
    {
      "train_step": 8578,
      "sup_loss": 0.469489155199064
    },
    {
      "train_step": 8579,
      "sup_loss": 0.5388293074745713
    },
    {
      "train_step": 8580,
      "sup_loss": 0.5110483842161379
    },
    {
      "train_step": 8581,
      "sup_loss": 0.7478353931476929
    },
    {
      "train_step": 8582,
      "sup_loss": 0.5958080330441197
    },
    {
      "train_step": 8583,
      "sup_loss": 0.7028783330656742
    },
    {
      "train_step": 8584,
      "sup_loss": 0.5128556456113013
    },
    {
      "train_step": 8585,
      "sup_loss": 0.6463254564587901
    },
    {
      "train_step": 8586,
      "sup_loss": 0.5653617081878909
    },
    {
      "train_step": 8587,
      "sup_loss": 0.5225484503658845
    },
    {
      "train_step": 8588,
      "sup_loss": 0.4330411859581826
    },
    {
      "train_step": 8589,
      "sup_loss": 0.6747488252221538
    },
    {
      "train_step": 8590,
      "sup_loss": 0.7615127828993761
    },
    {
      "train_step": 8591,
      "sup_loss": 0.5746764593662375
    },
    {
      "train_step": 8592,
      "sup_loss": 0.6448703517265197
    },
    {
      "train_step": 8593,
      "sup_loss": 0.6885494354507121
    },
    {
      "train_step": 8594,
      "sup_loss": 0.6203015774340002
    },
    {
      "train_step": 8595,
      "sup_loss": 0.5213787038650195
    },
    {
      "train_step": 8596,
      "sup_loss": 0.76698685637279
    },
    {
      "train_step": 8597,
      "sup_loss": 0.5044070340291632
    },
    {
      "train_step": 8598,
      "sup_loss": 0.47202757527086076
    },
    {
      "train_step": 8599,
      "sup_loss": 0.5578813724686013
    },
    {
      "train_step": 8600,
      "sup_loss": 0.5346751263996302
    },
    {
      "train_step": 8601,
      "sup_loss": 0.3912707805318576
    },
    {
      "train_step": 8602,
      "sup_loss": 0.504219546121143
    },
    {
      "train_step": 8603,
      "sup_loss": 0.32801882797934934
    },
    {
      "train_step": 8604,
      "sup_loss": 0.31791614373709426
    },
    {
      "train_step": 8605,
      "sup_loss": 0.6399004935193554
    },
    {
      "train_step": 8606,
      "sup_loss": 0.5470157808977691
    },
    {
      "train_step": 8607,
      "sup_loss": 0.48635636853538944
    },
    {
      "train_step": 8608,
      "sup_loss": 0.4392744958135158
    },
    {
      "train_step": 8609,
      "sup_loss": 0.42851377878110325
    },
    {
      "train_step": 8610,
      "sup_loss": 0.769656850841692
    },
    {
      "train_step": 8611,
      "sup_loss": 0.4733927170063735
    },
    {
      "train_step": 8612,
      "sup_loss": 0.6940446604448468
    },
    {
      "train_step": 8613,
      "sup_loss": 0.6084729915262306
    },
    {
      "train_step": 8614,
      "sup_loss": 0.649162588442739
    },
    {
      "train_step": 8615,
      "sup_loss": 0.5477945349847058
    },
    {
      "train_step": 8616,
      "sup_loss": 0.6810673951345982
    },
    {
      "train_step": 8617,
      "sup_loss": 0.4540192339102009
    },
    {
      "train_step": 8618,
      "sup_loss": 0.4948534807232802
    },
    {
      "train_step": 8619,
      "sup_loss": 0.41335749773847286
    },
    {
      "train_step": 8620,
      "sup_loss": 0.6157976649123851
    },
    {
      "train_step": 8621,
      "sup_loss": 0.5193227742647253
    },
    {
      "train_step": 8622,
      "sup_loss": 0.531038486165894
    },
    {
      "train_step": 8623,
      "sup_loss": 0.6513542950804718
    },
    {
      "train_step": 8624,
      "sup_loss": 0.3457068992871457
    },
    {
      "train_step": 8625,
      "sup_loss": 0.5314603726674839
    },
    {
      "train_step": 8626,
      "sup_loss": 0.5288001269747933
    },
    {
      "train_step": 8627,
      "sup_loss": 0.5090895817255091
    },
    {
      "train_step": 8628,
      "sup_loss": 0.5240875024782179
    },
    {
      "train_step": 8629,
      "sup_loss": 0.5489700102144716
    },
    {
      "train_step": 8630,
      "sup_loss": 0.44247167881753824
    },
    {
      "train_step": 8631,
      "sup_loss": 0.40425193247881597
    },
    {
      "train_step": 8632,
      "sup_loss": 0.6574491960555705
    },
    {
      "train_step": 8633,
      "sup_loss": 0.557064332099113
    },
    {
      "train_step": 8634,
      "sup_loss": 0.5050292158921639
    },
    {
      "train_step": 8635,
      "sup_loss": 0.6966356951930553
    },
    {
      "train_step": 8636,
      "sup_loss": 0.7130224928960227
    },
    {
      "train_step": 8637,
      "sup_loss": 0.7170284327646207
    },
    {
      "train_step": 8638,
      "sup_loss": 0.6711277924136535
    },
    {
      "train_step": 8639,
      "sup_loss": 0.7368222886111817
    },
    {
      "train_step": 8640,
      "sup_loss": 0.6126741931974848
    },
    {
      "train_step": 8641,
      "sup_loss": 0.521243576489028
    },
    {
      "train_step": 8642,
      "sup_loss": 0.4478914805812437
    },
    {
      "train_step": 8643,
      "sup_loss": 0.5684192398443684
    },
    {
      "train_step": 8644,
      "sup_loss": 0.625368033946255
    },
    {
      "train_step": 8645,
      "sup_loss": 0.2589201841305871
    },
    {
      "train_step": 8646,
      "sup_loss": 0.610429202782551
    },
    {
      "train_step": 8647,
      "sup_loss": 0.2674405995561899
    },
    {
      "train_step": 8648,
      "sup_loss": 0.60727118250692
    },
    {
      "train_step": 8649,
      "sup_loss": 0.4469595046906118
    },
    {
      "train_step": 8650,
      "sup_loss": 0.7003957456129112
    },
    {
      "train_step": 8651,
      "sup_loss": 0.37212329305413683
    },
    {
      "train_step": 8652,
      "sup_loss": 0.4544429177164229
    },
    {
      "train_step": 8653,
      "sup_loss": 0.6510771407199436
    },
    {
      "train_step": 8654,
      "sup_loss": 0.5558314397688493
    },
    {
      "train_step": 8655,
      "sup_loss": 0.5489058634001207
    },
    {
      "train_step": 8656,
      "sup_loss": 0.49737656991738294
    },
    {
      "train_step": 8657,
      "sup_loss": 0.3898709410901719
    },
    {
      "train_step": 8658,
      "sup_loss": 0.6221724053151194
    },
    {
      "train_step": 8659,
      "sup_loss": 0.5458313069259839
    },
    {
      "train_step": 8660,
      "sup_loss": 0.2621436906088736
    },
    {
      "train_step": 8661,
      "sup_loss": 0.5941704639259097
    },
    {
      "train_step": 8662,
      "sup_loss": 0.6285612373677502
    },
    {
      "train_step": 8663,
      "sup_loss": 0.5045296333737845
    },
    {
      "train_step": 8664,
      "sup_loss": 0.33332803030470504
    },
    {
      "train_step": 8665,
      "sup_loss": 0.6191026959743999
    },
    {
      "train_step": 8666,
      "sup_loss": 0.6002031479312923
    },
    {
      "train_step": 8667,
      "sup_loss": 0.2735856686980789
    },
    {
      "train_step": 8668,
      "sup_loss": 0.5926857990413787
    },
    {
      "train_step": 8669,
      "sup_loss": 0.537755164806124
    },
    {
      "train_step": 8670,
      "sup_loss": 0.4492071157236588
    },
    {
      "train_step": 8671,
      "sup_loss": 0.5228083331080039
    },
    {
      "train_step": 8672,
      "sup_loss": 0.5374702886745383
    },
    {
      "train_step": 8673,
      "sup_loss": 0.6519230715882434
    },
    {
      "train_step": 8674,
      "sup_loss": 0.7656512239419231
    },
    {
      "train_step": 8675,
      "sup_loss": 0.41706971926297426
    },
    {
      "train_step": 8676,
      "sup_loss": 0.5418598518351595
    },
    {
      "train_step": 8677,
      "sup_loss": 0.8288349967690973
    },
    {
      "train_step": 8678,
      "sup_loss": 0.5918827807538881
    },
    {
      "train_step": 8679,
      "sup_loss": 0.535849172455216
    },
    {
      "train_step": 8680,
      "sup_loss": 0.6296803454430082
    },
    {
      "train_step": 8681,
      "sup_loss": 0.6998511329479641
    },
    {
      "train_step": 8682,
      "sup_loss": 0.6326365573312808
    },
    {
      "train_step": 8683,
      "sup_loss": 0.595376603454461
    },
    {
      "train_step": 8684,
      "sup_loss": 0.6701793990774741
    },
    {
      "train_step": 8685,
      "sup_loss": 0.6473012354736207
    },
    {
      "train_step": 8686,
      "sup_loss": 0.5570923064170761
    },
    {
      "train_step": 8687,
      "sup_loss": 0.7795776824719373
    },
    {
      "train_step": 8688,
      "sup_loss": 0.646732876341452
    },
    {
      "train_step": 8689,
      "sup_loss": 0.65582336184175
    },
    {
      "train_step": 8690,
      "sup_loss": 0.5814038674533427
    },
    {
      "train_step": 8691,
      "sup_loss": 0.5880615671040451
    },
    {
      "train_step": 8692,
      "sup_loss": 0.43433899664369974
    },
    {
      "train_step": 8693,
      "sup_loss": 0.5636770755350338
    },
    {
      "train_step": 8694,
      "sup_loss": 0.722876458698326
    },
    {
      "train_step": 8695,
      "sup_loss": 0.613479328304211
    },
    {
      "train_step": 8696,
      "sup_loss": 0.39620108430095313
    },
    {
      "train_step": 8697,
      "sup_loss": 0.6685810861333153
    },
    {
      "train_step": 8698,
      "sup_loss": 0.4649101248084277
    },
    {
      "train_step": 8699,
      "sup_loss": 0.6566653839026373
    },
    {
      "train_step": 8700,
      "sup_loss": 0.6473093701487433
    },
    {
      "train_step": 8701,
      "sup_loss": 0.5409696299799618
    },
    {
      "train_step": 8702,
      "sup_loss": 0.3873111584197499
    },
    {
      "train_step": 8703,
      "sup_loss": 0.4936830773390439
    },
    {
      "train_step": 8704,
      "sup_loss": 0.44211931203338267
    },
    {
      "train_step": 8705,
      "sup_loss": 0.5479575207018056
    },
    {
      "train_step": 8706,
      "sup_loss": 0.28524641954799745
    },
    {
      "train_step": 8707,
      "sup_loss": 0.6697003902744031
    },
    {
      "train_step": 8708,
      "sup_loss": 0.6042295791852464
    },
    {
      "train_step": 8709,
      "sup_loss": 0.6453198357032847
    },
    {
      "train_step": 8710,
      "sup_loss": 0.6706735171348874
    },
    {
      "train_step": 8711,
      "sup_loss": 0.6823155184684494
    },
    {
      "train_step": 8712,
      "sup_loss": 0.4687485439614237
    },
    {
      "train_step": 8713,
      "sup_loss": 0.5990637876804937
    },
    {
      "train_step": 8714,
      "sup_loss": 0.6486413370468463
    },
    {
      "train_step": 8715,
      "sup_loss": 0.7620593018846498
    },
    {
      "train_step": 8716,
      "sup_loss": 0.5278376569391525
    },
    {
      "train_step": 8717,
      "sup_loss": 0.613963455611776
    },
    {
      "train_step": 8718,
      "sup_loss": 0.6227165415958008
    },
    {
      "train_step": 8719,
      "sup_loss": 0.49033609845460746
    },
    {
      "train_step": 8720,
      "sup_loss": 0.46177399701009003
    },
    {
      "train_step": 8721,
      "sup_loss": 0.4697187976972676
    },
    {
      "train_step": 8722,
      "sup_loss": 0.577235982579262
    },
    {
      "train_step": 8723,
      "sup_loss": 0.36285983050510173
    },
    {
      "train_step": 8724,
      "sup_loss": 0.6555040981860877
    },
    {
      "train_step": 8725,
      "sup_loss": 0.6548263802684742
    },
    {
      "train_step": 8726,
      "sup_loss": 0.5108702301399004
    },
    {
      "train_step": 8727,
      "sup_loss": 0.5906469745834676
    },
    {
      "train_step": 8728,
      "sup_loss": 0.5920805128479643
    },
    {
      "train_step": 8729,
      "sup_loss": 0.4113951957888794
    },
    {
      "train_step": 8730,
      "sup_loss": 0.8057530706504386
    },
    {
      "train_step": 8731,
      "sup_loss": 0.654222284213429
    },
    {
      "train_step": 8732,
      "sup_loss": 0.6049579957467361
    },
    {
      "train_step": 8733,
      "sup_loss": 0.5027998537627744
    },
    {
      "train_step": 8734,
      "sup_loss": 0.47149073252945795
    },
    {
      "train_step": 8735,
      "sup_loss": 0.4319660000688464
    },
    {
      "train_step": 8736,
      "sup_loss": 0.5139689373233681
    },
    {
      "train_step": 8737,
      "sup_loss": 0.6838398102020871
    },
    {
      "train_step": 8738,
      "sup_loss": 0.3136282135496382
    },
    {
      "train_step": 8739,
      "sup_loss": 0.607263477750948
    },
    {
      "train_step": 8740,
      "sup_loss": 0.6496789471628057
    },
    {
      "train_step": 8741,
      "sup_loss": 0.500172735082697
    },
    {
      "train_step": 8742,
      "sup_loss": 0.5135767229826134
    },
    {
      "train_step": 8743,
      "sup_loss": 0.6249441307675361
    },
    {
      "train_step": 8744,
      "sup_loss": 0.5542338682858469
    },
    {
      "train_step": 8745,
      "sup_loss": 0.5134611666366847
    },
    {
      "train_step": 8746,
      "sup_loss": 0.6074004245690489
    },
    {
      "train_step": 8747,
      "sup_loss": 0.4246653093831328
    },
    {
      "train_step": 8748,
      "sup_loss": 0.2943244629623085
    },
    {
      "train_step": 8749,
      "sup_loss": 0.8225833199422181
    },
    {
      "train_step": 8750,
      "sup_loss": 0.43388744476713653
    },
    {
      "train_step": 8751,
      "sup_loss": 0.4038811014989514
    },
    {
      "train_step": 8752,
      "sup_loss": 0.34289757186118186
    },
    {
      "train_step": 8753,
      "sup_loss": 0.41603097369494424
    },
    {
      "train_step": 8754,
      "sup_loss": 0.4935088805608188
    },
    {
      "train_step": 8755,
      "sup_loss": 0.8583553331138587
    },
    {
      "train_step": 8756,
      "sup_loss": 0.46675822924319804
    },
    {
      "train_step": 8757,
      "sup_loss": 0.8374434168236305
    },
    {
      "train_step": 8758,
      "sup_loss": 0.48144187883571715
    },
    {
      "train_step": 8759,
      "sup_loss": 0.5263561852475117
    },
    {
      "train_step": 8760,
      "sup_loss": 0.4110947640502397
    },
    {
      "train_step": 8761,
      "sup_loss": 0.5916610138458677
    },
    {
      "train_step": 8762,
      "sup_loss": 0.7268553747549352
    },
    {
      "train_step": 8763,
      "sup_loss": 0.6211262659584291
    },
    {
      "train_step": 8764,
      "sup_loss": 0.5711567788542192
    },
    {
      "train_step": 8765,
      "sup_loss": 0.46144272133080383
    },
    {
      "train_step": 8766,
      "sup_loss": 0.5111592067802402
    },
    {
      "train_step": 8767,
      "sup_loss": 0.6895241428256995
    },
    {
      "train_step": 8768,
      "sup_loss": 0.5842242805912655
    },
    {
      "train_step": 8769,
      "sup_loss": 0.5874895145597372
    },
    {
      "train_step": 8770,
      "sup_loss": 0.8766238813785602
    },
    {
      "train_step": 8771,
      "sup_loss": 0.5584250231502647
    },
    {
      "train_step": 8772,
      "sup_loss": 0.5198431542339514
    },
    {
      "train_step": 8773,
      "sup_loss": 0.6373222384609162
    },
    {
      "train_step": 8774,
      "sup_loss": 0.5110645020682183
    },
    {
      "train_step": 8775,
      "sup_loss": 0.5363786946579527
    },
    {
      "train_step": 8776,
      "sup_loss": 0.38818947580002366
    },
    {
      "train_step": 8777,
      "sup_loss": 0.4489056430492463
    },
    {
      "train_step": 8778,
      "sup_loss": 0.7414516555005538
    },
    {
      "train_step": 8779,
      "sup_loss": 0.6659357943966452
    },
    {
      "train_step": 8780,
      "sup_loss": 0.23206200361250653
    },
    {
      "train_step": 8781,
      "sup_loss": 0.42106103298558833
    },
    {
      "train_step": 8782,
      "sup_loss": 0.7912014042677727
    },
    {
      "train_step": 8783,
      "sup_loss": 0.5643701604479527
    },
    {
      "train_step": 8784,
      "sup_loss": 0.6497567619883865
    },
    {
      "train_step": 8785,
      "sup_loss": 0.742591152165114
    },
    {
      "train_step": 8786,
      "sup_loss": 0.6327261800737511
    },
    {
      "train_step": 8787,
      "sup_loss": 0.5765171944417473
    },
    {
      "train_step": 8788,
      "sup_loss": 0.5663654463241288
    },
    {
      "train_step": 8789,
      "sup_loss": 0.48138726427223405
    },
    {
      "train_step": 8790,
      "sup_loss": 0.43818821318029705
    },
    {
      "train_step": 8791,
      "sup_loss": 0.573149716071105
    },
    {
      "train_step": 8792,
      "sup_loss": 0.6896840265272303
    },
    {
      "train_step": 8793,
      "sup_loss": 0.7177892957000978
    },
    {
      "train_step": 8794,
      "sup_loss": 0.5878935572062567
    },
    {
      "train_step": 8795,
      "sup_loss": 0.5918956851192716
    },
    {
      "train_step": 8796,
      "sup_loss": 0.5266127320269269
    },
    {
      "train_step": 8797,
      "sup_loss": 0.5188172348646523
    },
    {
      "train_step": 8798,
      "sup_loss": 0.6767191783307526
    },
    {
      "train_step": 8799,
      "sup_loss": 0.5367676356022248
    },
    {
      "train_step": 8800,
      "sup_loss": 0.7304570600086417
    },
    {
      "train_step": 8801,
      "sup_loss": 0.5732531859213446
    },
    {
      "train_step": 8802,
      "sup_loss": 0.38008455615793957
    },
    {
      "train_step": 8803,
      "sup_loss": 0.6622595869812048
    },
    {
      "train_step": 8804,
      "sup_loss": 0.4018232803945495
    },
    {
      "train_step": 8805,
      "sup_loss": 0.8231975817284433
    },
    {
      "train_step": 8806,
      "sup_loss": 0.5603589293823539
    },
    {
      "train_step": 8807,
      "sup_loss": 0.48390355640967303
    },
    {
      "train_step": 8808,
      "sup_loss": 0.8949498627156041
    },
    {
      "train_step": 8809,
      "sup_loss": 0.46341774427910065
    },
    {
      "train_step": 8810,
      "sup_loss": 0.7422702465740072
    },
    {
      "train_step": 8811,
      "sup_loss": 0.510750752932377
    },
    {
      "train_step": 8812,
      "sup_loss": 0.6572145171964685
    },
    {
      "train_step": 8813,
      "sup_loss": 0.5168789033588962
    },
    {
      "train_step": 8814,
      "sup_loss": 0.6589529466182625
    },
    {
      "train_step": 8815,
      "sup_loss": 0.7354684197153504
    },
    {
      "train_step": 8816,
      "sup_loss": 0.43387017112769855
    },
    {
      "train_step": 8817,
      "sup_loss": 0.4479751321594171
    },
    {
      "train_step": 8818,
      "sup_loss": 0.6693284498854346
    },
    {
      "train_step": 8819,
      "sup_loss": 0.7010310819736613
    },
    {
      "train_step": 8820,
      "sup_loss": 0.6333716847303243
    },
    {
      "train_step": 8821,
      "sup_loss": 0.5467430003174314
    },
    {
      "train_step": 8822,
      "sup_loss": 0.649051002739232
    },
    {
      "train_step": 8823,
      "sup_loss": 0.5294704433279857
    },
    {
      "train_step": 8824,
      "sup_loss": 0.5468331316760935
    },
    {
      "train_step": 8825,
      "sup_loss": 0.7092407420199734
    },
    {
      "train_step": 8826,
      "sup_loss": 0.7305595383927538
    },
    {
      "train_step": 8827,
      "sup_loss": 0.5920626900563422
    },
    {
      "train_step": 8828,
      "sup_loss": 0.47031482908974076
    },
    {
      "train_step": 8829,
      "sup_loss": 0.7387631420108766
    },
    {
      "train_step": 8830,
      "sup_loss": 0.2027901533388753
    },
    {
      "train_step": 8831,
      "sup_loss": 0.6244106306705421
    },
    {
      "train_step": 8832,
      "sup_loss": 0.4573667545658644
    },
    {
      "train_step": 8833,
      "sup_loss": 0.5868696106251186
    },
    {
      "train_step": 8834,
      "sup_loss": 0.6412686863356822
    },
    {
      "train_step": 8835,
      "sup_loss": 0.672476796741949
    },
    {
      "train_step": 8836,
      "sup_loss": 0.508694993635788
    },
    {
      "train_step": 8837,
      "sup_loss": 0.4109743309546223
    },
    {
      "train_step": 8838,
      "sup_loss": 0.609886471675555
    },
    {
      "train_step": 8839,
      "sup_loss": 0.5330470094911226
    },
    {
      "train_step": 8840,
      "sup_loss": 0.6262767086874348
    },
    {
      "train_step": 8841,
      "sup_loss": 0.46905179719689466
    },
    {
      "train_step": 8842,
      "sup_loss": 0.6395538337155423
    },
    {
      "train_step": 8843,
      "sup_loss": 0.4202146046352754
    },
    {
      "train_step": 8844,
      "sup_loss": 0.7046161464697076
    },
    {
      "train_step": 8845,
      "sup_loss": 0.6999513056743063
    },
    {
      "train_step": 8846,
      "sup_loss": 0.6151774968125125
    },
    {
      "train_step": 8847,
      "sup_loss": 0.4036558332339791
    },
    {
      "train_step": 8848,
      "sup_loss": 0.5073187876348568
    },
    {
      "train_step": 8849,
      "sup_loss": 0.7761388689599533
    },
    {
      "train_step": 8850,
      "sup_loss": 0.569468145616692
    },
    {
      "train_step": 8851,
      "sup_loss": 0.6524315389104717
    },
    {
      "train_step": 8852,
      "sup_loss": 0.5479809852993902
    },
    {
      "train_step": 8853,
      "sup_loss": 0.5174440679945996
    },
    {
      "train_step": 8854,
      "sup_loss": 0.8476694590746164
    },
    {
      "train_step": 8855,
      "sup_loss": 0.5002597067365055
    },
    {
      "train_step": 8856,
      "sup_loss": 0.6536350952220465
    },
    {
      "train_step": 8857,
      "sup_loss": 0.6329429902909119
    },
    {
      "train_step": 8858,
      "sup_loss": 0.7426054304274002
    },
    {
      "train_step": 8859,
      "sup_loss": 0.4519845581045019
    },
    {
      "train_step": 8860,
      "sup_loss": 0.753661407587859
    },
    {
      "train_step": 8861,
      "sup_loss": 0.6394177608405596
    },
    {
      "train_step": 8862,
      "sup_loss": 0.569349450014045
    },
    {
      "train_step": 8863,
      "sup_loss": 0.6202035533270241
    },
    {
      "train_step": 8864,
      "sup_loss": 0.4967075889437503
    },
    {
      "train_step": 8865,
      "sup_loss": 0.5210546914362656
    },
    {
      "train_step": 8866,
      "sup_loss": 0.5587350428804401
    },
    {
      "train_step": 8867,
      "sup_loss": 0.7680364687134406
    },
    {
      "train_step": 8868,
      "sup_loss": 0.6164085681658846
    },
    {
      "train_step": 8869,
      "sup_loss": 0.4658516372039798
    },
    {
      "train_step": 8870,
      "sup_loss": 0.49948844728463787
    },
    {
      "train_step": 8871,
      "sup_loss": 0.5399859053538113
    },
    {
      "train_step": 8872,
      "sup_loss": 0.3885391407702582
    },
    {
      "train_step": 8873,
      "sup_loss": 0.38840037532352184
    },
    {
      "train_step": 8874,
      "sup_loss": 0.6186242173913016
    },
    {
      "train_step": 8875,
      "sup_loss": 0.6539206470456365
    },
    {
      "train_step": 8876,
      "sup_loss": 0.846334030793307
    },
    {
      "train_step": 8877,
      "sup_loss": 0.545755893491177
    },
    {
      "train_step": 8878,
      "sup_loss": 0.457598679915351
    },
    {
      "train_step": 8879,
      "sup_loss": 0.5478096150478386
    },
    {
      "train_step": 8880,
      "sup_loss": 0.6428784417796535
    },
    {
      "train_step": 8881,
      "sup_loss": 0.5070769085152256
    },
    {
      "train_step": 8882,
      "sup_loss": 0.6352247828648414
    },
    {
      "train_step": 8883,
      "sup_loss": 0.46391654084931866
    },
    {
      "train_step": 8884,
      "sup_loss": 0.7622318960775103
    },
    {
      "train_step": 8885,
      "sup_loss": 0.6797979236232236
    },
    {
      "train_step": 8886,
      "sup_loss": 0.6628149720841615
    },
    {
      "train_step": 8887,
      "sup_loss": 0.3929633667061256
    },
    {
      "train_step": 8888,
      "sup_loss": 0.5787009127742354
    },
    {
      "train_step": 8889,
      "sup_loss": 0.3985922860698956
    },
    {
      "train_step": 8890,
      "sup_loss": 0.4051206086575284
    },
    {
      "train_step": 8891,
      "sup_loss": 0.6349576207958826
    },
    {
      "train_step": 8892,
      "sup_loss": 0.7767804846035806
    },
    {
      "train_step": 8893,
      "sup_loss": 0.567729050756918
    },
    {
      "train_step": 8894,
      "sup_loss": 0.46307096420228294
    },
    {
      "train_step": 8895,
      "sup_loss": 0.5660026027815727
    },
    {
      "train_step": 8896,
      "sup_loss": 0.8080981296679457
    },
    {
      "train_step": 8897,
      "sup_loss": 0.5902803987591977
    },
    {
      "train_step": 8898,
      "sup_loss": 0.4468912187101554
    },
    {
      "train_step": 8899,
      "sup_loss": 0.5650349306557171
    },
    {
      "train_step": 8900,
      "sup_loss": 0.43694229050409533
    },
    {
      "train_step": 8901,
      "sup_loss": 0.6294647735603495
    },
    {
      "train_step": 8902,
      "sup_loss": 0.6084741685878443
    },
    {
      "train_step": 8903,
      "sup_loss": 0.33042537746477785
    },
    {
      "train_step": 8904,
      "sup_loss": 0.45781245197680337
    },
    {
      "train_step": 8905,
      "sup_loss": 0.5937130992649277
    },
    {
      "train_step": 8906,
      "sup_loss": 0.3338490941259615
    },
    {
      "train_step": 8907,
      "sup_loss": 0.3625351210384715
    },
    {
      "train_step": 8908,
      "sup_loss": 0.3728284796841924
    },
    {
      "train_step": 8909,
      "sup_loss": 0.45563179937038606
    },
    {
      "train_step": 8910,
      "sup_loss": 0.25522922273151877
    },
    {
      "train_step": 8911,
      "sup_loss": 0.4232983557984041
    },
    {
      "train_step": 8912,
      "sup_loss": 0.5243645837889377
    },
    {
      "train_step": 8913,
      "sup_loss": 0.45883835639564613
    },
    {
      "train_step": 8914,
      "sup_loss": 0.5268661378706578
    },
    {
      "train_step": 8915,
      "sup_loss": 0.42511760411525057
    },
    {
      "train_step": 8916,
      "sup_loss": 0.6976991898662063
    },
    {
      "train_step": 8917,
      "sup_loss": 0.31583941317851866
    },
    {
      "train_step": 8918,
      "sup_loss": 0.7515914103580486
    },
    {
      "train_step": 8919,
      "sup_loss": 0.42609154455194775
    },
    {
      "train_step": 8920,
      "sup_loss": 0.5820885918560545
    },
    {
      "train_step": 8921,
      "sup_loss": 0.7188861106519391
    },
    {
      "train_step": 8922,
      "sup_loss": 0.6937689881423615
    },
    {
      "train_step": 8923,
      "sup_loss": 0.5885958960170282
    },
    {
      "train_step": 8924,
      "sup_loss": 0.516730617686093
    },
    {
      "train_step": 8925,
      "sup_loss": 0.40946512233217075
    },
    {
      "train_step": 8926,
      "sup_loss": 0.7912443266054677
    },
    {
      "train_step": 8927,
      "sup_loss": 0.3812291331442197
    },
    {
      "train_step": 8928,
      "sup_loss": 0.6619559249195416
    },
    {
      "train_step": 8929,
      "sup_loss": 0.48298625929837624
    },
    {
      "train_step": 8930,
      "sup_loss": 0.529645187037105
    },
    {
      "train_step": 8931,
      "sup_loss": 0.6479782796530706
    },
    {
      "train_step": 8932,
      "sup_loss": 0.6035449904366722
    },
    {
      "train_step": 8933,
      "sup_loss": 0.7610382147871031
    },
    {
      "train_step": 8934,
      "sup_loss": 0.5436699267574746
    },
    {
      "train_step": 8935,
      "sup_loss": 0.6151164345566845
    },
    {
      "train_step": 8936,
      "sup_loss": 0.52586655484797
    },
    {
      "train_step": 8937,
      "sup_loss": 0.5108896259836837
    },
    {
      "train_step": 8938,
      "sup_loss": 0.5302674830551795
    },
    {
      "train_step": 8939,
      "sup_loss": 0.6037494638281913
    },
    {
      "train_step": 8940,
      "sup_loss": 0.6695071184596557
    },
    {
      "train_step": 8941,
      "sup_loss": 0.6911344516701038
    },
    {
      "train_step": 8942,
      "sup_loss": 0.7640968324539463
    },
    {
      "train_step": 8943,
      "sup_loss": 0.5935469490715304
    },
    {
      "train_step": 8944,
      "sup_loss": 0.6191757405696804
    },
    {
      "train_step": 8945,
      "sup_loss": 0.554086094246138
    },
    {
      "train_step": 8946,
      "sup_loss": 0.3235017294820199
    },
    {
      "train_step": 8947,
      "sup_loss": 0.9045494503011364
    },
    {
      "train_step": 8948,
      "sup_loss": 0.6746024975513465
    },
    {
      "train_step": 8949,
      "sup_loss": 0.6853530218144879
    },
    {
      "train_step": 8950,
      "sup_loss": 0.5915904914806215
    },
    {
      "train_step": 8951,
      "sup_loss": 0.6054092775330238
    },
    {
      "train_step": 8952,
      "sup_loss": 0.5336946334286751
    },
    {
      "train_step": 8953,
      "sup_loss": 0.49894066105929624
    },
    {
      "train_step": 8954,
      "sup_loss": 0.7654213831178855
    },
    {
      "train_step": 8955,
      "sup_loss": 0.5591695034198969
    },
    {
      "train_step": 8956,
      "sup_loss": 0.43566121835742216
    },
    {
      "train_step": 8957,
      "sup_loss": 0.5024847918003295
    },
    {
      "train_step": 8958,
      "sup_loss": 0.43415857592890567
    },
    {
      "train_step": 8959,
      "sup_loss": 0.5252869941544688
    },
    {
      "train_step": 8960,
      "sup_loss": 0.545827273520672
    },
    {
      "train_step": 8961,
      "sup_loss": 0.3661506635014068
    },
    {
      "train_step": 8962,
      "sup_loss": 0.6320365240450746
    },
    {
      "train_step": 8963,
      "sup_loss": 0.47195568503014507
    },
    {
      "train_step": 8964,
      "sup_loss": 0.44999482290476717
    },
    {
      "train_step": 8965,
      "sup_loss": 0.6854893993573099
    },
    {
      "train_step": 8966,
      "sup_loss": 0.8073224550553609
    },
    {
      "train_step": 8967,
      "sup_loss": 0.5100489161230516
    },
    {
      "train_step": 8968,
      "sup_loss": 0.4098666908178746
    },
    {
      "train_step": 8969,
      "sup_loss": 0.49081754795261884
    },
    {
      "train_step": 8970,
      "sup_loss": 0.582329201994874
    },
    {
      "train_step": 8971,
      "sup_loss": 0.5998278086190024
    },
    {
      "train_step": 8972,
      "sup_loss": 0.6193095002945082
    },
    {
      "train_step": 8973,
      "sup_loss": 0.5866711933642054
    },
    {
      "train_step": 8974,
      "sup_loss": 0.5899411199590053
    },
    {
      "train_step": 8975,
      "sup_loss": 0.3783508288365501
    },
    {
      "train_step": 8976,
      "sup_loss": 0.5500786334956015
    },
    {
      "train_step": 8977,
      "sup_loss": 0.7483722330939109
    },
    {
      "train_step": 8978,
      "sup_loss": 0.3624895267255733
    },
    {
      "train_step": 8979,
      "sup_loss": 0.5426691854467278
    },
    {
      "train_step": 8980,
      "sup_loss": 0.6767775345118481
    },
    {
      "train_step": 8981,
      "sup_loss": 0.7425058939877376
    },
    {
      "train_step": 8982,
      "sup_loss": 0.45766527647033267
    },
    {
      "train_step": 8983,
      "sup_loss": 0.5226852993465623
    },
    {
      "train_step": 8984,
      "sup_loss": 0.47811910526469814
    },
    {
      "train_step": 8985,
      "sup_loss": 0.6517655644479075
    },
    {
      "train_step": 8986,
      "sup_loss": 0.6274865034220232
    },
    {
      "train_step": 8987,
      "sup_loss": 0.4411150920945157
    },
    {
      "train_step": 8988,
      "sup_loss": 0.5421060128589917
    },
    {
      "train_step": 8989,
      "sup_loss": 0.26881600371200576
    },
    {
      "train_step": 8990,
      "sup_loss": 0.6060807060345315
    },
    {
      "train_step": 8991,
      "sup_loss": 0.7049424796867549
    },
    {
      "train_step": 8992,
      "sup_loss": 0.5794362920194639
    },
    {
      "train_step": 8993,
      "sup_loss": 0.6357391329557305
    },
    {
      "train_step": 8994,
      "sup_loss": 0.6196958466095589
    },
    {
      "train_step": 8995,
      "sup_loss": 0.7068409431453311
    },
    {
      "train_step": 8996,
      "sup_loss": 0.7606899646710986
    },
    {
      "train_step": 8997,
      "sup_loss": 0.6640539289803171
    },
    {
      "train_step": 8998,
      "sup_loss": 0.5940767157347528
    },
    {
      "train_step": 8999,
      "sup_loss": 0.7567706412169123
    },
    {
      "train_step": 9000,
      "sup_loss": 0.4733799706956071
    },
    {
      "train_step": 9001,
      "sup_loss": 0.5584660171452307
    },
    {
      "train_step": 9002,
      "sup_loss": 0.6608876315303391
    },
    {
      "train_step": 9003,
      "sup_loss": 0.6112806513321645
    },
    {
      "train_step": 9004,
      "sup_loss": 0.5821385365292913
    },
    {
      "train_step": 9005,
      "sup_loss": 0.6460768535304422
    },
    {
      "train_step": 9006,
      "sup_loss": 0.47642659382041974
    },
    {
      "train_step": 9007,
      "sup_loss": 0.5565456271966658
    },
    {
      "train_step": 9008,
      "sup_loss": 0.5547438029245462
    },
    {
      "train_step": 9009,
      "sup_loss": 0.5507164838187765
    },
    {
      "train_step": 9010,
      "sup_loss": 0.4946597005289586
    },
    {
      "train_step": 9011,
      "sup_loss": 0.39218520839524434
    },
    {
      "train_step": 9012,
      "sup_loss": 0.3240517610538413
    },
    {
      "train_step": 9013,
      "sup_loss": 0.570959741874072
    },
    {
      "train_step": 9014,
      "sup_loss": 0.6364693650035047
    },
    {
      "train_step": 9015,
      "sup_loss": 0.5182659166351002
    },
    {
      "train_step": 9016,
      "sup_loss": 0.7601496267304308
    },
    {
      "train_step": 9017,
      "sup_loss": 0.5292696633667817
    },
    {
      "train_step": 9018,
      "sup_loss": 0.5429064723399549
    },
    {
      "train_step": 9019,
      "sup_loss": 0.28803710550987455
    },
    {
      "train_step": 9020,
      "sup_loss": 0.5786584391420208
    },
    {
      "train_step": 9021,
      "sup_loss": 0.6042347107079828
    },
    {
      "train_step": 9022,
      "sup_loss": 0.4682907404611541
    },
    {
      "train_step": 9023,
      "sup_loss": 0.5870469136455729
    },
    {
      "train_step": 9024,
      "sup_loss": 0.38613376253282533
    },
    {
      "train_step": 9025,
      "sup_loss": 0.5373655021325003
    },
    {
      "train_step": 9026,
      "sup_loss": 0.6763061600594249
    },
    {
      "train_step": 9027,
      "sup_loss": 0.6129646759818976
    },
    {
      "train_step": 9028,
      "sup_loss": 0.689662660771186
    },
    {
      "train_step": 9029,
      "sup_loss": 0.42673552570918327
    },
    {
      "train_step": 9030,
      "sup_loss": 0.6673608229574223
    },
    {
      "train_step": 9031,
      "sup_loss": 0.4857986585263304
    },
    {
      "train_step": 9032,
      "sup_loss": 0.4686338365988252
    },
    {
      "train_step": 9033,
      "sup_loss": 0.5788185326948017
    },
    {
      "train_step": 9034,
      "sup_loss": 0.6188154141259884
    },
    {
      "train_step": 9035,
      "sup_loss": 0.6346488346428116
    },
    {
      "train_step": 9036,
      "sup_loss": 0.8727866132947825
    },
    {
      "train_step": 9037,
      "sup_loss": 0.6007232197869232
    },
    {
      "train_step": 9038,
      "sup_loss": 0.4984617840527657
    },
    {
      "train_step": 9039,
      "sup_loss": 0.5952279254375923
    },
    {
      "train_step": 9040,
      "sup_loss": 0.5162266774194548
    },
    {
      "train_step": 9041,
      "sup_loss": 0.4328150678583052
    },
    {
      "train_step": 9042,
      "sup_loss": 0.3782192577201243
    },
    {
      "train_step": 9043,
      "sup_loss": 0.6312299529335771
    },
    {
      "train_step": 9044,
      "sup_loss": 0.5749497733219985
    },
    {
      "train_step": 9045,
      "sup_loss": 0.3975640785558123
    },
    {
      "train_step": 9046,
      "sup_loss": 0.4343593232844751
    },
    {
      "train_step": 9047,
      "sup_loss": 0.5757604362560975
    },
    {
      "train_step": 9048,
      "sup_loss": 0.4578216064324182
    },
    {
      "train_step": 9049,
      "sup_loss": 0.6169548348729887
    },
    {
      "train_step": 9050,
      "sup_loss": 0.3189273388096813
    },
    {
      "train_step": 9051,
      "sup_loss": 0.6088296941209728
    },
    {
      "train_step": 9052,
      "sup_loss": 0.4438982743532276
    },
    {
      "train_step": 9053,
      "sup_loss": 0.37353393556952835
    },
    {
      "train_step": 9054,
      "sup_loss": 0.7262302575881382
    },
    {
      "train_step": 9055,
      "sup_loss": 0.5239470682233032
    },
    {
      "train_step": 9056,
      "sup_loss": 0.40788187000844445
    },
    {
      "train_step": 9057,
      "sup_loss": 0.5956777023605503
    },
    {
      "train_step": 9058,
      "sup_loss": 0.5396649540778162
    },
    {
      "train_step": 9059,
      "sup_loss": 0.48190623420086603
    },
    {
      "train_step": 9060,
      "sup_loss": 0.4757953391276812
    },
    {
      "train_step": 9061,
      "sup_loss": 0.4524970206589372
    },
    {
      "train_step": 9062,
      "sup_loss": 0.6240844782484103
    },
    {
      "train_step": 9063,
      "sup_loss": 0.5799679367282987
    },
    {
      "train_step": 9064,
      "sup_loss": 0.3595734975941229
    },
    {
      "train_step": 9065,
      "sup_loss": 0.4463506884189468
    },
    {
      "train_step": 9066,
      "sup_loss": 0.5324818632379349
    },
    {
      "train_step": 9067,
      "sup_loss": 0.490810251758754
    },
    {
      "train_step": 9068,
      "sup_loss": 0.31164027975742103
    },
    {
      "train_step": 9069,
      "sup_loss": 0.578307894039892
    },
    {
      "train_step": 9070,
      "sup_loss": 0.453699273639729
    },
    {
      "train_step": 9071,
      "sup_loss": 0.7314914637776572
    },
    {
      "train_step": 9072,
      "sup_loss": 0.5755951701541745
    },
    {
      "train_step": 9073,
      "sup_loss": 0.681421185676849
    },
    {
      "train_step": 9074,
      "sup_loss": 0.41056425199764357
    },
    {
      "train_step": 9075,
      "sup_loss": 0.5428147011870813
    },
    {
      "train_step": 9076,
      "sup_loss": 0.6157622166122907
    },
    {
      "train_step": 9077,
      "sup_loss": 0.48430329580029036
    },
    {
      "train_step": 9078,
      "sup_loss": 0.3971028865443763
    },
    {
      "train_step": 9079,
      "sup_loss": 0.407374096863935
    },
    {
      "train_step": 9080,
      "sup_loss": 0.7250981260057068
    },
    {
      "train_step": 9081,
      "sup_loss": 0.5080698393448134
    },
    {
      "train_step": 9082,
      "sup_loss": 0.665014314219632
    },
    {
      "train_step": 9083,
      "sup_loss": 0.47052762829302097
    },
    {
      "train_step": 9084,
      "sup_loss": 0.6276772952653843
    },
    {
      "train_step": 9085,
      "sup_loss": 0.497871840299381
    },
    {
      "train_step": 9086,
      "sup_loss": 0.6488545868264262
    },
    {
      "train_step": 9087,
      "sup_loss": 0.5970147735354414
    },
    {
      "train_step": 9088,
      "sup_loss": 0.6192133073339287
    },
    {
      "train_step": 9089,
      "sup_loss": 0.5258724762577486
    },
    {
      "train_step": 9090,
      "sup_loss": 0.5574688564220146
    },
    {
      "train_step": 9091,
      "sup_loss": 0.5634206753798396
    },
    {
      "train_step": 9092,
      "sup_loss": 0.5169228614408115
    },
    {
      "train_step": 9093,
      "sup_loss": 0.678752464362648
    },
    {
      "train_step": 9094,
      "sup_loss": 0.7109559277264829
    },
    {
      "train_step": 9095,
      "sup_loss": 0.5032440999846552
    },
    {
      "train_step": 9096,
      "sup_loss": 0.4378074235895447
    },
    {
      "train_step": 9097,
      "sup_loss": 0.6169515088238683
    },
    {
      "train_step": 9098,
      "sup_loss": 0.5594630450094897
    },
    {
      "train_step": 9099,
      "sup_loss": 0.44768070723292236
    },
    {
      "train_step": 9100,
      "sup_loss": 0.49675193427191877
    },
    {
      "train_step": 9101,
      "sup_loss": 0.5495602692252468
    },
    {
      "train_step": 9102,
      "sup_loss": 0.7450518225603153
    },
    {
      "train_step": 9103,
      "sup_loss": 0.4593565416892589
    },
    {
      "train_step": 9104,
      "sup_loss": 0.5112329749939691
    },
    {
      "train_step": 9105,
      "sup_loss": 0.44366023523875947
    },
    {
      "train_step": 9106,
      "sup_loss": 0.6158849871293176
    },
    {
      "train_step": 9107,
      "sup_loss": 0.6022335099728389
    },
    {
      "train_step": 9108,
      "sup_loss": 0.5950415990541678
    },
    {
      "train_step": 9109,
      "sup_loss": 0.5818188821423751
    },
    {
      "train_step": 9110,
      "sup_loss": 0.7021452905113459
    },
    {
      "train_step": 9111,
      "sup_loss": 0.6420596883130876
    },
    {
      "train_step": 9112,
      "sup_loss": 0.6191245937616445
    },
    {
      "train_step": 9113,
      "sup_loss": 0.6546803150297228
    },
    {
      "train_step": 9114,
      "sup_loss": 0.5615352400562584
    },
    {
      "train_step": 9115,
      "sup_loss": 0.36527855647616403
    },
    {
      "train_step": 9116,
      "sup_loss": 0.5896294294056018
    },
    {
      "train_step": 9117,
      "sup_loss": 0.6539425799744523
    },
    {
      "train_step": 9118,
      "sup_loss": 0.48819322245085295
    },
    {
      "train_step": 9119,
      "sup_loss": 0.7668989017575278
    },
    {
      "train_step": 9120,
      "sup_loss": 0.7263006941090386
    },
    {
      "train_step": 9121,
      "sup_loss": 0.5419606657579611
    },
    {
      "train_step": 9122,
      "sup_loss": 0.5779082602960222
    },
    {
      "train_step": 9123,
      "sup_loss": 0.6052773407054932
    },
    {
      "train_step": 9124,
      "sup_loss": 0.43179875843062077
    },
    {
      "train_step": 9125,
      "sup_loss": 0.4818235581070745
    },
    {
      "train_step": 9126,
      "sup_loss": 0.4464056696323407
    },
    {
      "train_step": 9127,
      "sup_loss": 0.5312752376328409
    },
    {
      "train_step": 9128,
      "sup_loss": 0.7317621347200356
    },
    {
      "train_step": 9129,
      "sup_loss": 0.7740392311790235
    },
    {
      "train_step": 9130,
      "sup_loss": 0.8960623271885846
    },
    {
      "train_step": 9131,
      "sup_loss": 0.3097873367284195
    },
    {
      "train_step": 9132,
      "sup_loss": 0.34606748367496065
    },
    {
      "train_step": 9133,
      "sup_loss": 0.553966218168877
    },
    {
      "train_step": 9134,
      "sup_loss": 0.33240357755080596
    },
    {
      "train_step": 9135,
      "sup_loss": 0.47689404299518107
    },
    {
      "train_step": 9136,
      "sup_loss": 0.29371050990032516
    },
    {
      "train_step": 9137,
      "sup_loss": 0.5439936819545474
    },
    {
      "train_step": 9138,
      "sup_loss": 0.49477581912172397
    },
    {
      "train_step": 9139,
      "sup_loss": 0.464810354430594
    },
    {
      "train_step": 9140,
      "sup_loss": 0.5561551423089465
    },
    {
      "train_step": 9141,
      "sup_loss": 0.44700740355363694
    },
    {
      "train_step": 9142,
      "sup_loss": 0.7348363449212748
    },
    {
      "train_step": 9143,
      "sup_loss": 0.5741295682788179
    },
    {
      "train_step": 9144,
      "sup_loss": 0.4581999299561861
    },
    {
      "train_step": 9145,
      "sup_loss": 0.7874393992307721
    },
    {
      "train_step": 9146,
      "sup_loss": 0.4880359230657798
    },
    {
      "train_step": 9147,
      "sup_loss": 0.49155669961663445
    },
    {
      "train_step": 9148,
      "sup_loss": 0.7274959020944405
    },
    {
      "train_step": 9149,
      "sup_loss": 0.3108539697398508
    },
    {
      "train_step": 9150,
      "sup_loss": 0.4131785626068902
    },
    {
      "train_step": 9151,
      "sup_loss": 0.5993651131350874
    },
    {
      "train_step": 9152,
      "sup_loss": 0.489419390626476
    },
    {
      "train_step": 9153,
      "sup_loss": 0.5216449858959213
    },
    {
      "train_step": 9154,
      "sup_loss": 0.2205411465563387
    },
    {
      "train_step": 9155,
      "sup_loss": 1.121653010585956
    },
    {
      "train_step": 9156,
      "sup_loss": 0.6244148781875384
    },
    {
      "train_step": 9157,
      "sup_loss": 0.497893873406193
    },
    {
      "train_step": 9158,
      "sup_loss": 0.6340825919610302
    },
    {
      "train_step": 9159,
      "sup_loss": 0.3055893246165315
    },
    {
      "train_step": 9160,
      "sup_loss": 0.6236893270123057
    },
    {
      "train_step": 9161,
      "sup_loss": 0.6951699622479577
    },
    {
      "train_step": 9162,
      "sup_loss": 0.4855629329562897
    },
    {
      "train_step": 9163,
      "sup_loss": 0.4981700356152766
    },
    {
      "train_step": 9164,
      "sup_loss": 0.5920153681393445
    },
    {
      "train_step": 9165,
      "sup_loss": 0.43549145384682086
    },
    {
      "train_step": 9166,
      "sup_loss": 0.5341320015351989
    },
    {
      "train_step": 9167,
      "sup_loss": 0.5859251250963053
    },
    {
      "train_step": 9168,
      "sup_loss": 0.3850777779745689
    },
    {
      "train_step": 9169,
      "sup_loss": 0.6338189916108216
    },
    {
      "train_step": 9170,
      "sup_loss": 0.47274622454980403
    },
    {
      "train_step": 9171,
      "sup_loss": 0.3830991232462069
    },
    {
      "train_step": 9172,
      "sup_loss": 0.7097534128013516
    },
    {
      "train_step": 9173,
      "sup_loss": 0.5405903539968313
    },
    {
      "train_step": 9174,
      "sup_loss": 0.3702622903070072
    },
    {
      "train_step": 9175,
      "sup_loss": 0.44855546014216735
    },
    {
      "train_step": 9176,
      "sup_loss": 0.6672826564833614
    },
    {
      "train_step": 9177,
      "sup_loss": 0.505317056287724
    },
    {
      "train_step": 9178,
      "sup_loss": 0.6923170555467789
    },
    {
      "train_step": 9179,
      "sup_loss": 0.48837370102447925
    },
    {
      "train_step": 9180,
      "sup_loss": 0.7000610979313079
    },
    {
      "train_step": 9181,
      "sup_loss": 0.6976843547791405
    },
    {
      "train_step": 9182,
      "sup_loss": 0.5733273730634338
    },
    {
      "train_step": 9183,
      "sup_loss": 0.46109558881394613
    },
    {
      "train_step": 9184,
      "sup_loss": 0.4320245599794613
    },
    {
      "train_step": 9185,
      "sup_loss": 0.6344882895360878
    },
    {
      "train_step": 9186,
      "sup_loss": 0.7122844295951052
    },
    {
      "train_step": 9187,
      "sup_loss": 0.44498741408724346
    },
    {
      "train_step": 9188,
      "sup_loss": 0.5494529267508846
    },
    {
      "train_step": 9189,
      "sup_loss": 0.4083177235196288
    },
    {
      "train_step": 9190,
      "sup_loss": 0.573329680065112
    },
    {
      "train_step": 9191,
      "sup_loss": 0.4366255365742745
    },
    {
      "train_step": 9192,
      "sup_loss": 0.5196678118363716
    },
    {
      "train_step": 9193,
      "sup_loss": 0.799874992349036
    },
    {
      "train_step": 9194,
      "sup_loss": 0.5581807367246078
    },
    {
      "train_step": 9195,
      "sup_loss": 0.7661536586420274
    },
    {
      "train_step": 9196,
      "sup_loss": 0.41669660753964
    },
    {
      "train_step": 9197,
      "sup_loss": 0.6430184662392537
    },
    {
      "train_step": 9198,
      "sup_loss": 0.4726241906853935
    },
    {
      "train_step": 9199,
      "sup_loss": 0.39644774307006236
    },
    {
      "train_step": 9200,
      "sup_loss": 0.5511634109054271
    },
    {
      "train_step": 9201,
      "sup_loss": 0.4539542541077628
    },
    {
      "train_step": 9202,
      "sup_loss": 0.5202446528091937
    },
    {
      "train_step": 9203,
      "sup_loss": 0.5395018601145211
    },
    {
      "train_step": 9204,
      "sup_loss": 0.7785403871520733
    },
    {
      "train_step": 9205,
      "sup_loss": 0.5119249140953608
    },
    {
      "train_step": 9206,
      "sup_loss": 0.6286059700926062
    },
    {
      "train_step": 9207,
      "sup_loss": 0.7146433504429645
    },
    {
      "train_step": 9208,
      "sup_loss": 0.6126327352717342
    },
    {
      "train_step": 9209,
      "sup_loss": 0.663061206837106
    },
    {
      "train_step": 9210,
      "sup_loss": 0.6502456870939497
    },
    {
      "train_step": 9211,
      "sup_loss": 0.6213479514296966
    },
    {
      "train_step": 9212,
      "sup_loss": 0.6495020385431399
    },
    {
      "train_step": 9213,
      "sup_loss": 0.3929977210265017
    },
    {
      "train_step": 9214,
      "sup_loss": 0.5170396706377101
    },
    {
      "train_step": 9215,
      "sup_loss": 0.46059899704450086
    },
    {
      "train_step": 9216,
      "sup_loss": 0.5276681873404533
    },
    {
      "train_step": 9217,
      "sup_loss": 0.5221432574095691
    },
    {
      "train_step": 9218,
      "sup_loss": 0.5262839781918246
    },
    {
      "train_step": 9219,
      "sup_loss": 0.5419370381401821
    },
    {
      "train_step": 9220,
      "sup_loss": 0.49790371903612
    },
    {
      "train_step": 9221,
      "sup_loss": 0.7159988470504522
    },
    {
      "train_step": 9222,
      "sup_loss": 0.6580138389001697
    },
    {
      "train_step": 9223,
      "sup_loss": 0.3710398864698248
    },
    {
      "train_step": 9224,
      "sup_loss": 0.6354743595764509
    },
    {
      "train_step": 9225,
      "sup_loss": 0.6223824924049552
    },
    {
      "train_step": 9226,
      "sup_loss": 0.6146361046167644
    },
    {
      "train_step": 9227,
      "sup_loss": 0.7559857427512662
    },
    {
      "train_step": 9228,
      "sup_loss": 0.5444896266092789
    },
    {
      "train_step": 9229,
      "sup_loss": 0.5131666453892307
    },
    {
      "train_step": 9230,
      "sup_loss": 0.6158047192547313
    },
    {
      "train_step": 9231,
      "sup_loss": 0.5646097601120712
    },
    {
      "train_step": 9232,
      "sup_loss": 0.6372118662078134
    },
    {
      "train_step": 9233,
      "sup_loss": 0.6105986496195805
    },
    {
      "train_step": 9234,
      "sup_loss": 0.46084184955021035
    },
    {
      "train_step": 9235,
      "sup_loss": 0.5222563324776219
    },
    {
      "train_step": 9236,
      "sup_loss": 0.5153087834596554
    },
    {
      "train_step": 9237,
      "sup_loss": 0.5890945278796746
    },
    {
      "train_step": 9238,
      "sup_loss": 0.6353440262294389
    },
    {
      "train_step": 9239,
      "sup_loss": 0.7241369205938645
    },
    {
      "train_step": 9240,
      "sup_loss": 0.395589314161908
    },
    {
      "train_step": 9241,
      "sup_loss": 0.5488916041847562
    },
    {
      "train_step": 9242,
      "sup_loss": 0.5343011700515911
    },
    {
      "train_step": 9243,
      "sup_loss": 0.638964369610741
    },
    {
      "train_step": 9244,
      "sup_loss": 0.43637601425503963
    },
    {
      "train_step": 9245,
      "sup_loss": 0.4937359752180349
    },
    {
      "train_step": 9246,
      "sup_loss": 0.5923234054552041
    },
    {
      "train_step": 9247,
      "sup_loss": 0.4017968489520614
    },
    {
      "train_step": 9248,
      "sup_loss": 0.5865721571562009
    },
    {
      "train_step": 9249,
      "sup_loss": 0.7994249359737857
    },
    {
      "train_step": 9250,
      "sup_loss": 0.4565746287045219
    },
    {
      "train_step": 9251,
      "sup_loss": 0.5954472999919194
    },
    {
      "train_step": 9252,
      "sup_loss": 0.26591835998861396
    },
    {
      "train_step": 9253,
      "sup_loss": 0.4147414145993057
    },
    {
      "train_step": 9254,
      "sup_loss": 0.5093274362953705
    },
    {
      "train_step": 9255,
      "sup_loss": 0.7763389913251333
    },
    {
      "train_step": 9256,
      "sup_loss": 0.7242831284071447
    },
    {
      "train_step": 9257,
      "sup_loss": 0.4869089021132899
    },
    {
      "train_step": 9258,
      "sup_loss": 0.31393428073669566
    },
    {
      "train_step": 9259,
      "sup_loss": 0.5713942245265445
    },
    {
      "train_step": 9260,
      "sup_loss": 0.48775345285351496
    },
    {
      "train_step": 9261,
      "sup_loss": 0.49919575810133243
    },
    {
      "train_step": 9262,
      "sup_loss": 0.6139306891864253
    },
    {
      "train_step": 9263,
      "sup_loss": 0.6462744576199552
    },
    {
      "train_step": 9264,
      "sup_loss": 0.5393105497207901
    },
    {
      "train_step": 9265,
      "sup_loss": 0.6605712097831021
    },
    {
      "train_step": 9266,
      "sup_loss": 0.3303646198467689
    },
    {
      "train_step": 9267,
      "sup_loss": 0.5540449760674535
    },
    {
      "train_step": 9268,
      "sup_loss": 0.5301573668360148
    },
    {
      "train_step": 9269,
      "sup_loss": 0.5224586317209025
    },
    {
      "train_step": 9270,
      "sup_loss": 0.5658029951094836
    },
    {
      "train_step": 9271,
      "sup_loss": 0.5338055238154135
    },
    {
      "train_step": 9272,
      "sup_loss": 0.49509260009087647
    },
    {
      "train_step": 9273,
      "sup_loss": 0.5524668291155304
    },
    {
      "train_step": 9274,
      "sup_loss": 0.756057050713615
    },
    {
      "train_step": 9275,
      "sup_loss": 0.5431258612893363
    },
    {
      "train_step": 9276,
      "sup_loss": 0.4596830373087945
    },
    {
      "train_step": 9277,
      "sup_loss": 0.5452109265591302
    },
    {
      "train_step": 9278,
      "sup_loss": 0.6762063068237607
    },
    {
      "train_step": 9279,
      "sup_loss": 0.6921223657620633
    },
    {
      "train_step": 9280,
      "sup_loss": 0.5597932793028616
    },
    {
      "train_step": 9281,
      "sup_loss": 0.6247724717678577
    },
    {
      "train_step": 9282,
      "sup_loss": 0.4017358808485634
    },
    {
      "train_step": 9283,
      "sup_loss": 0.68998391146746
    },
    {
      "train_step": 9284,
      "sup_loss": 0.5898332126189272
    },
    {
      "train_step": 9285,
      "sup_loss": 0.7576148446937455
    },
    {
      "train_step": 9286,
      "sup_loss": 0.3910641052217199
    },
    {
      "train_step": 9287,
      "sup_loss": 0.657163099458536
    },
    {
      "train_step": 9288,
      "sup_loss": 0.5893752684877016
    },
    {
      "train_step": 9289,
      "sup_loss": 0.6918124603109906
    },
    {
      "train_step": 9290,
      "sup_loss": 0.4896338248694625
    },
    {
      "train_step": 9291,
      "sup_loss": 0.43354641535003946
    },
    {
      "train_step": 9292,
      "sup_loss": 0.567747810058083
    },
    {
      "train_step": 9293,
      "sup_loss": 0.6363683334881387
    },
    {
      "train_step": 9294,
      "sup_loss": 0.4433505036101362
    },
    {
      "train_step": 9295,
      "sup_loss": 0.5229028389919845
    },
    {
      "train_step": 9296,
      "sup_loss": 0.7361529379576407
    },
    {
      "train_step": 9297,
      "sup_loss": 0.4664291560908545
    },
    {
      "train_step": 9298,
      "sup_loss": 0.42421219997512877
    },
    {
      "train_step": 9299,
      "sup_loss": 0.46742434338596706
    },
    {
      "train_step": 9300,
      "sup_loss": 0.5584511288062284
    },
    {
      "train_step": 9301,
      "sup_loss": 0.6541002075834079
    },
    {
      "train_step": 9302,
      "sup_loss": 0.43287334961569607
    },
    {
      "train_step": 9303,
      "sup_loss": 0.47475680318432417
    },
    {
      "train_step": 9304,
      "sup_loss": 0.48205445588815715
    },
    {
      "train_step": 9305,
      "sup_loss": 0.44594936362298054
    },
    {
      "train_step": 9306,
      "sup_loss": 0.8187023510431066
    },
    {
      "train_step": 9307,
      "sup_loss": 0.41575265721154264
    },
    {
      "train_step": 9308,
      "sup_loss": 0.5814308672985024
    },
    {
      "train_step": 9309,
      "sup_loss": 0.6952713616647775
    },
    {
      "train_step": 9310,
      "sup_loss": 0.6191507382163206
    },
    {
      "train_step": 9311,
      "sup_loss": 0.5853990284442231
    },
    {
      "train_step": 9312,
      "sup_loss": 0.4662929462896651
    },
    {
      "train_step": 9313,
      "sup_loss": 0.6848827093286323
    },
    {
      "train_step": 9314,
      "sup_loss": 0.6040534602154363
    },
    {
      "train_step": 9315,
      "sup_loss": 0.6031656674511878
    },
    {
      "train_step": 9316,
      "sup_loss": 0.7321970203673764
    },
    {
      "train_step": 9317,
      "sup_loss": 0.6582666709916607
    },
    {
      "train_step": 9318,
      "sup_loss": 0.5362217881020073
    },
    {
      "train_step": 9319,
      "sup_loss": 0.4640461805353338
    },
    {
      "train_step": 9320,
      "sup_loss": 0.4562169455840719
    },
    {
      "train_step": 9321,
      "sup_loss": 0.4034416592718485
    },
    {
      "train_step": 9322,
      "sup_loss": 0.8612441592035082
    },
    {
      "train_step": 9323,
      "sup_loss": 0.5384429736445074
    },
    {
      "train_step": 9324,
      "sup_loss": 0.32694433699499464
    },
    {
      "train_step": 9325,
      "sup_loss": 0.3339148197407112
    },
    {
      "train_step": 9326,
      "sup_loss": 0.43787689892327875
    },
    {
      "train_step": 9327,
      "sup_loss": 0.6461401593748654
    },
    {
      "train_step": 9328,
      "sup_loss": 0.7202300458847876
    },
    {
      "train_step": 9329,
      "sup_loss": 0.3998948785048506
    },
    {
      "train_step": 9330,
      "sup_loss": 0.5603844453285918
    },
    {
      "train_step": 9331,
      "sup_loss": 0.7967923840397196
    },
    {
      "train_step": 9332,
      "sup_loss": 0.6693263572558061
    },
    {
      "train_step": 9333,
      "sup_loss": 0.6527817892979287
    },
    {
      "train_step": 9334,
      "sup_loss": 1.3973144165208211
    },
    {
      "train_step": 9335,
      "sup_loss": 0.5040941540262179
    },
    {
      "train_step": 9336,
      "sup_loss": 0.6010316205489795
    },
    {
      "train_step": 9337,
      "sup_loss": 0.5211183384897876
    },
    {
      "train_step": 9338,
      "sup_loss": 0.5698458423733664
    },
    {
      "train_step": 9339,
      "sup_loss": 0.5594713455467129
    },
    {
      "train_step": 9340,
      "sup_loss": 0.502373093025798
    },
    {
      "train_step": 9341,
      "sup_loss": 0.37542084960451866
    },
    {
      "train_step": 9342,
      "sup_loss": 0.3700057708526357
    },
    {
      "train_step": 9343,
      "sup_loss": 0.5074381707744264
    },
    {
      "train_step": 9344,
      "sup_loss": 0.5896299007739656
    },
    {
      "train_step": 9345,
      "sup_loss": 0.7042158968735333
    },
    {
      "train_step": 9346,
      "sup_loss": 0.2941257431493464
    },
    {
      "train_step": 9347,
      "sup_loss": 0.5812080725588686
    },
    {
      "train_step": 9348,
      "sup_loss": 0.7804224009915632
    },
    {
      "train_step": 9349,
      "sup_loss": 0.9340239935574884
    },
    {
      "train_step": 9350,
      "sup_loss": 0.5472378417730962
    },
    {
      "train_step": 9351,
      "sup_loss": 0.5854164754176513
    },
    {
      "train_step": 9352,
      "sup_loss": 0.44115117048890273
    },
    {
      "train_step": 9353,
      "sup_loss": 0.5338883627843609
    },
    {
      "train_step": 9354,
      "sup_loss": 0.7740726645464302
    },
    {
      "train_step": 9355,
      "sup_loss": 0.6295343887827655
    },
    {
      "train_step": 9356,
      "sup_loss": 0.5788424665005627
    },
    {
      "train_step": 9357,
      "sup_loss": 0.7045286526711082
    },
    {
      "train_step": 9358,
      "sup_loss": 0.5007848319830981
    },
    {
      "train_step": 9359,
      "sup_loss": 0.5678266023328388
    },
    {
      "train_step": 9360,
      "sup_loss": 0.6503839949570575
    },
    {
      "train_step": 9361,
      "sup_loss": 0.5069757452580794
    },
    {
      "train_step": 9362,
      "sup_loss": 0.48187680530944366
    },
    {
      "train_step": 9363,
      "sup_loss": 0.5905764541509032
    },
    {
      "train_step": 9364,
      "sup_loss": 0.7459181570881164
    },
    {
      "train_step": 9365,
      "sup_loss": 0.7005801178509634
    },
    {
      "train_step": 9366,
      "sup_loss": 0.6150956612332894
    },
    {
      "train_step": 9367,
      "sup_loss": 0.6259520748269229
    },
    {
      "train_step": 9368,
      "sup_loss": 0.5189648562400838
    },
    {
      "train_step": 9369,
      "sup_loss": 0.8051021059194167
    },
    {
      "train_step": 9370,
      "sup_loss": 0.4305640469925954
    },
    {
      "train_step": 9371,
      "sup_loss": 0.5834676404434354
    },
    {
      "train_step": 9372,
      "sup_loss": 0.6737063918351769
    },
    {
      "train_step": 9373,
      "sup_loss": 0.5432170334592319
    },
    {
      "train_step": 9374,
      "sup_loss": 0.592097256048801
    },
    {
      "train_step": 9375,
      "sup_loss": 0.3519231543323644
    },
    {
      "train_step": 9376,
      "sup_loss": 0.44910715940203483
    },
    {
      "train_step": 9377,
      "sup_loss": 0.41821676008795955
    },
    {
      "train_step": 9378,
      "sup_loss": 0.39795304406804854
    },
    {
      "train_step": 9379,
      "sup_loss": 0.5586461969649764
    },
    {
      "train_step": 9380,
      "sup_loss": 0.5033881083069615
    },
    {
      "train_step": 9381,
      "sup_loss": 0.5056468301647968
    },
    {
      "train_step": 9382,
      "sup_loss": 0.5590349047879941
    },
    {
      "train_step": 9383,
      "sup_loss": 0.5212989296074485
    },
    {
      "train_step": 9384,
      "sup_loss": 0.748754761883275
    },
    {
      "train_step": 9385,
      "sup_loss": 0.6853668847970429
    },
    {
      "train_step": 9386,
      "sup_loss": 0.4046459158806546
    },
    {
      "train_step": 9387,
      "sup_loss": 0.5967267248318981
    },
    {
      "train_step": 9388,
      "sup_loss": 0.4318657694100896
    },
    {
      "train_step": 9389,
      "sup_loss": 0.6560305166183853
    },
    {
      "train_step": 9390,
      "sup_loss": 0.5974864048368882
    },
    {
      "train_step": 9391,
      "sup_loss": 0.7149950324905805
    },
    {
      "train_step": 9392,
      "sup_loss": 0.5799230215926144
    },
    {
      "train_step": 9393,
      "sup_loss": 0.4672728870839795
    },
    {
      "train_step": 9394,
      "sup_loss": 0.6995320790155742
    },
    {
      "train_step": 9395,
      "sup_loss": 0.49486118659090617
    },
    {
      "train_step": 9396,
      "sup_loss": 0.5711266423900176
    },
    {
      "train_step": 9397,
      "sup_loss": 0.4902856670194518
    },
    {
      "train_step": 9398,
      "sup_loss": 0.6088877473109514
    },
    {
      "train_step": 9399,
      "sup_loss": 0.5130513701858557
    },
    {
      "train_step": 9400,
      "sup_loss": 0.5140686045043596
    },
    {
      "train_step": 9401,
      "sup_loss": 0.44501556528534086
    },
    {
      "train_step": 9402,
      "sup_loss": 0.5063690127423929
    },
    {
      "train_step": 9403,
      "sup_loss": 0.5555263851329844
    },
    {
      "train_step": 9404,
      "sup_loss": 0.6688267490638
    },
    {
      "train_step": 9405,
      "sup_loss": 0.4101347331362942
    },
    {
      "train_step": 9406,
      "sup_loss": 0.7169018165450628
    },
    {
      "train_step": 9407,
      "sup_loss": 0.5028516773293955
    },
    {
      "train_step": 9408,
      "sup_loss": 0.6188267515308107
    },
    {
      "train_step": 9409,
      "sup_loss": 0.6130322247792779
    },
    {
      "train_step": 9410,
      "sup_loss": 0.5664634148591875
    },
    {
      "train_step": 9411,
      "sup_loss": 0.44498306057955395
    },
    {
      "train_step": 9412,
      "sup_loss": 0.6683464869097343
    },
    {
      "train_step": 9413,
      "sup_loss": 0.4628285163315097
    },
    {
      "train_step": 9414,
      "sup_loss": 0.6342917252477002
    },
    {
      "train_step": 9415,
      "sup_loss": 0.6780571813397445
    },
    {
      "train_step": 9416,
      "sup_loss": 0.654271126288617
    },
    {
      "train_step": 9417,
      "sup_loss": 0.559588986394534
    },
    {
      "train_step": 9418,
      "sup_loss": 0.5417692749199093
    },
    {
      "train_step": 9419,
      "sup_loss": 0.6442118979130017
    },
    {
      "train_step": 9420,
      "sup_loss": 0.5525369124823674
    },
    {
      "train_step": 9421,
      "sup_loss": 0.6755419143081909
    },
    {
      "train_step": 9422,
      "sup_loss": 0.479261365632044
    },
    {
      "train_step": 9423,
      "sup_loss": 0.4974626044927796
    },
    {
      "train_step": 9424,
      "sup_loss": 0.6071807043050987
    },
    {
      "train_step": 9425,
      "sup_loss": 0.6433048229506909
    },
    {
      "train_step": 9426,
      "sup_loss": 0.6395371872758354
    },
    {
      "train_step": 9427,
      "sup_loss": 0.3815463836963814
    },
    {
      "train_step": 9428,
      "sup_loss": 0.39611536308544243
    },
    {
      "train_step": 9429,
      "sup_loss": 0.7941810946171997
    },
    {
      "train_step": 9430,
      "sup_loss": 0.49887833151320937
    },
    {
      "train_step": 9431,
      "sup_loss": 0.2393735416262075
    },
    {
      "train_step": 9432,
      "sup_loss": 0.4626926677195175
    },
    {
      "train_step": 9433,
      "sup_loss": 0.6222065081086533
    },
    {
      "train_step": 9434,
      "sup_loss": 0.6553809422543593
    },
    {
      "train_step": 9435,
      "sup_loss": 0.48505016129763145
    },
    {
      "train_step": 9436,
      "sup_loss": 0.4749197909446046
    },
    {
      "train_step": 9437,
      "sup_loss": 0.5183126704700685
    },
    {
      "train_step": 9438,
      "sup_loss": 0.6229860000793774
    },
    {
      "train_step": 9439,
      "sup_loss": 0.44852555160187213
    },
    {
      "train_step": 9440,
      "sup_loss": 0.46636237947508896
    },
    {
      "train_step": 9441,
      "sup_loss": 0.5406857416784682
    },
    {
      "train_step": 9442,
      "sup_loss": 0.49675326718229823
    },
    {
      "train_step": 9443,
      "sup_loss": 0.6397672753015585
    },
    {
      "train_step": 9444,
      "sup_loss": 0.6009961335189314
    },
    {
      "train_step": 9445,
      "sup_loss": 0.47202771499124385
    },
    {
      "train_step": 9446,
      "sup_loss": 0.6693668364822338
    },
    {
      "train_step": 9447,
      "sup_loss": 0.3977679834381411
    },
    {
      "train_step": 9448,
      "sup_loss": 0.7283402804707784
    },
    {
      "train_step": 9449,
      "sup_loss": 0.7380916558465233
    },
    {
      "train_step": 9450,
      "sup_loss": 0.47275699001318594
    },
    {
      "train_step": 9451,
      "sup_loss": 0.6116943795352104
    },
    {
      "train_step": 9452,
      "sup_loss": 0.5852640184347198
    },
    {
      "train_step": 9453,
      "sup_loss": 0.30920310750509905
    },
    {
      "train_step": 9454,
      "sup_loss": 0.6819216046216464
    },
    {
      "train_step": 9455,
      "sup_loss": 0.6951072800134015
    },
    {
      "train_step": 9456,
      "sup_loss": 0.48612645146919264
    },
    {
      "train_step": 9457,
      "sup_loss": 0.6710227683963234
    },
    {
      "train_step": 9458,
      "sup_loss": 0.7271592211220784
    },
    {
      "train_step": 9459,
      "sup_loss": 0.4197344559123843
    },
    {
      "train_step": 9460,
      "sup_loss": 0.5035641281719568
    },
    {
      "train_step": 9461,
      "sup_loss": 0.5988368509645877
    },
    {
      "train_step": 9462,
      "sup_loss": 0.5636728453484514
    },
    {
      "train_step": 9463,
      "sup_loss": 0.4992718312807465
    },
    {
      "train_step": 9464,
      "sup_loss": 0.412678919252288
    },
    {
      "train_step": 9465,
      "sup_loss": 0.6079840130701077
    },
    {
      "train_step": 9466,
      "sup_loss": 0.47965229618173677
    },
    {
      "train_step": 9467,
      "sup_loss": 0.8518925044832415
    },
    {
      "train_step": 9468,
      "sup_loss": 0.8347713778121746
    },
    {
      "train_step": 9469,
      "sup_loss": 0.46410195320556913
    },
    {
      "train_step": 9470,
      "sup_loss": 0.4935410247792916
    },
    {
      "train_step": 9471,
      "sup_loss": 0.7015058023572265
    },
    {
      "train_step": 9472,
      "sup_loss": 0.4876222052665704
    },
    {
      "train_step": 9473,
      "sup_loss": 0.5800415625894939
    },
    {
      "train_step": 9474,
      "sup_loss": 0.5234991489087537
    },
    {
      "train_step": 9475,
      "sup_loss": 0.40669857018164257
    },
    {
      "train_step": 9476,
      "sup_loss": 0.6208411905788603
    },
    {
      "train_step": 9477,
      "sup_loss": 0.6470897099116842
    },
    {
      "train_step": 9478,
      "sup_loss": 0.421118201058064
    },
    {
      "train_step": 9479,
      "sup_loss": 0.6244616079138368
    },
    {
      "train_step": 9480,
      "sup_loss": 0.6815268861334234
    },
    {
      "train_step": 9481,
      "sup_loss": 0.4219757345624015
    },
    {
      "train_step": 9482,
      "sup_loss": 0.5719307079887431
    },
    {
      "train_step": 9483,
      "sup_loss": 0.35224142472588144
    },
    {
      "train_step": 9484,
      "sup_loss": 0.42648724304113084
    },
    {
      "train_step": 9485,
      "sup_loss": 0.5233290627963882
    },
    {
      "train_step": 9486,
      "sup_loss": 0.3286823729774494
    },
    {
      "train_step": 9487,
      "sup_loss": 0.31521733504907057
    },
    {
      "train_step": 9488,
      "sup_loss": 0.6465211141633813
    },
    {
      "train_step": 9489,
      "sup_loss": 0.7216745869364304
    },
    {
      "train_step": 9490,
      "sup_loss": 0.5331086073785067
    },
    {
      "train_step": 9491,
      "sup_loss": 0.6904611286710817
    },
    {
      "train_step": 9492,
      "sup_loss": 0.4623389666315417
    },
    {
      "train_step": 9493,
      "sup_loss": 0.6205273236792388
    },
    {
      "train_step": 9494,
      "sup_loss": 0.74767076239868
    },
    {
      "train_step": 9495,
      "sup_loss": 0.6724914750541168
    },
    {
      "train_step": 9496,
      "sup_loss": 0.6212051425781806
    },
    {
      "train_step": 9497,
      "sup_loss": 0.4035639457437083
    },
    {
      "train_step": 9498,
      "sup_loss": 0.50640845331402
    },
    {
      "train_step": 9499,
      "sup_loss": 0.46153018327269774
    },
    {
      "train_step": 9500,
      "sup_loss": 0.444355239682049
    },
    {
      "train_step": 9501,
      "sup_loss": 0.40504317743289886
    },
    {
      "train_step": 9502,
      "sup_loss": 0.7201597657250389
    },
    {
      "train_step": 9503,
      "sup_loss": 0.6329741352350804
    },
    {
      "train_step": 9504,
      "sup_loss": 0.5531019677856133
    },
    {
      "train_step": 9505,
      "sup_loss": 0.6065539444811253
    },
    {
      "train_step": 9506,
      "sup_loss": 0.4053015255122271
    },
    {
      "train_step": 9507,
      "sup_loss": 0.24656174715123036
    },
    {
      "train_step": 9508,
      "sup_loss": 0.6398503950786483
    },
    {
      "train_step": 9509,
      "sup_loss": 0.5210345734867255
    },
    {
      "train_step": 9510,
      "sup_loss": 0.6649631359674306
    },
    {
      "train_step": 9511,
      "sup_loss": 0.597669703249402
    },
    {
      "train_step": 9512,
      "sup_loss": 0.4076735760966789
    },
    {
      "train_step": 9513,
      "sup_loss": 0.6202271904018101
    },
    {
      "train_step": 9514,
      "sup_loss": 0.676972813455796
    },
    {
      "train_step": 9515,
      "sup_loss": 0.34347340176479146
    },
    {
      "train_step": 9516,
      "sup_loss": 0.5923824295733187
    },
    {
      "train_step": 9517,
      "sup_loss": 0.749800253815947
    },
    {
      "train_step": 9518,
      "sup_loss": 0.5987809948358095
    },
    {
      "train_step": 9519,
      "sup_loss": 0.4560290410128994
    },
    {
      "train_step": 9520,
      "sup_loss": 0.6498550098884089
    },
    {
      "train_step": 9521,
      "sup_loss": 0.38236734850374504
    },
    {
      "train_step": 9522,
      "sup_loss": 0.5399481795963463
    },
    {
      "train_step": 9523,
      "sup_loss": 0.5702027186752178
    },
    {
      "train_step": 9524,
      "sup_loss": 0.48623430298787124
    },
    {
      "train_step": 9525,
      "sup_loss": 0.6294098601952419
    },
    {
      "train_step": 9526,
      "sup_loss": 0.15016843194375307
    },
    {
      "train_step": 9527,
      "sup_loss": 0.51537376307593
    },
    {
      "train_step": 9528,
      "sup_loss": 0.5668889627259684
    },
    {
      "train_step": 9529,
      "sup_loss": 0.5825626522092164
    },
    {
      "train_step": 9530,
      "sup_loss": 0.5133507526779161
    },
    {
      "train_step": 9531,
      "sup_loss": 0.5329371118905536
    },
    {
      "train_step": 9532,
      "sup_loss": 0.6541039538944403
    },
    {
      "train_step": 9533,
      "sup_loss": 0.5879186981995697
    },
    {
      "train_step": 9534,
      "sup_loss": 0.538503934552458
    },
    {
      "train_step": 9535,
      "sup_loss": 0.6184567375152369
    },
    {
      "train_step": 9536,
      "sup_loss": 0.6524675766837081
    },
    {
      "train_step": 9537,
      "sup_loss": 0.6967741384026138
    },
    {
      "train_step": 9538,
      "sup_loss": 0.4382787168451676
    },
    {
      "train_step": 9539,
      "sup_loss": 0.5678391539078831
    },
    {
      "train_step": 9540,
      "sup_loss": 0.6791549054430643
    },
    {
      "train_step": 9541,
      "sup_loss": 0.3919680962661765
    },
    {
      "train_step": 9542,
      "sup_loss": 0.8042736757333532
    },
    {
      "train_step": 9543,
      "sup_loss": 0.544468911220517
    },
    {
      "train_step": 9544,
      "sup_loss": 0.4063950830923949
    },
    {
      "train_step": 9545,
      "sup_loss": 0.8075662458052346
    },
    {
      "train_step": 9546,
      "sup_loss": 0.4858655346499294
    },
    {
      "train_step": 9547,
      "sup_loss": 0.711490734396014
    },
    {
      "train_step": 9548,
      "sup_loss": 0.7886672967310757
    },
    {
      "train_step": 9549,
      "sup_loss": 0.5505170772715157
    },
    {
      "train_step": 9550,
      "sup_loss": 0.45676362377331325
    },
    {
      "train_step": 9551,
      "sup_loss": 0.4747955142567957
    },
    {
      "train_step": 9552,
      "sup_loss": 0.48516749992564534
    },
    {
      "train_step": 9553,
      "sup_loss": 0.5822158568756478
    },
    {
      "train_step": 9554,
      "sup_loss": 0.6683789210684546
    },
    {
      "train_step": 9555,
      "sup_loss": 0.6469894357069874
    },
    {
      "train_step": 9556,
      "sup_loss": 0.6121946541203409
    },
    {
      "train_step": 9557,
      "sup_loss": 0.7918673143553769
    },
    {
      "train_step": 9558,
      "sup_loss": 0.3650917705390889
    },
    {
      "train_step": 9559,
      "sup_loss": 0.5660435846355131
    },
    {
      "train_step": 9560,
      "sup_loss": 0.2914349420284051
    },
    {
      "train_step": 9561,
      "sup_loss": 0.5030198823326671
    },
    {
      "train_step": 9562,
      "sup_loss": 0.7012158659954646
    },
    {
      "train_step": 9563,
      "sup_loss": 0.39306006122514253
    },
    {
      "train_step": 9564,
      "sup_loss": 0.5480571549054856
    },
    {
      "train_step": 9565,
      "sup_loss": 0.3756762679779178
    },
    {
      "train_step": 9566,
      "sup_loss": 0.5587627145140066
    },
    {
      "train_step": 9567,
      "sup_loss": 0.7712497099807343
    },
    {
      "train_step": 9568,
      "sup_loss": 0.7318508364735792
    },
    {
      "train_step": 9569,
      "sup_loss": 0.6021630520539647
    },
    {
      "train_step": 9570,
      "sup_loss": 0.5608068510037222
    },
    {
      "train_step": 9571,
      "sup_loss": 0.6788944724070235
    },
    {
      "train_step": 9572,
      "sup_loss": 0.6405859611927991
    },
    {
      "train_step": 9573,
      "sup_loss": 0.49076837523799455
    },
    {
      "train_step": 9574,
      "sup_loss": 0.5649346244419261
    },
    {
      "train_step": 9575,
      "sup_loss": 0.33749581422257313
    },
    {
      "train_step": 9576,
      "sup_loss": 0.5086587876073446
    },
    {
      "train_step": 9577,
      "sup_loss": 0.7813579601943286
    },
    {
      "train_step": 9578,
      "sup_loss": 0.6489339960087984
    },
    {
      "train_step": 9579,
      "sup_loss": 0.39867355000152527
    },
    {
      "train_step": 9580,
      "sup_loss": 0.49020150813874097
    },
    {
      "train_step": 9581,
      "sup_loss": 0.678663074736676
    },
    {
      "train_step": 9582,
      "sup_loss": 0.7224755496624694
    },
    {
      "train_step": 9583,
      "sup_loss": 0.7172743183586888
    },
    {
      "train_step": 9584,
      "sup_loss": 0.433701747050195
    },
    {
      "train_step": 9585,
      "sup_loss": 0.45998966012556514
    },
    {
      "train_step": 9586,
      "sup_loss": 0.6250039356101711
    },
    {
      "train_step": 9587,
      "sup_loss": 0.7036238555191446
    },
    {
      "train_step": 9588,
      "sup_loss": 0.36480791575954064
    },
    {
      "train_step": 9589,
      "sup_loss": 0.7172336533443359
    },
    {
      "train_step": 9590,
      "sup_loss": 0.5142952613359171
    },
    {
      "train_step": 9591,
      "sup_loss": 0.640366504852322
    },
    {
      "train_step": 9592,
      "sup_loss": 0.8209648121117876
    },
    {
      "train_step": 9593,
      "sup_loss": 0.512906680762978
    },
    {
      "train_step": 9594,
      "sup_loss": 0.5664860580410239
    },
    {
      "train_step": 9595,
      "sup_loss": 0.356399742681933
    },
    {
      "train_step": 9596,
      "sup_loss": 0.5883559481157778
    },
    {
      "train_step": 9597,
      "sup_loss": 0.50142524343439
    },
    {
      "train_step": 9598,
      "sup_loss": 0.6020125009747201
    },
    {
      "train_step": 9599,
      "sup_loss": 0.48983070146457974
    },
    {
      "train_step": 9600,
      "sup_loss": 0.6914285259471532
    },
    {
      "train_step": 9601,
      "sup_loss": 0.3540863258423052
    },
    {
      "train_step": 9602,
      "sup_loss": 0.5184491762623026
    },
    {
      "train_step": 9603,
      "sup_loss": 0.7081008160868401
    },
    {
      "train_step": 9604,
      "sup_loss": 0.8012116657913297
    },
    {
      "train_step": 9605,
      "sup_loss": 0.5179351703171592
    },
    {
      "train_step": 9606,
      "sup_loss": 0.4537322509036912
    },
    {
      "train_step": 9607,
      "sup_loss": 0.6293576805391669
    },
    {
      "train_step": 9608,
      "sup_loss": 0.5511149676298432
    },
    {
      "train_step": 9609,
      "sup_loss": 0.4136126593667645
    },
    {
      "train_step": 9610,
      "sup_loss": 0.2786210401265494
    },
    {
      "train_step": 9611,
      "sup_loss": 0.55557574823729
    },
    {
      "train_step": 9612,
      "sup_loss": 0.6532774087127372
    },
    {
      "train_step": 9613,
      "sup_loss": 0.6372252168026507
    },
    {
      "train_step": 9614,
      "sup_loss": 0.41370356615068915
    },
    {
      "train_step": 9615,
      "sup_loss": 0.5373499167966569
    },
    {
      "train_step": 9616,
      "sup_loss": 0.44507433632950244
    },
    {
      "train_step": 9617,
      "sup_loss": 0.516120009997365
    },
    {
      "train_step": 9618,
      "sup_loss": 0.5443413155820148
    },
    {
      "train_step": 9619,
      "sup_loss": 0.7437910594592445
    },
    {
      "train_step": 9620,
      "sup_loss": 0.6922422185458594
    },
    {
      "train_step": 9621,
      "sup_loss": 0.5381703666237159
    },
    {
      "train_step": 9622,
      "sup_loss": 0.4854570641534152
    },
    {
      "train_step": 9623,
      "sup_loss": 0.7318970230236522
    },
    {
      "train_step": 9624,
      "sup_loss": 0.5092903723732272
    },
    {
      "train_step": 9625,
      "sup_loss": 0.7816088553662762
    },
    {
      "train_step": 9626,
      "sup_loss": 0.40139644352023474
    },
    {
      "train_step": 9627,
      "sup_loss": 0.5629761459408329
    },
    {
      "train_step": 9628,
      "sup_loss": 0.810803523652056
    },
    {
      "train_step": 9629,
      "sup_loss": 0.710317511663844
    },
    {
      "train_step": 9630,
      "sup_loss": 0.38200376815153975
    },
    {
      "train_step": 9631,
      "sup_loss": 0.626897329203437
    },
    {
      "train_step": 9632,
      "sup_loss": 0.6277689734770969
    },
    {
      "train_step": 9633,
      "sup_loss": 0.5833560012656253
    },
    {
      "train_step": 9634,
      "sup_loss": 0.608135285301854
    },
    {
      "train_step": 9635,
      "sup_loss": 0.5017531010143929
    },
    {
      "train_step": 9636,
      "sup_loss": 0.6514960719136659
    },
    {
      "train_step": 9637,
      "sup_loss": 0.6218473840412848
    },
    {
      "train_step": 9638,
      "sup_loss": 0.5259211524944802
    },
    {
      "train_step": 9639,
      "sup_loss": 0.5890703296648074
    },
    {
      "train_step": 9640,
      "sup_loss": 0.775437446690239
    },
    {
      "train_step": 9641,
      "sup_loss": 0.6055683035707131
    },
    {
      "train_step": 9642,
      "sup_loss": 0.4196880767713335
    },
    {
      "train_step": 9643,
      "sup_loss": 0.6005995350295656
    },
    {
      "train_step": 9644,
      "sup_loss": 0.6682658444784755
    },
    {
      "train_step": 9645,
      "sup_loss": 0.5869143294590634
    },
    {
      "train_step": 9646,
      "sup_loss": 0.49982169093473844
    },
    {
      "train_step": 9647,
      "sup_loss": 0.4679543170653463
    },
    {
      "train_step": 9648,
      "sup_loss": 0.5290861605836205
    },
    {
      "train_step": 9649,
      "sup_loss": 0.5002922270220049
    },
    {
      "train_step": 9650,
      "sup_loss": 0.44240034363652664
    },
    {
      "train_step": 9651,
      "sup_loss": 0.5986742582106249
    },
    {
      "train_step": 9652,
      "sup_loss": 0.4582700837824564
    },
    {
      "train_step": 9653,
      "sup_loss": 0.5485350783574262
    },
    {
      "train_step": 9654,
      "sup_loss": 0.6418677736351439
    },
    {
      "train_step": 9655,
      "sup_loss": 0.673504437546886
    },
    {
      "train_step": 9656,
      "sup_loss": 0.5732188060862616
    },
    {
      "train_step": 9657,
      "sup_loss": 0.5077859376524029
    },
    {
      "train_step": 9658,
      "sup_loss": 0.5107119572863006
    },
    {
      "train_step": 9659,
      "sup_loss": 0.561159076581184
    },
    {
      "train_step": 9660,
      "sup_loss": 0.5081933776546929
    },
    {
      "train_step": 9661,
      "sup_loss": 0.5539584759035626
    },
    {
      "train_step": 9662,
      "sup_loss": 0.35907725839432714
    },
    {
      "train_step": 9663,
      "sup_loss": 0.5090498746659121
    },
    {
      "train_step": 9664,
      "sup_loss": 0.7011274642265429
    },
    {
      "train_step": 9665,
      "sup_loss": 0.6839446167721283
    },
    {
      "train_step": 9666,
      "sup_loss": 0.40276378284371955
    },
    {
      "train_step": 9667,
      "sup_loss": 0.6314029762136842
    },
    {
      "train_step": 9668,
      "sup_loss": 0.5184376816703732
    },
    {
      "train_step": 9669,
      "sup_loss": 0.6467871265699233
    },
    {
      "train_step": 9670,
      "sup_loss": 0.4717022119603055
    },
    {
      "train_step": 9671,
      "sup_loss": 0.7252792928126266
    },
    {
      "train_step": 9672,
      "sup_loss": 0.6038823870635822
    },
    {
      "train_step": 9673,
      "sup_loss": 0.7231333100797499
    },
    {
      "train_step": 9674,
      "sup_loss": 0.5516951457603024
    },
    {
      "train_step": 9675,
      "sup_loss": 0.9100046269968342
    },
    {
      "train_step": 9676,
      "sup_loss": 0.31965656621373156
    },
    {
      "train_step": 9677,
      "sup_loss": 0.3874619919916081
    },
    {
      "train_step": 9678,
      "sup_loss": 0.4701032217192047
    },
    {
      "train_step": 9679,
      "sup_loss": 0.6760304111703478
    },
    {
      "train_step": 9680,
      "sup_loss": 0.5583542493043527
    },
    {
      "train_step": 9681,
      "sup_loss": 0.6606716327616723
    },
    {
      "train_step": 9682,
      "sup_loss": 0.6598655148673732
    },
    {
      "train_step": 9683,
      "sup_loss": 0.6319233351243024
    },
    {
      "train_step": 9684,
      "sup_loss": 0.4878275541374334
    },
    {
      "train_step": 9685,
      "sup_loss": 0.5384885416602707
    },
    {
      "train_step": 9686,
      "sup_loss": 0.49535116635232695
    },
    {
      "train_step": 9687,
      "sup_loss": 0.4175726423102092
    },
    {
      "train_step": 9688,
      "sup_loss": 0.7356051626752121
    },
    {
      "train_step": 9689,
      "sup_loss": 0.7293489100001928
    },
    {
      "train_step": 9690,
      "sup_loss": 0.4929778465726145
    },
    {
      "train_step": 9691,
      "sup_loss": 0.6330727061915896
    },
    {
      "train_step": 9692,
      "sup_loss": 0.5692031278723423
    },
    {
      "train_step": 9693,
      "sup_loss": 0.5374413278309497
    },
    {
      "train_step": 9694,
      "sup_loss": 0.4315503418116766
    },
    {
      "train_step": 9695,
      "sup_loss": 0.585773859340286
    },
    {
      "train_step": 9696,
      "sup_loss": 0.3918647572324325
    },
    {
      "train_step": 9697,
      "sup_loss": 0.595853387386591
    },
    {
      "train_step": 9698,
      "sup_loss": 0.7891902214095723
    },
    {
      "train_step": 9699,
      "sup_loss": 0.5589624259121833
    },
    {
      "train_step": 9700,
      "sup_loss": 0.6216667894127811
    },
    {
      "train_step": 9701,
      "sup_loss": 0.5112652694267179
    },
    {
      "train_step": 9702,
      "sup_loss": 0.8228841235614665
    },
    {
      "train_step": 9703,
      "sup_loss": 0.47343779489707394
    },
    {
      "train_step": 9704,
      "sup_loss": 0.34642074176579657
    },
    {
      "train_step": 9705,
      "sup_loss": 0.5852393530125848
    },
    {
      "train_step": 9706,
      "sup_loss": 0.7763391470831149
    },
    {
      "train_step": 9707,
      "sup_loss": 0.4026364555545824
    },
    {
      "train_step": 9708,
      "sup_loss": 0.5497325522182721
    },
    {
      "train_step": 9709,
      "sup_loss": 0.7328462326218752
    },
    {
      "train_step": 9710,
      "sup_loss": 0.4713913880271602
    },
    {
      "train_step": 9711,
      "sup_loss": 0.6492279063407511
    },
    {
      "train_step": 9712,
      "sup_loss": 0.7350944631819457
    },
    {
      "train_step": 9713,
      "sup_loss": 0.5854518745086978
    },
    {
      "train_step": 9714,
      "sup_loss": 0.6698086742899243
    },
    {
      "train_step": 9715,
      "sup_loss": 0.6166587931520717
    },
    {
      "train_step": 9716,
      "sup_loss": 0.5536645090298172
    },
    {
      "train_step": 9717,
      "sup_loss": 0.49935059808088517
    },
    {
      "train_step": 9718,
      "sup_loss": 0.4979741636881948
    },
    {
      "train_step": 9719,
      "sup_loss": 0.5592970433853739
    },
    {
      "train_step": 9720,
      "sup_loss": 0.6734783075105488
    },
    {
      "train_step": 9721,
      "sup_loss": 0.47215687189975897
    },
    {
      "train_step": 9722,
      "sup_loss": 0.5204078221757942
    },
    {
      "train_step": 9723,
      "sup_loss": 0.5002685011337186
    },
    {
      "train_step": 9724,
      "sup_loss": 0.4502533609021278
    },
    {
      "train_step": 9725,
      "sup_loss": 0.46991698526623904
    },
    {
      "train_step": 9726,
      "sup_loss": 0.6160475559389073
    },
    {
      "train_step": 9727,
      "sup_loss": 0.538120829705429
    },
    {
      "train_step": 9728,
      "sup_loss": 0.6499341293661185
    },
    {
      "train_step": 9729,
      "sup_loss": 0.814758374570753
    },
    {
      "train_step": 9730,
      "sup_loss": 0.6534637841756648
    },
    {
      "train_step": 9731,
      "sup_loss": 0.6779624530519739
    },
    {
      "train_step": 9732,
      "sup_loss": 0.5071878216861163
    },
    {
      "train_step": 9733,
      "sup_loss": 0.5693862862180422
    },
    {
      "train_step": 9734,
      "sup_loss": 0.4462340379947757
    },
    {
      "train_step": 9735,
      "sup_loss": 0.44828971591378125
    },
    {
      "train_step": 9736,
      "sup_loss": 0.5163196562649227
    },
    {
      "train_step": 9737,
      "sup_loss": 0.832550563891029
    },
    {
      "train_step": 9738,
      "sup_loss": 0.4666647766756344
    },
    {
      "train_step": 9739,
      "sup_loss": 0.5131841692169073
    },
    {
      "train_step": 9740,
      "sup_loss": 0.5470463691258298
    },
    {
      "train_step": 9741,
      "sup_loss": 0.5972495305425632
    },
    {
      "train_step": 9742,
      "sup_loss": 0.592733539928792
    },
    {
      "train_step": 9743,
      "sup_loss": 0.4987711811417206
    },
    {
      "train_step": 9744,
      "sup_loss": 0.598614259677165
    },
    {
      "train_step": 9745,
      "sup_loss": 0.3954248873252851
    },
    {
      "train_step": 9746,
      "sup_loss": 0.4696128754258996
    },
    {
      "train_step": 9747,
      "sup_loss": 0.5920032618812267
    },
    {
      "train_step": 9748,
      "sup_loss": 0.7702539397721023
    },
    {
      "train_step": 9749,
      "sup_loss": 0.38928342325380577
    },
    {
      "train_step": 9750,
      "sup_loss": 0.6650121421729371
    },
    {
      "train_step": 9751,
      "sup_loss": 0.7911949268299485
    },
    {
      "train_step": 9752,
      "sup_loss": 0.4803508312725518
    },
    {
      "train_step": 9753,
      "sup_loss": 0.651913903930968
    },
    {
      "train_step": 9754,
      "sup_loss": 0.6527071125332509
    },
    {
      "train_step": 9755,
      "sup_loss": 0.6258920321141175
    },
    {
      "train_step": 9756,
      "sup_loss": 0.5113876320700766
    },
    {
      "train_step": 9757,
      "sup_loss": 0.6739197116202226
    },
    {
      "train_step": 9758,
      "sup_loss": 0.4950468763652438
    },
    {
      "train_step": 9759,
      "sup_loss": 0.6105571045230465
    },
    {
      "train_step": 9760,
      "sup_loss": 0.5223269896353637
    },
    {
      "train_step": 9761,
      "sup_loss": 0.5518761858076477
    },
    {
      "train_step": 9762,
      "sup_loss": 0.6060769899250762
    },
    {
      "train_step": 9763,
      "sup_loss": 0.3450528701568464
    },
    {
      "train_step": 9764,
      "sup_loss": 0.35878049146946356
    },
    {
      "train_step": 9765,
      "sup_loss": 0.38505035603606175
    },
    {
      "train_step": 9766,
      "sup_loss": 0.5587998964822416
    },
    {
      "train_step": 9767,
      "sup_loss": 0.5137961914185551
    },
    {
      "train_step": 9768,
      "sup_loss": 0.5564100560034455
    },
    {
      "train_step": 9769,
      "sup_loss": 0.6603617756304186
    },
    {
      "train_step": 9770,
      "sup_loss": 0.4498509621155012
    },
    {
      "train_step": 9771,
      "sup_loss": 0.7004222190946873
    },
    {
      "train_step": 9772,
      "sup_loss": 0.7695559291215722
    },
    {
      "train_step": 9773,
      "sup_loss": 0.4993611176674768
    },
    {
      "train_step": 9774,
      "sup_loss": 0.8322485861257277
    },
    {
      "train_step": 9775,
      "sup_loss": 0.5564273697044071
    },
    {
      "train_step": 9776,
      "sup_loss": 0.3599554774559269
    },
    {
      "train_step": 9777,
      "sup_loss": 0.6503559547617486
    },
    {
      "train_step": 9778,
      "sup_loss": 0.3703686868727485
    },
    {
      "train_step": 9779,
      "sup_loss": 0.20491875372810844
    },
    {
      "train_step": 9780,
      "sup_loss": 0.5881934287540038
    },
    {
      "train_step": 9781,
      "sup_loss": 0.5427689792263923
    },
    {
      "train_step": 9782,
      "sup_loss": 0.4121472512628308
    },
    {
      "train_step": 9783,
      "sup_loss": 0.5989505077626283
    },
    {
      "train_step": 9784,
      "sup_loss": 0.7589307217131971
    },
    {
      "train_step": 9785,
      "sup_loss": 0.4032861541498897
    },
    {
      "train_step": 9786,
      "sup_loss": 0.5648729071209679
    },
    {
      "train_step": 9787,
      "sup_loss": 0.5244269905950261
    },
    {
      "train_step": 9788,
      "sup_loss": 0.5656813872341117
    },
    {
      "train_step": 9789,
      "sup_loss": 0.44731094326194976
    },
    {
      "train_step": 9790,
      "sup_loss": 0.6227713615185364
    },
    {
      "train_step": 9791,
      "sup_loss": 0.5588422689073195
    },
    {
      "train_step": 9792,
      "sup_loss": 0.26969653358313767
    },
    {
      "train_step": 9793,
      "sup_loss": 0.6360184328766778
    },
    {
      "train_step": 9794,
      "sup_loss": 0.6684995443067976
    },
    {
      "train_step": 9795,
      "sup_loss": 0.5095984711326229
    },
    {
      "train_step": 9796,
      "sup_loss": 0.5578759887010117
    },
    {
      "train_step": 9797,
      "sup_loss": 0.6072598780643916
    },
    {
      "train_step": 9798,
      "sup_loss": 0.8512320855208535
    },
    {
      "train_step": 9799,
      "sup_loss": 0.4241326739112016
    },
    {
      "train_step": 9800,
      "sup_loss": 0.6827023871849774
    },
    {
      "train_step": 9801,
      "sup_loss": 0.6264473827112408
    },
    {
      "train_step": 9802,
      "sup_loss": 0.5905242260756312
    },
    {
      "train_step": 9803,
      "sup_loss": 0.7013894657584789
    },
    {
      "train_step": 9804,
      "sup_loss": 0.5270829749080163
    },
    {
      "train_step": 9805,
      "sup_loss": 0.22208963301981405
    },
    {
      "train_step": 9806,
      "sup_loss": 0.5930414759900366
    },
    {
      "train_step": 9807,
      "sup_loss": 0.3042993338619385
    },
    {
      "train_step": 9808,
      "sup_loss": 0.6503281729618628
    },
    {
      "train_step": 9809,
      "sup_loss": 0.6455238000685308
    },
    {
      "train_step": 9810,
      "sup_loss": 0.4702910833504848
    },
    {
      "train_step": 9811,
      "sup_loss": 0.3288362326804918
    },
    {
      "train_step": 9812,
      "sup_loss": 0.6215894783585506
    },
    {
      "train_step": 9813,
      "sup_loss": 0.48672283230962743
    },
    {
      "train_step": 9814,
      "sup_loss": 0.5831317043061881
    },
    {
      "train_step": 9815,
      "sup_loss": 0.3618022637531916
    },
    {
      "train_step": 9816,
      "sup_loss": 0.6276999139675375
    },
    {
      "train_step": 9817,
      "sup_loss": 0.6715679926274724
    },
    {
      "train_step": 9818,
      "sup_loss": 0.5056268637189867
    },
    {
      "train_step": 9819,
      "sup_loss": 0.6257006544476021
    },
    {
      "train_step": 9820,
      "sup_loss": 0.7777699144771593
    },
    {
      "train_step": 9821,
      "sup_loss": 0.7513371147885544
    },
    {
      "train_step": 9822,
      "sup_loss": 0.7136155625973606
    },
    {
      "train_step": 9823,
      "sup_loss": 0.6497629499780149
    },
    {
      "train_step": 9824,
      "sup_loss": 0.46901001770773476
    },
    {
      "train_step": 9825,
      "sup_loss": 0.8001518610604393
    },
    {
      "train_step": 9826,
      "sup_loss": 0.5222409254291316
    },
    {
      "train_step": 9827,
      "sup_loss": 0.45271060881495023
    },
    {
      "train_step": 9828,
      "sup_loss": 0.6112205620144104
    },
    {
      "train_step": 9829,
      "sup_loss": 0.4874724676755591
    },
    {
      "train_step": 9830,
      "sup_loss": 0.399614896932061
    },
    {
      "train_step": 9831,
      "sup_loss": 0.7478085100027516
    },
    {
      "train_step": 9832,
      "sup_loss": 0.425002342724102
    },
    {
      "train_step": 9833,
      "sup_loss": 0.4151445407853818
    },
    {
      "train_step": 9834,
      "sup_loss": 0.606467528572555
    },
    {
      "train_step": 9835,
      "sup_loss": 0.6289593063486774
    },
    {
      "train_step": 9836,
      "sup_loss": 0.6731621812959557
    },
    {
      "train_step": 9837,
      "sup_loss": 0.7321821733739643
    },
    {
      "train_step": 9838,
      "sup_loss": 0.6313867324044143
    },
    {
      "train_step": 9839,
      "sup_loss": 0.5881375834190553
    },
    {
      "train_step": 9840,
      "sup_loss": 0.6156190486852275
    },
    {
      "train_step": 9841,
      "sup_loss": 0.46227532201208604
    },
    {
      "train_step": 9842,
      "sup_loss": 0.7970366641228996
    },
    {
      "train_step": 9843,
      "sup_loss": 0.7788744330889155
    },
    {
      "train_step": 9844,
      "sup_loss": 0.5613123168813124
    },
    {
      "train_step": 9845,
      "sup_loss": 0.6609730647861338
    },
    {
      "train_step": 9846,
      "sup_loss": 0.42097731632143953
    },
    {
      "train_step": 9847,
      "sup_loss": 0.34820616497031687
    },
    {
      "train_step": 9848,
      "sup_loss": 0.420574990865407
    },
    {
      "train_step": 9849,
      "sup_loss": 0.6498405929162652
    },
    {
      "train_step": 9850,
      "sup_loss": 0.33921505613440306
    },
    {
      "train_step": 9851,
      "sup_loss": 0.5424599549733837
    },
    {
      "train_step": 9852,
      "sup_loss": 0.7426453601830751
    },
    {
      "train_step": 9853,
      "sup_loss": 0.6613573061242933
    },
    {
      "train_step": 9854,
      "sup_loss": 0.534131453794151
    },
    {
      "train_step": 9855,
      "sup_loss": 0.39178089862409665
    },
    {
      "train_step": 9856,
      "sup_loss": 0.639892171394951
    },
    {
      "train_step": 9857,
      "sup_loss": 0.5868124256920177
    },
    {
      "train_step": 9858,
      "sup_loss": 0.4057208521941193
    },
    {
      "train_step": 9859,
      "sup_loss": 0.5542889043082726
    },
    {
      "train_step": 9860,
      "sup_loss": 0.5610450993381373
    },
    {
      "train_step": 9861,
      "sup_loss": 0.6211263848037414
    },
    {
      "train_step": 9862,
      "sup_loss": 0.5121253179091603
    },
    {
      "train_step": 9863,
      "sup_loss": 0.6616812827695416
    },
    {
      "train_step": 9864,
      "sup_loss": 0.6249276268873097
    },
    {
      "train_step": 9865,
      "sup_loss": 0.5830278458278813
    },
    {
      "train_step": 9866,
      "sup_loss": 0.568028922335423
    },
    {
      "train_step": 9867,
      "sup_loss": 0.6604727394287915
    },
    {
      "train_step": 9868,
      "sup_loss": 0.6708306026922324
    },
    {
      "train_step": 9869,
      "sup_loss": 0.5736697121348514
    },
    {
      "train_step": 9870,
      "sup_loss": 0.31635922191138716
    },
    {
      "train_step": 9871,
      "sup_loss": 0.5408240618390904
    },
    {
      "train_step": 9872,
      "sup_loss": 0.37808045734886
    },
    {
      "train_step": 9873,
      "sup_loss": 0.5572826439468858
    },
    {
      "train_step": 9874,
      "sup_loss": 0.5209886314570622
    },
    {
      "train_step": 9875,
      "sup_loss": 0.527549348247785
    },
    {
      "train_step": 9876,
      "sup_loss": 0.5303394743542875
    },
    {
      "train_step": 9877,
      "sup_loss": 0.5544902403073398
    },
    {
      "train_step": 9878,
      "sup_loss": 0.6094821975586361
    },
    {
      "train_step": 9879,
      "sup_loss": 0.5675857482194446
    },
    {
      "train_step": 9880,
      "sup_loss": 0.4909138901925767
    },
    {
      "train_step": 9881,
      "sup_loss": 0.6057262880132768
    },
    {
      "train_step": 9882,
      "sup_loss": 0.6283898043309318
    },
    {
      "train_step": 9883,
      "sup_loss": 0.6615687626499326
    },
    {
      "train_step": 9884,
      "sup_loss": 0.6449673586540448
    },
    {
      "train_step": 9885,
      "sup_loss": 0.7248523487234514
    },
    {
      "train_step": 9886,
      "sup_loss": 0.6746991521268341
    },
    {
      "train_step": 9887,
      "sup_loss": 0.4614818642542782
    },
    {
      "train_step": 9888,
      "sup_loss": 0.6886352862073247
    },
    {
      "train_step": 9889,
      "sup_loss": 0.4151192381930596
    },
    {
      "train_step": 9890,
      "sup_loss": 0.5235814846902853
    },
    {
      "train_step": 9891,
      "sup_loss": 0.5719220252130109
    },
    {
      "train_step": 9892,
      "sup_loss": 0.42804233985128076
    },
    {
      "train_step": 9893,
      "sup_loss": 0.7254433599155065
    },
    {
      "train_step": 9894,
      "sup_loss": 0.5637087173166206
    },
    {
      "train_step": 9895,
      "sup_loss": 0.5869396567020525
    },
    {
      "train_step": 9896,
      "sup_loss": 0.3834723384808709
    },
    {
      "train_step": 9897,
      "sup_loss": 0.6298462511436631
    },
    {
      "train_step": 9898,
      "sup_loss": 0.6438527128622883
    },
    {
      "train_step": 9899,
      "sup_loss": 0.4856264844351656
    },
    {
      "train_step": 9900,
      "sup_loss": 0.5461482493585437
    },
    {
      "train_step": 9901,
      "sup_loss": 0.5563409872168895
    },
    {
      "train_step": 9902,
      "sup_loss": 0.5711692479749216
    },
    {
      "train_step": 9903,
      "sup_loss": 0.6151536753240471
    },
    {
      "train_step": 9904,
      "sup_loss": 0.6479822036729562
    },
    {
      "train_step": 9905,
      "sup_loss": 0.5501936313949625
    },
    {
      "train_step": 9906,
      "sup_loss": 0.4628002511418493
    },
    {
      "train_step": 9907,
      "sup_loss": 0.592615019924905
    },
    {
      "train_step": 9908,
      "sup_loss": 0.4708191781544495
    },
    {
      "train_step": 9909,
      "sup_loss": 0.39724384861715484
    },
    {
      "train_step": 9910,
      "sup_loss": 0.2208159064523917
    },
    {
      "train_step": 9911,
      "sup_loss": 0.4803021214426183
    },
    {
      "train_step": 9912,
      "sup_loss": 0.5851976696459888
    },
    {
      "train_step": 9913,
      "sup_loss": 0.8565261044753639
    },
    {
      "train_step": 9914,
      "sup_loss": 0.7176820734147022
    },
    {
      "train_step": 9915,
      "sup_loss": 0.46799684986082846
    },
    {
      "train_step": 9916,
      "sup_loss": 0.4136088737254108
    },
    {
      "train_step": 9917,
      "sup_loss": 0.7215936127389707
    },
    {
      "train_step": 9918,
      "sup_loss": 0.4179495303161547
    },
    {
      "train_step": 9919,
      "sup_loss": 0.72151518181051
    },
    {
      "train_step": 9920,
      "sup_loss": 0.502028943865615
    },
    {
      "train_step": 9921,
      "sup_loss": 0.896178009707296
    },
    {
      "train_step": 9922,
      "sup_loss": 0.6656520283088616
    },
    {
      "train_step": 9923,
      "sup_loss": 0.4930171547492053
    },
    {
      "train_step": 9924,
      "sup_loss": 0.6939627026849655
    },
    {
      "train_step": 9925,
      "sup_loss": 0.47208752490254674
    },
    {
      "train_step": 9926,
      "sup_loss": 0.40973959021055534
    },
    {
      "train_step": 9927,
      "sup_loss": 0.6499164378111478
    },
    {
      "train_step": 9928,
      "sup_loss": 0.6153091695395257
    },
    {
      "train_step": 9929,
      "sup_loss": 0.655458463342722
    },
    {
      "train_step": 9930,
      "sup_loss": 0.5976492543869292
    },
    {
      "train_step": 9931,
      "sup_loss": 0.6016499250828706
    },
    {
      "train_step": 9932,
      "sup_loss": 0.36468878991901915
    },
    {
      "train_step": 9933,
      "sup_loss": 0.5375490196500778
    },
    {
      "train_step": 9934,
      "sup_loss": 0.5720625796136244
    },
    {
      "train_step": 9935,
      "sup_loss": 0.5852236467673718
    },
    {
      "train_step": 9936,
      "sup_loss": 0.41591306590500404
    },
    {
      "train_step": 9937,
      "sup_loss": 0.532663509540993
    },
    {
      "train_step": 9938,
      "sup_loss": 0.5997835885358528
    },
    {
      "train_step": 9939,
      "sup_loss": 0.7174893971696423
    },
    {
      "train_step": 9940,
      "sup_loss": 0.5748482398529099
    },
    {
      "train_step": 9941,
      "sup_loss": 0.4976068656794813
    },
    {
      "train_step": 9942,
      "sup_loss": 0.6421489624344161
    },
    {
      "train_step": 9943,
      "sup_loss": 0.40017310268631395
    },
    {
      "train_step": 9944,
      "sup_loss": 0.5385549783411913
    },
    {
      "train_step": 9945,
      "sup_loss": 0.3802876471534366
    },
    {
      "train_step": 9946,
      "sup_loss": 0.6492325399621292
    },
    {
      "train_step": 9947,
      "sup_loss": 0.49836409438185725
    },
    {
      "train_step": 9948,
      "sup_loss": 0.4610505827374667
    },
    {
      "train_step": 9949,
      "sup_loss": 0.4552951431920271
    },
    {
      "train_step": 9950,
      "sup_loss": 0.8500854362489205
    },
    {
      "train_step": 9951,
      "sup_loss": 0.5994088990353629
    },
    {
      "train_step": 9952,
      "sup_loss": 0.6450334829797048
    },
    {
      "train_step": 9953,
      "sup_loss": 0.6706338837023532
    },
    {
      "train_step": 9954,
      "sup_loss": 0.6375058300971903
    },
    {
      "train_step": 9955,
      "sup_loss": 0.6308810874587627
    },
    {
      "train_step": 9956,
      "sup_loss": 0.670157379536159
    },
    {
      "train_step": 9957,
      "sup_loss": 0.4032333436418881
    },
    {
      "train_step": 9958,
      "sup_loss": 0.4693870730140358
    },
    {
      "train_step": 9959,
      "sup_loss": 0.42925412097735044
    },
    {
      "train_step": 9960,
      "sup_loss": 0.5916553215644287
    },
    {
      "train_step": 9961,
      "sup_loss": 0.562131342247948
    },
    {
      "train_step": 9962,
      "sup_loss": 0.8538973824105532
    },
    {
      "train_step": 9963,
      "sup_loss": 0.7507131400395607
    },
    {
      "train_step": 9964,
      "sup_loss": 0.38548821229545877
    },
    {
      "train_step": 9965,
      "sup_loss": 0.5636237294544043
    },
    {
      "train_step": 9966,
      "sup_loss": 0.5403332224005061
    },
    {
      "train_step": 9967,
      "sup_loss": 0.7050194314605209
    },
    {
      "train_step": 9968,
      "sup_loss": 0.3601464514443089
    },
    {
      "train_step": 9969,
      "sup_loss": 0.5671343645402848
    },
    {
      "train_step": 9970,
      "sup_loss": 0.509085953687162
    },
    {
      "train_step": 9971,
      "sup_loss": 0.6836863502258205
    },
    {
      "train_step": 9972,
      "sup_loss": 0.4460796871319767
    },
    {
      "train_step": 9973,
      "sup_loss": 0.48404659416226725
    },
    {
      "train_step": 9974,
      "sup_loss": 0.37345827247472113
    },
    {
      "train_step": 9975,
      "sup_loss": 0.755534590038511
    },
    {
      "train_step": 9976,
      "sup_loss": 0.637867887059635
    },
    {
      "train_step": 9977,
      "sup_loss": 0.5813068378523616
    },
    {
      "train_step": 9978,
      "sup_loss": 0.5891866537277841
    },
    {
      "train_step": 9979,
      "sup_loss": 0.5983562798875859
    },
    {
      "train_step": 9980,
      "sup_loss": 0.7124695427454346
    },
    {
      "train_step": 9981,
      "sup_loss": 0.596790785330947
    },
    {
      "train_step": 9982,
      "sup_loss": 0.5275991387796316
    },
    {
      "train_step": 9983,
      "sup_loss": 0.5848029862414774
    },
    {
      "train_step": 9984,
      "sup_loss": 0.5096456648998587
    },
    {
      "train_step": 9985,
      "sup_loss": 0.7052303171189532
    },
    {
      "train_step": 9986,
      "sup_loss": 0.524835205775318
    },
    {
      "train_step": 9987,
      "sup_loss": 0.6904590661227823
    },
    {
      "train_step": 9988,
      "sup_loss": 0.6364948090541879
    },
    {
      "train_step": 9989,
      "sup_loss": 0.5736498516139721
    },
    {
      "train_step": 9990,
      "sup_loss": 0.6291315682109904
    },
    {
      "train_step": 9991,
      "sup_loss": 0.8457207487390196
    },
    {
      "train_step": 9992,
      "sup_loss": 0.44074778229593675
    },
    {
      "train_step": 9993,
      "sup_loss": 0.5185940773027593
    },
    {
      "train_step": 9994,
      "sup_loss": 0.4399952105027313
    },
    {
      "train_step": 9995,
      "sup_loss": 0.520846402661939
    },
    {
      "train_step": 9996,
      "sup_loss": 0.7639733189550135
    },
    {
      "train_step": 9997,
      "sup_loss": 0.6896359689517666
    },
    {
      "train_step": 9998,
      "sup_loss": 0.9420635419277491
    },
    {
      "train_step": 9999,
      "sup_loss": 0.7556307179583747
    },
    {
      "train_step": 10000,
      "sup_loss": 0.6106767335039616
    }
  ],
  "floss_episodes": [
    {
      "episode": 0,
      "train_step": 0,
      "steps": [
        {
          "step": 0,
          "loss": 0.0008834931650198996,
          "floss_loss": 0.0008834907785058022,
          "kl_loss": 2.395612597183572e-10,
          "lyap1_mean": 0.008908528834581375,
          "lyap1_max": 0.05537918210029602,
          "lyap_mean": 0.0023297073785215616,
          "volume_mean": 0.0023297073785215616,
          "volume_max": 0.03715287894010544,
          "frac_active": 0.5625
        },
        {
          "step": 1,
          "loss": 0.0005437883082777262,
          "floss_loss": 0.0005437883082777262,
          "kl_loss": -1.0664339613428246e-20,
          "lyap1_mean": 0.025454748421907425,
          "lyap1_max": 0.05097999423742294,
          "lyap_mean": 0.01733190193772316,
          "volume_mean": 0.01733190193772316,
          "volume_max": 0.03217626363039017,
          "frac_active": 0.875
        },
        {
          "step": 2,
          "loss": 48.47990417480469,
          "floss_loss": 0.0006907657370902598,
          "kl_loss": 4.847921371459961,
          "lyap1_mean": -0.00677112303674221,
          "lyap1_max": 0.029939183965325356,
          "lyap_mean": -0.00924393068999052,
          "volume_mean": -0.00924393068999052,
          "volume_max": 0.026449080556631088,
          "frac_active": 0.25
        },
        {
          "step": 3,
          "loss": 12.63366985321045,
          "floss_loss": 0.0009103554766625166,
          "kl_loss": 1.2632758617401123,
          "lyap1_mean": 0.03862243518233299,
          "lyap1_max": 0.058936819434165955,
          "lyap_mean": 0.023545771837234497,
          "volume_mean": 0.023545771837234497,
          "volume_max": 0.04011932760477066,
          "frac_active": 0.875
        },
        {
          "step": 4,
          "loss": 36.69914245605469,
          "floss_loss": 8.462732512271032e-05,
          "kl_loss": 3.669905662536621,
          "lyap1_mean": 0.0015251198783516884,
          "lyap1_max": 0.01546127162873745,
          "lyap_mean": -0.0025836899876594543,
          "volume_mean": -0.0025836899876594543,
          "volume_max": 0.004392252769321203,
          "frac_active": 0.3125
        },
        {
          "step": 5,
          "loss": 41.7454833984375,
          "floss_loss": 0.0001538900105515495,
          "kl_loss": 4.174532890319824,
          "lyap1_mean": 0.01285372581332922,
          "lyap1_max": 0.024069635197520256,
          "lyap_mean": 0.0070623839274048805,
          "volume_mean": 0.0070623839274048805,
          "volume_max": 0.019012082368135452,
          "frac_active": 0.8125
        },
        {
          "step": 6,
          "loss": 45.32136154174805,
          "floss_loss": 0.0005749655538238585,
          "kl_loss": 4.532078742980957,
          "lyap1_mean": 0.02227272093296051,
          "lyap1_max": 0.04010874778032303,
          "lyap_mean": 0.016352791339159012,
          "volume_mean": 0.016352791339159012,
          "volume_max": 0.03832796961069107,
          "frac_active": 0.875
        },
        {
          "step": 7,
          "loss": 36.91105270385742,
          "floss_loss": 0.0003563040809240192,
          "kl_loss": 3.6910698413848877,
          "lyap1_mean": 0.0065550366416573524,
          "lyap1_max": 0.03697400540113449,
          "lyap_mean": 0.005472937133163214,
          "volume_mean": 0.005472937133163214,
          "volume_max": 0.035762060433626175,
          "frac_active": 0.3125
        },
        {
          "step": 8,
          "loss": 23.307538986206055,
          "floss_loss": 0.00014743482461199164,
          "kl_loss": 2.3307392597198486,
          "lyap1_mean": 0.01440929900854826,
          "lyap1_max": 0.03144616633653641,
          "lyap_mean": 0.005620843730866909,
          "volume_mean": 0.005620843730866909,
          "volume_max": 0.014997245743870735,
          "frac_active": 0.75
        },
        {
          "step": 9,
          "loss": 48.39494705200195,
          "floss_loss": 0.0002914927317760885,
          "kl_loss": 4.839465618133545,
          "lyap1_mean": 0.0056801531463861465,
          "lyap1_max": 0.03919310122728348,
          "lyap_mean": 0.0019881827756762505,
          "volume_mean": 0.0019881827756762505,
          "volume_max": 0.02870098687708378,
          "frac_active": 0.3125
        },
        {
          "step": 10,
          "loss": 17.454030990600586,
          "floss_loss": 2.804514952003956e-05,
          "kl_loss": 1.7454001903533936,
          "lyap1_mean": 0.001485426677390933,
          "lyap1_max": 0.0050211441703140736,
          "lyap_mean": -0.0022049471735954285,
          "volume_mean": -0.0022049471735954285,
          "volume_max": 0.0013362378813326359,
          "frac_active": 0.25
        },
        {
          "step": 11,
          "loss": 14.026840209960938,
          "floss_loss": 3.845579703920521e-05,
          "kl_loss": 1.4026801586151123,
          "lyap1_mean": -0.004541628062725067,
          "lyap1_max": 0.00382432178594172,
          "lyap_mean": -0.0031566305551677942,
          "volume_mean": -0.0031566305551677942,
          "volume_max": 0.002625253750011325,
          "frac_active": 0.25
        },
        {
          "step": 12,
          "loss": 21.333127975463867,
          "floss_loss": 0.00013472353748511523,
          "kl_loss": 2.1332993507385254,
          "lyap1_mean": 0.007292181719094515,
          "lyap1_max": 0.027924882248044014,
          "lyap_mean": 0.003050178289413452,
          "volume_mean": 0.003050178289413452,
          "volume_max": 0.019201770424842834,
          "frac_active": 0.5
        },
        {
          "step": 13,
          "loss": 15.557368278503418,
          "floss_loss": 0.0002221540780737996,
          "kl_loss": 1.5557146072387695,
          "lyap1_mean": 0.010208778083324432,
          "lyap1_max": 0.016121486201882362,
          "lyap_mean": 0.0078115020878612995,
          "volume_mean": 0.0078115020878612995,
          "volume_max": 0.025799348950386047,
          "frac_active": 0.6875
        },
        {
          "step": 14,
          "loss": 28.046058654785156,
          "floss_loss": 0.00017866649432107806,
          "kl_loss": 2.8045878410339355,
          "lyap1_mean": 0.011626102030277252,
          "lyap1_max": 0.02583501860499382,
          "lyap_mean": 0.0065911198034882545,
          "volume_mean": 0.0065911198034882545,
          "volume_max": 0.017254356294870377,
          "frac_active": 0.5
        },
        {
          "step": 15,
          "loss": 22.82073211669922,
          "floss_loss": 0.0003385572927072644,
          "kl_loss": 2.282039165496826,
          "lyap1_mean": 0.015888862311840057,
          "lyap1_max": 0.034447334706783295,
          "lyap_mean": 0.014448706060647964,
          "volume_mean": 0.014448706060647964,
          "volume_max": 0.024506254121661186,
          "frac_active": 1.0
        },
        {
          "step": 16,
          "loss": 19.776336669921875,
          "floss_loss": 5.7998237025458366e-05,
          "kl_loss": 1.9776279926300049,
          "lyap1_mean": 0.0073298243805766106,
          "lyap1_max": 0.016479669138789177,
          "lyap_mean": 0.005497301463037729,
          "volume_mean": 0.005497301463037729,
          "volume_max": 0.012968097813427448,
          "frac_active": 0.8125
        },
        {
          "step": 17,
          "loss": 73.0819320678711,
          "floss_loss": 0.0006014661630615592,
          "kl_loss": 7.308132648468018,
          "lyap1_mean": 0.029645180329680443,
          "lyap1_max": 0.05365302041172981,
          "lyap_mean": 0.02117857336997986,
          "volume_mean": 0.02117857336997986,
          "volume_max": 0.034581974148750305,
          "frac_active": 1.0
        },
        {
          "step": 18,
          "loss": 30.95783042907715,
          "floss_loss": 0.00014809668937232345,
          "kl_loss": 3.0957682132720947,
          "lyap1_mean": 0.009530870243906975,
          "lyap1_max": 0.012272289022803307,
          "lyap_mean": 0.009462134912610054,
          "volume_mean": 0.009462134912610054,
          "volume_max": 0.018979767337441444,
          "frac_active": 0.9375
        },
        {
          "step": 19,
          "loss": 11.806107521057129,
          "floss_loss": 0.00023386391694657505,
          "kl_loss": 1.1805874109268188,
          "lyap1_mean": 0.016420438885688782,
          "lyap1_max": 0.02855597250163555,
          "lyap_mean": 0.013652510941028595,
          "volume_mean": 0.013652510941028595,
          "volume_max": 0.023160964250564575,
          "frac_active": 1.0
        },
        {
          "step": 20,
          "loss": 26.68339729309082,
          "floss_loss": 0.00020127734751440585,
          "kl_loss": 2.6683194637298584,
          "lyap1_mean": 0.012436607852578163,
          "lyap1_max": 0.03543863072991371,
          "lyap_mean": 0.01101602241396904,
          "volume_mean": 0.01101602241396904,
          "volume_max": 0.02410876378417015,
          "frac_active": 1.0
        },
        {
          "step": 21,
          "loss": 25.35411262512207,
          "floss_loss": 0.00028364011086523533,
          "kl_loss": 2.5353827476501465,
          "lyap1_mean": 0.012345340102910995,
          "lyap1_max": 0.02388525940477848,
          "lyap_mean": 0.011917940340936184,
          "volume_mean": 0.011917940340936184,
          "volume_max": 0.023842349648475647,
          "frac_active": 0.75
        },
        {
          "step": 22,
          "loss": 20.476131439208984,
          "floss_loss": 2.1082163584651425e-05,
          "kl_loss": 2.0476109981536865,
          "lyap1_mean": 0.0037156837061047554,
          "lyap1_max": 0.008434640243649483,
          "lyap_mean": 0.00338406627997756,
          "volume_mean": 0.00338406627997756,
          "volume_max": 0.005380659364163876,
          "frac_active": 0.875
        },
        {
          "step": 23,
          "loss": 11.614727020263672,
          "floss_loss": 1.2731325114145875e-05,
          "kl_loss": 1.1614714860916138,
          "lyap1_mean": 0.0051222797483205795,
          "lyap1_max": 0.008285995572805405,
          "lyap_mean": 0.0016059760237112641,
          "volume_mean": 0.0016059760237112641,
          "volume_max": 0.003590562380850315,
          "frac_active": 0.5625
        },
        {
          "step": 24,
          "loss": 10.405295372009277,
          "floss_loss": 2.7236594178248197e-05,
          "kl_loss": 1.0405267477035522,
          "lyap1_mean": -0.0018970688106492162,
          "lyap1_max": 0.005201906897127628,
          "lyap_mean": -0.0021343883126974106,
          "volume_mean": -0.0021343883126974106,
          "volume_max": 0.004026214592158794,
          "frac_active": 0.3125
        },
        {
          "step": 25,
          "loss": 26.422801971435547,
          "floss_loss": 3.5503064282238483e-05,
          "kl_loss": 2.6422765254974365,
          "lyap1_mean": 0.0026590542402118444,
          "lyap1_max": 0.009327986277639866,
          "lyap_mean": 0.00022601312957704067,
          "volume_mean": 0.00022601312957704067,
          "volume_max": 0.006930793169885874,
          "frac_active": 0.625
        },
        {
          "step": 26,
          "loss": 9.279953002929688,
          "floss_loss": 7.213581557152793e-05,
          "kl_loss": 0.9279880523681641,
          "lyap1_mean": 0.007693742401897907,
          "lyap1_max": 0.022599821910262108,
          "lyap_mean": 0.0059457020834088326,
          "volume_mean": 0.0059457020834088326,
          "volume_max": 0.01012200303375721,
          "frac_active": 0.875
        },
        {
          "step": 27,
          "loss": 13.969918251037598,
          "floss_loss": 3.17144877044484e-05,
          "kl_loss": 1.3969886302947998,
          "lyap1_mean": 0.007585657760500908,
          "lyap1_max": 0.017142407596111298,
          "lyap_mean": 0.002073966432362795,
          "volume_mean": 0.002073966432362795,
          "volume_max": 0.0034764939919114113,
          "frac_active": 0.625
        },
        {
          "step": 28,
          "loss": 9.840996742248535,
          "floss_loss": 3.897798160323873e-05,
          "kl_loss": 0.9840958118438721,
          "lyap1_mean": 0.006142022088170052,
          "lyap1_max": 0.00751948356628418,
          "lyap_mean": 0.003907849080860615,
          "volume_mean": 0.003907849080860615,
          "volume_max": 0.006329408846795559,
          "frac_active": 0.8125
        },
        {
          "step": 29,
          "loss": 10.288400650024414,
          "floss_loss": 1.9653885829029605e-05,
          "kl_loss": 1.028838038444519,
          "lyap1_mean": 0.0005338359042070806,
          "lyap1_max": 0.00238565425388515,
          "lyap_mean": -2.9312301194295287e-05,
          "volume_mean": -2.9312301194295287e-05,
          "volume_max": 0.0037022577598690987,
          "frac_active": 0.625
        },
        {
          "step": 30,
          "loss": 13.671530723571777,
          "floss_loss": 5.119681736687198e-05,
          "kl_loss": 1.3671479225158691,
          "lyap1_mean": 0.00668254354968667,
          "lyap1_max": 0.014912196435034275,
          "lyap_mean": 0.0024577081203460693,
          "volume_mean": 0.0024577081203460693,
          "volume_max": 0.005385218653827906,
          "frac_active": 0.5625
        },
        {
          "step": 31,
          "loss": 15.837989807128906,
          "floss_loss": 2.813652463373728e-05,
          "kl_loss": 1.5837961435317993,
          "lyap1_mean": 0.004991140216588974,
          "lyap1_max": 0.011401329189538956,
          "lyap_mean": 0.003074285574257374,
          "volume_mean": 0.003074285574257374,
          "volume_max": 0.006720458157360554,
          "frac_active": 0.6875
        },
        {
          "step": 32,
          "loss": 11.940606117248535,
          "floss_loss": 3.1289426260627806e-05,
          "kl_loss": 1.1940574645996094,
          "lyap1_mean": -0.00021214148728176951,
          "lyap1_max": 0.009039456956088543,
          "lyap_mean": -0.0009529589442536235,
          "volume_mean": -0.0009529589442536235,
          "volume_max": 0.0028007724322378635,
          "frac_active": 0.4375
        },
        {
          "step": 33,
          "loss": 20.83019256591797,
          "floss_loss": 1.93382402358111e-05,
          "kl_loss": 2.083017349243164,
          "lyap1_mean": 0.004024961031973362,
          "lyap1_max": 0.010093280114233494,
          "lyap_mean": -9.466067422181368e-05,
          "volume_mean": -9.466067422181368e-05,
          "volume_max": 0.0020027831196784973,
          "frac_active": 0.1875
        },
        {
          "step": 34,
          "loss": 13.809103012084961,
          "floss_loss": 4.2026862502098083e-05,
          "kl_loss": 1.380906105041504,
          "lyap1_mean": 0.007553845643997192,
          "lyap1_max": 0.014937730506062508,
          "lyap_mean": 0.003026338992640376,
          "volume_mean": 0.003026338992640376,
          "volume_max": 0.006592629477381706,
          "frac_active": 0.6875
        },
        {
          "step": 35,
          "loss": 11.550023078918457,
          "floss_loss": 2.0667883291025646e-05,
          "kl_loss": 1.1550002098083496,
          "lyap1_mean": 0.003822278929874301,
          "lyap1_max": 0.006193685345351696,
          "lyap_mean": 0.002657113131135702,
          "volume_mean": 0.002657113131135702,
          "volume_max": 0.0057522887364029884,
          "frac_active": 0.75
        },
        {
          "step": 36,
          "loss": 10.453292846679688,
          "floss_loss": 7.738728891126812e-05,
          "kl_loss": 1.0453215837478638,
          "lyap1_mean": 0.008002945221960545,
          "lyap1_max": 0.0123771782964468,
          "lyap_mean": 0.006559967994689941,
          "volume_mean": 0.006559967994689941,
          "volume_max": 0.011758093722164631,
          "frac_active": 0.8125
        },
        {
          "step": 37,
          "loss": 7.966409206390381,
          "floss_loss": 0.00024441879941150546,
          "kl_loss": 0.7966164350509644,
          "lyap1_mean": 0.01796749234199524,
          "lyap1_max": 0.0513046495616436,
          "lyap_mean": 0.009123722091317177,
          "volume_mean": 0.009123722091317177,
          "volume_max": 0.018763400614261627,
          "frac_active": 0.8125
        },
        {
          "step": 38,
          "loss": 9.426351547241211,
          "floss_loss": 8.264341886388138e-05,
          "kl_loss": 0.9426268935203552,
          "lyap1_mean": 0.007178575731813908,
          "lyap1_max": 0.013576292432844639,
          "lyap_mean": 0.006307824049144983,
          "volume_mean": 0.006307824049144983,
          "volume_max": 0.013939794152975082,
          "frac_active": 0.875
        },
        {
          "step": 39,
          "loss": 8.821261405944824,
          "floss_loss": 0.00011366333637852222,
          "kl_loss": 0.8821147680282593,
          "lyap1_mean": 0.007497448939830065,
          "lyap1_max": 0.00988492090255022,
          "lyap_mean": 0.00820883084088564,
          "volume_mean": 0.00820883084088564,
          "volume_max": 0.012607550248503685,
          "frac_active": 0.9375
        },
        {
          "step": 40,
          "loss": 8.175566673278809,
          "floss_loss": 0.0001637009554542601,
          "kl_loss": 0.8175402879714966,
          "lyap1_mean": 0.013897324912250042,
          "lyap1_max": 0.02547696977853775,
          "lyap_mean": 0.009532525204122066,
          "volume_mean": 0.009532525204122066,
          "volume_max": 0.019047122448682785,
          "frac_active": 1.0
        },
        {
          "step": 41,
          "loss": 10.558281898498535,
          "floss_loss": 0.00019979610806331038,
          "kl_loss": 1.055808186531067,
          "lyap1_mean": 0.020458634942770004,
          "lyap1_max": 0.03574757277965546,
          "lyap_mean": 0.009924707934260368,
          "volume_mean": 0.009924707934260368,
          "volume_max": 0.017402246594429016,
          "frac_active": 0.9375
        },
        {
          "step": 42,
          "loss": 6.545205593109131,
          "floss_loss": 0.00015203695511445403,
          "kl_loss": 0.6545053720474243,
          "lyap1_mean": 0.009527547284960747,
          "lyap1_max": 0.015624008141458035,
          "lyap_mean": 0.010366814211010933,
          "volume_mean": 0.010366814211010933,
          "volume_max": 0.015476996079087257,
          "frac_active": 1.0
        },
        {
          "step": 43,
          "loss": 8.112367630004883,
          "floss_loss": 9.023451275425032e-05,
          "kl_loss": 0.8112276792526245,
          "lyap1_mean": 0.008823014795780182,
          "lyap1_max": 0.017065536230802536,
          "lyap_mean": 0.007757123559713364,
          "volume_mean": 0.007757123559713364,
          "volume_max": 0.015893690288066864,
          "frac_active": 1.0
        },
        {
          "step": 44,
          "loss": 6.778186798095703,
          "floss_loss": 8.506621816195548e-05,
          "kl_loss": 0.6778101921081543,
          "lyap1_mean": 0.011656437069177628,
          "lyap1_max": 0.0175991989672184,
          "lyap_mean": 0.008256718516349792,
          "volume_mean": 0.008256718516349792,
          "volume_max": 0.011132581159472466,
          "frac_active": 1.0
        },
        {
          "step": 45,
          "loss": 7.227631092071533,
          "floss_loss": 2.702154597500339e-05,
          "kl_loss": 0.7227603793144226,
          "lyap1_mean": 0.0056719789281487465,
          "lyap1_max": 0.008047474548220634,
          "lyap_mean": 0.003892537672072649,
          "volume_mean": 0.003892537672072649,
          "volume_max": 0.008006628602743149,
          "frac_active": 0.8125
        },
        {
          "step": 46,
          "loss": 8.019438743591309,
          "floss_loss": 0.00010335906699765474,
          "kl_loss": 0.8019335865974426,
          "lyap1_mean": 0.009692142717540264,
          "lyap1_max": 0.01967768371105194,
          "lyap_mean": 0.008359501138329506,
          "volume_mean": 0.008359501138329506,
          "volume_max": 0.01624712347984314,
          "frac_active": 1.0
        },
        {
          "step": 47,
          "loss": 9.107064247131348,
          "floss_loss": 3.593409564928152e-05,
          "kl_loss": 0.9107027649879456,
          "lyap1_mean": 0.00624876469373703,
          "lyap1_max": 0.01272211316972971,
          "lyap_mean": 0.004666995257139206,
          "volume_mean": 0.004666995257139206,
          "volume_max": 0.009941361844539642,
          "frac_active": 1.0
        },
        {
          "step": 48,
          "loss": 6.861966609954834,
          "floss_loss": 4.264970630174503e-05,
          "kl_loss": 0.6861923933029175,
          "lyap1_mean": 0.008595552295446396,
          "lyap1_max": 0.014381427317857742,
          "lyap_mean": 0.005209186114370823,
          "volume_mean": 0.005209186114370823,
          "volume_max": 0.008269238285720348,
          "frac_active": 0.875
        },
        {
          "step": 49,
          "loss": 7.141152381896973,
          "floss_loss": 1.4526299310091417e-05,
          "kl_loss": 0.7141138315200806,
          "lyap1_mean": 0.00335716363042593,
          "lyap1_max": 0.007622826844453812,
          "lyap_mean": 0.0019197511719539762,
          "volume_mean": 0.0019197511719539762,
          "volume_max": 0.00548149086534977,
          "frac_active": 0.625
        },
        {
          "step": 50,
          "loss": 8.592986106872559,
          "floss_loss": 8.568327757529914e-06,
          "kl_loss": 0.8592977523803711,
          "lyap1_mean": 0.0032184168230742216,
          "lyap1_max": 0.007546586915850639,
          "lyap_mean": 0.001226147636771202,
          "volume_mean": 0.001226147636771202,
          "volume_max": 0.002962095895782113,
          "frac_active": 0.4375
        },
        {
          "step": 51,
          "loss": 7.397765636444092,
          "floss_loss": 6.218575435923412e-05,
          "kl_loss": 0.7397703528404236,
          "lyap1_mean": 0.008727183565497398,
          "lyap1_max": 0.025699930265545845,
          "lyap_mean": 0.003431531134992838,
          "volume_mean": 0.003431531134992838,
          "volume_max": 0.011709874495863914,
          "frac_active": 0.6875
        },
        {
          "step": 52,
          "loss": 6.4594316482543945,
          "floss_loss": 2.156967639166396e-05,
          "kl_loss": 0.6459410190582275,
          "lyap1_mean": -0.0028970048297196627,
          "lyap1_max": 0.00013754836982116103,
          "lyap_mean": -0.0029524872079491615,
          "volume_mean": -0.0029524872079491615,
          "volume_max": -0.0004442830104380846,
          "frac_active": 0.25
        },
        {
          "step": 53,
          "loss": 4.193436145782471,
          "floss_loss": 1.2975409845239483e-05,
          "kl_loss": 0.4193423390388489,
          "lyap1_mean": -8.031658944673836e-05,
          "lyap1_max": 0.0027091463562101126,
          "lyap_mean": -0.00023947001318447292,
          "volume_mean": -0.00023947001318447292,
          "volume_max": 0.002832037629559636,
          "frac_active": 0.4375
        },
        {
          "step": 54,
          "loss": 5.176523685455322,
          "floss_loss": 1.4811014807492029e-05,
          "kl_loss": 0.5176509022712708,
          "lyap1_mean": 0.0003109418030362576,
          "lyap1_max": 0.006022188812494278,
          "lyap_mean": -0.002220283029600978,
          "volume_mean": -0.002220283029600978,
          "volume_max": 0.0008931170450523496,
          "frac_active": 0.25
        },
        {
          "step": 55,
          "loss": 4.609138011932373,
          "floss_loss": 4.4933105527888983e-05,
          "kl_loss": 0.46090930700302124,
          "lyap1_mean": -0.0002534879604354501,
          "lyap1_max": 0.005919219460338354,
          "lyap_mean": -0.0006125810323283076,
          "volume_mean": -0.0006125810323283076,
          "volume_max": 0.007352783810347319,
          "frac_active": 0.375
        },
        {
          "step": 56,
          "loss": 3.0538930892944336,
          "floss_loss": 1.7853879398899153e-05,
          "kl_loss": 0.3053875267505646,
          "lyap1_mean": 0.0023036329075694084,
          "lyap1_max": 0.004028362222015858,
          "lyap_mean": 0.000668379187118262,
          "volume_mean": 0.000668379187118262,
          "volume_max": 0.004356772173196077,
          "frac_active": 0.5
        },
        {
          "step": 57,
          "loss": 6.3637776374816895,
          "floss_loss": 3.838569318759255e-05,
          "kl_loss": 0.6363738775253296,
          "lyap1_mean": 4.1794730350375175e-05,
          "lyap1_max": 0.005522242281585932,
          "lyap_mean": -0.001134099205955863,
          "volume_mean": -0.001134099205955863,
          "volume_max": 0.006469286046922207,
          "frac_active": 0.4375
        },
        {
          "step": 58,
          "loss": 5.770496368408203,
          "floss_loss": 1.3044940715190023e-05,
          "kl_loss": 0.5770483613014221,
          "lyap1_mean": -0.0004747959319502115,
          "lyap1_max": 0.0031692914199084044,
          "lyap_mean": -0.0010711568174883723,
          "volume_mean": -0.0010711568174883723,
          "volume_max": 0.0012728606816381216,
          "frac_active": 0.375
        },
        {
          "step": 59,
          "loss": 6.059235095977783,
          "floss_loss": 2.9767759770038538e-05,
          "kl_loss": 0.6059205532073975,
          "lyap1_mean": 0.00026016467018052936,
          "lyap1_max": 0.0035180968698114157,
          "lyap_mean": 0.00037563941441476345,
          "volume_mean": 0.00037563941441476345,
          "volume_max": 0.004920287989079952,
          "frac_active": 0.4375
        },
        {
          "step": 60,
          "loss": 3.0710508823394775,
          "floss_loss": 9.326804138254374e-05,
          "kl_loss": 0.3070957660675049,
          "lyap1_mean": 0.007970706559717655,
          "lyap1_max": 0.02636088617146015,
          "lyap_mean": 0.0012055790284648538,
          "volume_mean": 0.0012055790284648538,
          "volume_max": 0.014430164359509945,
          "frac_active": 0.3125
        },
        {
          "step": 61,
          "loss": 2.728327512741089,
          "floss_loss": 3.090799873461947e-05,
          "kl_loss": 0.27282965183258057,
          "lyap1_mean": 0.0008092437637969851,
          "lyap1_max": 0.003599909832701087,
          "lyap_mean": -0.0018452010117471218,
          "volume_mean": -0.0018452010117471218,
          "volume_max": 0.005546596832573414,
          "frac_active": 0.3125
        },
        {
          "step": 62,
          "loss": 3.179337739944458,
          "floss_loss": 2.9281167371664196e-05,
          "kl_loss": 0.31793084740638733,
          "lyap1_mean": -0.002648575697094202,
          "lyap1_max": 0.003166818292811513,
          "lyap_mean": -0.003485799999907613,
          "volume_mean": -0.003485799999907613,
          "volume_max": 0.000741938129067421,
          "frac_active": 0.25
        },
        {
          "step": 63,
          "loss": 3.4704947471618652,
          "floss_loss": 0.00018265888502355665,
          "kl_loss": 0.34703120589256287,
          "lyap1_mean": 0.015149278566241264,
          "lyap1_max": 0.049811478704214096,
          "lyap_mean": 0.0034335164818912745,
          "volume_mean": 0.0034335164818912745,
          "volume_max": 0.01635216362774372,
          "frac_active": 0.5625
        },
        {
          "step": 64,
          "loss": 6.5017409324646,
          "floss_loss": 4.778735456056893e-05,
          "kl_loss": 0.650169312953949,
          "lyap1_mean": -0.0016899893525987864,
          "lyap1_max": 0.01047984603792429,
          "lyap_mean": -0.0036202638875693083,
          "volume_mean": -0.0036202638875693083,
          "volume_max": 0.0030477007385343313,
          "frac_active": 0.25
        },
        {
          "step": 65,
          "loss": 5.716618537902832,
          "floss_loss": 5.877756848349236e-05,
          "kl_loss": 0.5716559886932373,
          "lyap1_mean": 0.005181933753192425,
          "lyap1_max": 0.011178028769791126,
          "lyap_mean": -0.0009072496904991567,
          "volume_mean": -0.0009072496904991567,
          "volume_max": 0.0029224161989986897,
          "frac_active": 0.375
        },
        {
          "step": 66,
          "loss": 6.074275970458984,
          "floss_loss": 0.00030295999022200704,
          "kl_loss": 0.6073973178863525,
          "lyap1_mean": 0.016291014850139618,
          "lyap1_max": 0.04566998779773712,
          "lyap_mean": 0.007581738755106926,
          "volume_mean": 0.007581738755106926,
          "volume_max": 0.030360950157046318,
          "frac_active": 0.5625
        },
        {
          "step": 67,
          "loss": 3.2253193855285645,
          "floss_loss": 6.616850441787392e-05,
          "kl_loss": 0.3225253224372864,
          "lyap1_mean": -0.005290730390697718,
          "lyap1_max": 0.006540573667734861,
          "lyap_mean": -0.0057611120864748955,
          "volume_mean": -0.0057611120864748955,
          "volume_max": -4.855729639530182e-05,
          "frac_active": 0.125
        },
        {
          "step": 68,
          "loss": 4.591973304748535,
          "floss_loss": 5.9116166085004807e-05,
          "kl_loss": 0.4591914415359497,
          "lyap1_mean": 0.0013255961239337921,
          "lyap1_max": 0.013648617081344128,
          "lyap_mean": -0.0013794589322060347,
          "volume_mean": -0.0013794589322060347,
          "volume_max": 0.007197726983577013,
          "frac_active": 0.4375
        },
        {
          "step": 69,
          "loss": 2.116894245147705,
          "floss_loss": 9.054351539816707e-05,
          "kl_loss": 0.2116803526878357,
          "lyap1_mean": 0.0005533723160624504,
          "lyap1_max": 0.006080044899135828,
          "lyap_mean": -0.003028788138180971,
          "volume_mean": -0.003028788138180971,
          "volume_max": 0.0029664975591003895,
          "frac_active": 0.375
        },
        {
          "step": 70,
          "loss": 1.2709022760391235,
          "floss_loss": 8.047982555581257e-05,
          "kl_loss": 0.12708218395709991,
          "lyap1_mean": -0.00851445086300373,
          "lyap1_max": -0.005159728229045868,
          "lyap_mean": -0.008573111146688461,
          "volume_mean": -0.008573111146688461,
          "volume_max": -0.006472461856901646,
          "frac_active": 0.0
        },
        {
          "step": 71,
          "loss": 3.5480129718780518,
          "floss_loss": 0.00012860834249295294,
          "kl_loss": 0.3547884523868561,
          "lyap1_mean": 0.001514551811851561,
          "lyap1_max": 0.023798927664756775,
          "lyap_mean": -0.0017934885108843446,
          "volume_mean": -0.0017934885108843446,
          "volume_max": 0.005287295673042536,
          "frac_active": 0.25
        },
        {
          "step": 72,
          "loss": 1.1274735927581787,
          "floss_loss": 7.873759750509635e-05,
          "kl_loss": 0.11273949593305588,
          "lyap1_mean": 0.004058745689690113,
          "lyap1_max": 0.020738588646054268,
          "lyap_mean": -0.003913328051567078,
          "volume_mean": -0.003913328051567078,
          "volume_max": 0.0004050454590469599,
          "frac_active": 0.1875
        },
        {
          "step": 73,
          "loss": 1.3544015884399414,
          "floss_loss": 0.0003110565594397485,
          "kl_loss": 0.13540905714035034,
          "lyap1_mean": 0.019784247502684593,
          "lyap1_max": 0.0407557338476181,
          "lyap_mean": 0.0028125024400651455,
          "volume_mean": 0.0028125024400651455,
          "volume_max": 0.011255732737481594,
          "frac_active": 0.375
        },
        {
          "step": 74,
          "loss": 8.448148727416992,
          "floss_loss": 0.0003118333697784692,
          "kl_loss": 0.8447837233543396,
          "lyap1_mean": 0.01572171226143837,
          "lyap1_max": 0.05290202423930168,
          "lyap_mean": 0.0012649563141167164,
          "volume_mean": 0.0012649563141167164,
          "volume_max": 0.013589683920145035,
          "frac_active": 0.3125
        },
        {
          "step": 75,
          "loss": 4.0576276779174805,
          "floss_loss": 0.00014190345245879143,
          "kl_loss": 0.40574854612350464,
          "lyap1_mean": 0.0014138896949589252,
          "lyap1_max": 0.028415853157639503,
          "lyap_mean": -0.00582937803119421,
          "volume_mean": -0.00582937803119421,
          "volume_max": 0.004668714012950659,
          "frac_active": 0.0625
        },
        {
          "step": 76,
          "loss": 2.185250997543335,
          "floss_loss": 0.0001152718105004169,
          "kl_loss": 0.2185135781764984,
          "lyap1_mean": 3.2460084185004234e-05,
          "lyap1_max": 0.021313190460205078,
          "lyap_mean": -0.005456503480672836,
          "volume_mean": -0.005456503480672836,
          "volume_max": 0.005647796671837568,
          "frac_active": 0.25
        },
        {
          "step": 77,
          "loss": 1.6434764862060547,
          "floss_loss": 0.0001985628332477063,
          "kl_loss": 0.16432778537273407,
          "lyap1_mean": -0.009988054633140564,
          "lyap1_max": 0.0015632540453225374,
          "lyap_mean": -0.013002711348235607,
          "volume_mean": -0.013002711348235607,
          "volume_max": -0.00676264613866806,
          "frac_active": 0.0625
        },
        {
          "step": 78,
          "loss": 2.7143847942352295,
          "floss_loss": 0.00014422401727642864,
          "kl_loss": 0.2714240550994873,
          "lyap1_mean": -0.0032965161371976137,
          "lyap1_max": 0.0018705890979617834,
          "lyap_mean": -0.010250218212604523,
          "volume_mean": -0.010250218212604523,
          "volume_max": -0.006797193083912134,
          "frac_active": 0.0625
        },
        {
          "step": 79,
          "loss": 3.748526096343994,
          "floss_loss": 0.00012054595572408289,
          "kl_loss": 0.37484055757522583,
          "lyap1_mean": -0.0035375962033867836,
          "lyap1_max": -0.0003021480224560946,
          "lyap_mean": -0.009136868640780449,
          "volume_mean": -0.009136868640780449,
          "volume_max": -0.004730890970677137,
          "frac_active": 0.0625
        },
        {
          "step": 80,
          "loss": 3.5782103538513184,
          "floss_loss": 8.139623969327658e-05,
          "kl_loss": 0.35781291127204895,
          "lyap1_mean": -0.0029977799858897924,
          "lyap1_max": -0.0004776974383275956,
          "lyap_mean": -0.007251285947859287,
          "volume_mean": -0.007251285947859287,
          "volume_max": -0.0015916235279291868,
          "frac_active": 0.0625
        },
        {
          "step": 81,
          "loss": 2.5192315578460693,
          "floss_loss": 0.00015007343608886003,
          "kl_loss": 0.25190815329551697,
          "lyap1_mean": 0.006730047054588795,
          "lyap1_max": 0.023556256666779518,
          "lyap_mean": -0.005645365919917822,
          "volume_mean": -0.005645365919917822,
          "volume_max": -0.0004512346349656582,
          "frac_active": 0.125
        },
        {
          "step": 82,
          "loss": 3.7795779705047607,
          "floss_loss": 0.00014416564954444766,
          "kl_loss": 0.37794336676597595,
          "lyap1_mean": -0.005331074353307486,
          "lyap1_max": 0.0033492459915578365,
          "lyap_mean": -0.009968806989490986,
          "volume_mean": -0.009968806989490986,
          "volume_max": -0.004011011682450771,
          "frac_active": 0.125
        },
        {
          "step": 83,
          "loss": 1.047003984451294,
          "floss_loss": 0.00020933602354489267,
          "kl_loss": 0.10467946529388428,
          "lyap1_mean": 0.011271539144217968,
          "lyap1_max": 0.03198865428566933,
          "lyap_mean": -0.004112533759325743,
          "volume_mean": -0.004112533759325743,
          "volume_max": 0.004370697773993015,
          "frac_active": 0.3125
        },
        {
          "step": 84,
          "loss": 5.4554548263549805,
          "floss_loss": 0.00022676223306916654,
          "kl_loss": 0.5455228090286255,
          "lyap1_mean": -0.00829806737601757,
          "lyap1_max": 0.004565069917589426,
          "lyap_mean": -0.012996233999729156,
          "volume_mean": -0.012996233999729156,
          "volume_max": -0.008565814234316349,
          "frac_active": 0.0625
        },
        {
          "step": 85,
          "loss": 0.3378761112689972,
          "floss_loss": 0.00022637724759988487,
          "kl_loss": 0.033764973282814026,
          "lyap1_mean": -0.011325396597385406,
          "lyap1_max": -0.007674462161958218,
          "lyap_mean": -0.013665023259818554,
          "volume_mean": -0.013665023259818554,
          "volume_max": -0.005578001029789448,
          "frac_active": 0.0
        },
        {
          "step": 86,
          "loss": 1.1604382991790771,
          "floss_loss": 0.00024267325352411717,
          "kl_loss": 0.11601955443620682,
          "lyap1_mean": -0.006364420056343079,
          "lyap1_max": 0.004940081853419542,
          "lyap_mean": -0.007150905206799507,
          "volume_mean": -0.007150905206799507,
          "volume_max": 0.004892322234809399,
          "frac_active": 0.25
        },
        {
          "step": 87,
          "loss": 1.1435595750808716,
          "floss_loss": 0.00020066233992110938,
          "kl_loss": 0.11433589458465576,
          "lyap1_mean": -0.00565579067915678,
          "lyap1_max": 0.020505478605628014,
          "lyap_mean": -0.008611653000116348,
          "volume_mean": -0.008611653000116348,
          "volume_max": 0.0030555319972336292,
          "frac_active": 0.125
        },
        {
          "step": 88,
          "loss": 0.6583454608917236,
          "floss_loss": 0.00015822434215806425,
          "kl_loss": 0.06581871956586838,
          "lyap1_mean": 0.006111658178269863,
          "lyap1_max": 0.02280968241393566,
          "lyap_mean": -0.004693640395998955,
          "volume_mean": -0.004693640395998955,
          "volume_max": 0.010472467169165611,
          "frac_active": 0.3125
        },
        {
          "step": 89,
          "loss": 3.6331939697265625,
          "floss_loss": 0.00024032662622630596,
          "kl_loss": 0.36329537630081177,
          "lyap1_mean": 0.005626411642879248,
          "lyap1_max": 0.030061380937695503,
          "lyap_mean": -0.005643717013299465,
          "volume_mean": -0.005643717013299465,
          "volume_max": 0.0022815223783254623,
          "frac_active": 0.25
        },
        {
          "step": 90,
          "loss": 2.958897829055786,
          "floss_loss": 0.00017860272782854736,
          "kl_loss": 0.29587191343307495,
          "lyap1_mean": -0.0003538280725479126,
          "lyap1_max": 0.010941417887806892,
          "lyap_mean": -0.00526502076536417,
          "volume_mean": -0.00526502076536417,
          "volume_max": 0.006975875236093998,
          "frac_active": 0.375
        },
        {
          "step": 91,
          "loss": 2.3586506843566895,
          "floss_loss": 7.18318551662378e-05,
          "kl_loss": 0.23585788905620575,
          "lyap1_mean": 0.007625012658536434,
          "lyap1_max": 0.014565344899892807,
          "lyap_mean": -0.0027984939515590668,
          "volume_mean": -0.0027984939515590668,
          "volume_max": 0.0001584992278367281,
          "frac_active": 0.375
        },
        {
          "step": 92,
          "loss": 0.8572322130203247,
          "floss_loss": 0.0001638491521589458,
          "kl_loss": 0.08570683747529984,
          "lyap1_mean": -0.0021047419868409634,
          "lyap1_max": 0.009991692379117012,
          "lyap_mean": -0.008550667203962803,
          "volume_mean": -0.008550667203962803,
          "volume_max": -0.002763418946415186,
          "frac_active": 0.25
        },
        {
          "step": 93,
          "loss": 1.0627504587173462,
          "floss_loss": 9.774244244908914e-05,
          "kl_loss": 0.10626526921987534,
          "lyap1_mean": -0.0015934796538203955,
          "lyap1_max": 0.007251949515193701,
          "lyap_mean": -0.0061514489352703094,
          "volume_mean": -0.0061514489352703094,
          "volume_max": 0.00011605690815486014,
          "frac_active": 0.25
        },
        {
          "step": 94,
          "loss": 0.4201700985431671,
          "floss_loss": 7.019293843768537e-05,
          "kl_loss": 0.04200999066233635,
          "lyap1_mean": 0.0037217948120087385,
          "lyap1_max": 0.007152872625738382,
          "lyap_mean": -0.00229018647223711,
          "volume_mean": -0.00229018647223711,
          "volume_max": 0.003716873936355114,
          "frac_active": 0.5
        },
        {
          "step": 95,
          "loss": 2.094393491744995,
          "floss_loss": 0.00012000423157587647,
          "kl_loss": 0.2094273567199707,
          "lyap1_mean": 0.013120876625180244,
          "lyap1_max": 0.023281743749976158,
          "lyap_mean": 0.0006199036724865437,
          "volume_mean": 0.0006199036724865437,
          "volume_max": 0.009512066841125488,
          "frac_active": 0.4375
        },
        {
          "step": 96,
          "loss": 2.116873025894165,
          "floss_loss": 0.0001604474091436714,
          "kl_loss": 0.21167126297950745,
          "lyap1_mean": 0.00763712776824832,
          "lyap1_max": 0.010339374653995037,
          "lyap_mean": 0.0031443892512470484,
          "volume_mean": 0.0031443892512470484,
          "volume_max": 0.008131827227771282,
          "frac_active": 0.625
        },
        {
          "step": 97,
          "loss": 0.06760725378990173,
          "floss_loss": 0.00019105325918644667,
          "kl_loss": 0.006741619668900967,
          "lyap1_mean": 0.006286992225795984,
          "lyap1_max": 0.017004339024424553,
          "lyap_mean": -0.00297286338172853,
          "volume_mean": -0.00297286338172853,
          "volume_max": 0.01085321418941021,
          "frac_active": 0.375
        },
        {
          "step": 98,
          "loss": 1.4627796411514282,
          "floss_loss": 0.00020028326252941042,
          "kl_loss": 0.1462579369544983,
          "lyap1_mean": 0.005292606074362993,
          "lyap1_max": 0.01828385703265667,
          "lyap_mean": -0.00346547388471663,
          "volume_mean": -0.00346547388471663,
          "volume_max": 0.010911939665675163,
          "frac_active": 0.4375
        },
        {
          "step": 99,
          "loss": 1.070796251296997,
          "floss_loss": 0.0002749108534771949,
          "kl_loss": 0.10705213248729706,
          "lyap1_mean": 0.0041526793502271175,
          "lyap1_max": 0.03776339814066887,
          "lyap_mean": -0.005182874388992786,
          "volume_mean": -0.005182874388992786,
          "volume_max": 0.008012955076992512,
          "frac_active": 0.3125
        },
        {
          "step": 100,
          "loss": 0.9274143576622009,
          "floss_loss": 0.0002045349683612585,
          "kl_loss": 0.09272097796201706,
          "lyap1_mean": 0.01393674686551094,
          "lyap1_max": 0.036338504403829575,
          "lyap_mean": 0.0018028132617473602,
          "volume_mean": 0.0018028132617473602,
          "volume_max": 0.00950065441429615,
          "frac_active": 0.5
        },
        {
          "step": 101,
          "loss": 0.2807978391647339,
          "floss_loss": 0.0005702966591343284,
          "kl_loss": 0.028022754937410355,
          "lyap1_mean": 0.025822581723332405,
          "lyap1_max": 0.07575615495443344,
          "lyap_mean": 0.006595582235604525,
          "volume_mean": 0.006595582235604525,
          "volume_max": 0.03535681590437889,
          "frac_active": 0.4375
        },
        {
          "step": 102,
          "loss": 1.4858136177062988,
          "floss_loss": 0.0010232257191091776,
          "kl_loss": 0.14847904443740845,
          "lyap1_mean": 0.027217498049139977,
          "lyap1_max": 0.07834650576114655,
          "lyap_mean": 0.005877180956304073,
          "volume_mean": 0.005877180956304073,
          "volume_max": 0.03135961666703224,
          "frac_active": 0.5
        },
        {
          "step": 103,
          "loss": 1.4700500965118408,
          "floss_loss": 0.0001959005167009309,
          "kl_loss": 0.14698542654514313,
          "lyap1_mean": 0.008115150034427643,
          "lyap1_max": 0.024276049807667732,
          "lyap_mean": -0.0026299834717065096,
          "volume_mean": -0.0026299834717065096,
          "volume_max": 0.004290609620511532,
          "frac_active": 0.3125
        },
        {
          "step": 104,
          "loss": 4.242083549499512,
          "floss_loss": 0.0002747339603956789,
          "kl_loss": 0.42418086528778076,
          "lyap1_mean": 0.0025288567412644625,
          "lyap1_max": 0.031881485134363174,
          "lyap_mean": -0.004935965407639742,
          "volume_mean": -0.004935965407639742,
          "volume_max": 0.015554467216134071,
          "frac_active": 0.375
        },
        {
          "step": 105,
          "loss": 0.0033085732720792294,
          "floss_loss": 0.0003093802370131016,
          "kl_loss": 0.00029991930932737887,
          "lyap1_mean": 0.0022648111917078495,
          "lyap1_max": 0.01948627457022667,
          "lyap_mean": -0.011074485257267952,
          "volume_mean": -0.011074485257267952,
          "volume_max": -0.0009879572317004204,
          "frac_active": 0.125
        },
        {
          "step": 106,
          "loss": 3.6550910472869873,
          "floss_loss": 0.00022950477432459593,
          "kl_loss": 0.36548614501953125,
          "lyap1_mean": -0.00800858810544014,
          "lyap1_max": 0.003648527432233095,
          "lyap_mean": -0.013072540983557701,
          "volume_mean": -0.013072540983557701,
          "volume_max": -0.007247472647577524,
          "frac_active": 0.0625
        },
        {
          "step": 107,
          "loss": 1.4847556352615356,
          "floss_loss": 0.00031161727383732796,
          "kl_loss": 0.14844439923763275,
          "lyap1_mean": 0.000867471331730485,
          "lyap1_max": 0.017333578318357468,
          "lyap_mean": -0.01340620219707489,
          "volume_mean": -0.01340620219707489,
          "volume_max": -0.008234137669205666,
          "frac_active": 0.1875
        },
        {
          "step": 108,
          "loss": 0.40983688831329346,
          "floss_loss": 0.00033621606417000294,
          "kl_loss": 0.04095006734132767,
          "lyap1_mean": -0.01147374790161848,
          "lyap1_max": 0.0038292265962809324,
          "lyap_mean": -0.014751652255654335,
          "volume_mean": -0.014751652255654335,
          "volume_max": 0.00218050554394722,
          "frac_active": 0.1875
        },
        {
          "step": 109,
          "loss": 0.5484265089035034,
          "floss_loss": 0.0005320428754203022,
          "kl_loss": 0.05478945001959801,
          "lyap1_mean": -0.010154353454709053,
          "lyap1_max": 0.008063085377216339,
          "lyap_mean": -0.02133135311305523,
          "volume_mean": -0.02133135311305523,
          "volume_max": -0.017419110983610153,
          "frac_active": 0.0625
        },
        {
          "step": 110,
          "loss": 2.2150943279266357,
          "floss_loss": 0.00041867000982165337,
          "kl_loss": 0.22146756947040558,
          "lyap1_mean": -0.01117838267236948,
          "lyap1_max": 0.02344372868537903,
          "lyap_mean": -0.014181626960635185,
          "volume_mean": -0.014181626960635185,
          "volume_max": 0.00853494182229042,
          "frac_active": 0.1875
        },
        {
          "step": 111,
          "loss": 0.34524133801460266,
          "floss_loss": 0.0005175912519916892,
          "kl_loss": 0.034472376108169556,
          "lyap1_mean": -0.020591648295521736,
          "lyap1_max": -0.012977790087461472,
          "lyap_mean": -0.02222517319023609,
          "volume_mean": -0.02222517319023609,
          "volume_max": -0.01567956805229187,
          "frac_active": 0.0
        },
        {
          "step": 112,
          "loss": 1.1910240650177002,
          "floss_loss": 0.0002708513056859374,
          "kl_loss": 0.11907532811164856,
          "lyap1_mean": -0.0054539879783988,
          "lyap1_max": 0.005889564286917448,
          "lyap_mean": -0.014516389928758144,
          "volume_mean": -0.014516389928758144,
          "volume_max": -0.010502870194613934,
          "frac_active": 0.0625
        },
        {
          "step": 113,
          "loss": 0.01158562395721674,
          "floss_loss": 0.00030761410016566515,
          "kl_loss": 0.001127801020629704,
          "lyap1_mean": -0.0024396232329308987,
          "lyap1_max": 0.026449821889400482,
          "lyap_mean": -0.012594971805810928,
          "volume_mean": -0.012594971805810928,
          "volume_max": -0.003496580058708787,
          "frac_active": 0.125
        },
        {
          "step": 114,
          "loss": 2.279907703399658,
          "floss_loss": 0.00041616102680563927,
          "kl_loss": 0.2279491424560547,
          "lyap1_mean": -0.018927399069070816,
          "lyap1_max": -0.00873451866209507,
          "lyap_mean": -0.019095633178949356,
          "volume_mean": -0.019095633178949356,
          "volume_max": -0.009467218071222305,
          "frac_active": 0.0
        },
        {
          "step": 115,
          "loss": 0.26411786675453186,
          "floss_loss": 0.00018604268552735448,
          "kl_loss": 0.026393180713057518,
          "lyap1_mean": -0.005979974754154682,
          "lyap1_max": -0.0008585585164837539,
          "lyap_mean": -0.012043805792927742,
          "volume_mean": -0.012043805792927742,
          "volume_max": -0.00733604934066534,
          "frac_active": 0.0
        },
        {
          "step": 116,
          "loss": 1.6691482067108154,
          "floss_loss": 0.00013581309758592397,
          "kl_loss": 0.16690124571323395,
          "lyap1_mean": -0.004920231178402901,
          "lyap1_max": 0.0015642776852473617,
          "lyap_mean": -0.007132369559258223,
          "volume_mean": -0.007132369559258223,
          "volume_max": 0.003325965255498886,
          "frac_active": 0.3125
        },
        {
          "step": 117,
          "loss": 0.29820749163627625,
          "floss_loss": 0.000153324130224064,
          "kl_loss": 0.029805416241288185,
          "lyap1_mean": 0.012272018939256668,
          "lyap1_max": 0.02971954643726349,
          "lyap_mean": -0.00142203108407557,
          "volume_mean": -0.00142203108407557,
          "volume_max": 0.005222578998655081,
          "frac_active": 0.375
        },
        {
          "step": 118,
          "loss": 3.5688841342926025,
          "floss_loss": 0.00014440233644563705,
          "kl_loss": 0.3568739593029022,
          "lyap1_mean": 0.005235112272202969,
          "lyap1_max": 0.023895544931292534,
          "lyap_mean": -0.004616226069629192,
          "volume_mean": -0.004616226069629192,
          "volume_max": 0.003564100246876478,
          "frac_active": 0.25
        },
        {
          "step": 119,
          "loss": 7.570523738861084,
          "floss_loss": 0.0001860886550275609,
          "kl_loss": 0.7570337653160095,
          "lyap1_mean": 0.01210652943700552,
          "lyap1_max": 0.04422162473201752,
          "lyap_mean": 0.004320289473980665,
          "volume_mean": 0.004320289473980665,
          "volume_max": 0.013792555779218674,
          "frac_active": 0.5625
        },
        {
          "step": 120,
          "loss": 0.5411747097969055,
          "floss_loss": 0.00011969131446676329,
          "kl_loss": 0.054105501621961594,
          "lyap1_mean": 0.01484673097729683,
          "lyap1_max": 0.01923203282058239,
          "lyap_mean": -0.0006749651511199772,
          "volume_mean": -0.0006749651511199772,
          "volume_max": 0.0025111509021371603,
          "frac_active": 0.375
        },
        {
          "step": 121,
          "loss": 0.0037705234717577696,
          "floss_loss": 0.00018636831373441964,
          "kl_loss": 0.00035841550561599433,
          "lyap1_mean": 0.011695001274347305,
          "lyap1_max": 0.028708260506391525,
          "lyap_mean": -0.0015078161377459764,
          "volume_mean": -0.0015078161377459764,
          "volume_max": 0.009289518930017948,
          "frac_active": 0.375
        },
        {
          "step": 122,
          "loss": 3.042957305908203,
          "floss_loss": 0.00015488025383092463,
          "kl_loss": 0.30428022146224976,
          "lyap1_mean": -0.004219934344291687,
          "lyap1_max": 0.012022801674902439,
          "lyap_mean": -0.003747382201254368,
          "volume_mean": -0.003747382201254368,
          "volume_max": 0.005349328741431236,
          "frac_active": 0.375
        },
        {
          "step": 123,
          "loss": 0.8635880351066589,
          "floss_loss": 0.00014412221207749099,
          "kl_loss": 0.0863443911075592,
          "lyap1_mean": 0.008423242717981339,
          "lyap1_max": 0.02141321636736393,
          "lyap_mean": 0.0003034892724826932,
          "volume_mean": 0.0003034892724826932,
          "volume_max": 0.010226847603917122,
          "frac_active": 0.4375
        },
        {
          "step": 124,
          "loss": 0.12107706815004349,
          "floss_loss": 0.00014636365813203156,
          "kl_loss": 0.012093069963157177,
          "lyap1_mean": -0.003939760383218527,
          "lyap1_max": 0.0075214640237390995,
          "lyap_mean": -0.004951176233589649,
          "volume_mean": -0.004951176233589649,
          "volume_max": 0.005887769162654877,
          "frac_active": 0.3125
        },
        {
          "step": 125,
          "loss": 0.29771608114242554,
          "floss_loss": 0.0001703509478829801,
          "kl_loss": 0.029754573479294777,
          "lyap1_mean": -0.0020722183398902416,
          "lyap1_max": 0.014647109434008598,
          "lyap_mean": -0.00932619534432888,
          "volume_mean": -0.00932619534432888,
          "volume_max": 0.0017039324156939983,
          "frac_active": 0.125
        },
        {
          "step": 126,
          "loss": 0.011610272340476513,
          "floss_loss": 0.0001939692592713982,
          "kl_loss": 0.001141630345955491,
          "lyap1_mean": -0.0074827345088124275,
          "lyap1_max": 0.005248044151812792,
          "lyap_mean": -0.010183028876781464,
          "volume_mean": -0.010183028876781464,
          "volume_max": 0.0008976487442851067,
          "frac_active": 0.25
        },
        {
          "step": 127,
          "loss": 0.2868655025959015,
          "floss_loss": 0.0003414214588701725,
          "kl_loss": 0.028652409091591835,
          "lyap1_mean": -0.01323654130101204,
          "lyap1_max": 0.004994675051420927,
          "lyap_mean": -0.01701052114367485,
          "volume_mean": -0.01701052114367485,
          "volume_max": -0.013555610552430153,
          "frac_active": 0.0625
        },
        {
          "step": 128,
          "loss": 0.5137044191360474,
          "floss_loss": 0.00035560314427129924,
          "kl_loss": 0.05133488029241562,
          "lyap1_mean": -0.00749682541936636,
          "lyap1_max": 0.0213702991604805,
          "lyap_mean": -0.015255356207489967,
          "volume_mean": -0.015255356207489967,
          "volume_max": -0.00416016997769475,
          "frac_active": 0.0625
        },
        {
          "step": 129,
          "loss": 0.0024789823219180107,
          "floss_loss": 0.00033996853744611144,
          "kl_loss": 0.00021390136680565774,
          "lyap1_mean": -0.0172407366335392,
          "lyap1_max": -0.010978199541568756,
          "lyap_mean": -0.017650697380304337,
          "volume_mean": -0.017650697380304337,
          "volume_max": -0.0126658845692873,
          "frac_active": 0.0
        },
        {
          "step": 130,
          "loss": 0.3001336455345154,
          "floss_loss": 0.0003513184783514589,
          "kl_loss": 0.029978234320878983,
          "lyap1_mean": -0.012919559143483639,
          "lyap1_max": -0.0034620834048837423,
          "lyap_mean": -0.016516827046871185,
          "volume_mean": -0.016516827046871185,
          "volume_max": -0.008527321740984917,
          "frac_active": 0.0
        },
        {
          "step": 131,
          "loss": 0.12882846593856812,
          "floss_loss": 0.00010275550448568538,
          "kl_loss": 0.012872571125626564,
          "lyap1_mean": 0.0044431742280721664,
          "lyap1_max": 0.012664207257330418,
          "lyap_mean": -0.00016360182780772448,
          "volume_mean": -0.00016360182780772448,
          "volume_max": 0.0071470728144049644,
          "frac_active": 0.5625
        },
        {
          "step": 132,
          "loss": 0.41210252046585083,
          "floss_loss": 0.00020405836403369904,
          "kl_loss": 0.04118984565138817,
          "lyap1_mean": 0.008199668489396572,
          "lyap1_max": 0.021096544340252876,
          "lyap_mean": -0.005618504248559475,
          "volume_mean": -0.005618504248559475,
          "volume_max": 0.0008417111821472645,
          "frac_active": 0.25
        },
        {
          "step": 133,
          "loss": 2.402600049972534,
          "floss_loss": 0.00017357913020532578,
          "kl_loss": 0.24024264514446259,
          "lyap1_mean": -0.0035932071041315794,
          "lyap1_max": 0.013103503733873367,
          "lyap_mean": -0.010542850010097027,
          "volume_mean": -0.010542850010097027,
          "volume_max": -0.005878115072846413,
          "frac_active": 0.0625
        },
        {
          "step": 134,
          "loss": 1.1974879503250122,
          "floss_loss": 0.00019746151519939303,
          "kl_loss": 0.11972905695438385,
          "lyap1_mean": -0.012338291853666306,
          "lyap1_max": -0.006550394464284182,
          "lyap_mean": -0.012708880938589573,
          "volume_mean": -0.012708880938589573,
          "volume_max": -0.009726512245833874,
          "frac_active": 0.0
        },
        {
          "step": 135,
          "loss": 0.8875940442085266,
          "floss_loss": 0.0005476134247146547,
          "kl_loss": 0.08870464563369751,
          "lyap1_mean": -0.01988230086863041,
          "lyap1_max": -0.0046743182465434074,
          "lyap_mean": -0.02223099209368229,
          "volume_mean": -0.02223099209368229,
          "volume_max": -0.013918493874371052,
          "frac_active": 0.0
        },
        {
          "step": 136,
          "loss": 0.5204285383224487,
          "floss_loss": 0.00015063495084177703,
          "kl_loss": 0.05202779173851013,
          "lyap1_mean": 0.001207052730023861,
          "lyap1_max": 0.012030462734401226,
          "lyap_mean": -0.006647659931331873,
          "volume_mean": -0.006647659931331873,
          "volume_max": -0.0013911512214690447,
          "frac_active": 0.25
        },
        {
          "step": 137,
          "loss": 0.004446906037628651,
          "floss_loss": 0.00016622574185021222,
          "kl_loss": 0.0004280680150259286,
          "lyap1_mean": 3.25450673699379e-05,
          "lyap1_max": 0.010582247748970985,
          "lyap_mean": -0.006769417319446802,
          "volume_mean": -0.006769417319446802,
          "volume_max": 0.0005450411699712276,
          "frac_active": 0.25
        },
        {
          "step": 138,
          "loss": 13.176660537719727,
          "floss_loss": 0.0003795830998569727,
          "kl_loss": 1.3176281452178955,
          "lyap1_mean": 0.015555249527096748,
          "lyap1_max": 0.030012222006917,
          "lyap_mean": -0.006229585502296686,
          "volume_mean": -0.006229585502296686,
          "volume_max": 0.005599673371762037,
          "frac_active": 0.4375
        },
        {
          "step": 139,
          "loss": 6.445018291473389,
          "floss_loss": 0.0003374567022547126,
          "kl_loss": 0.6444680690765381,
          "lyap1_mean": -0.016811396926641464,
          "lyap1_max": -0.004240986425429583,
          "lyap_mean": -0.016533177345991135,
          "volume_mean": -0.016533177345991135,
          "volume_max": -0.010995173826813698,
          "frac_active": 0.0
        },
        {
          "step": 140,
          "loss": 0.32387775182724,
          "floss_loss": 0.00031169294379651546,
          "kl_loss": 0.03235660493373871,
          "lyap1_mean": -0.010328255593776703,
          "lyap1_max": 0.004922541789710522,
          "lyap_mean": -0.015411930158734322,
          "volume_mean": -0.015411930158734322,
          "volume_max": -0.005647452548146248,
          "frac_active": 0.125
        },
        {
          "step": 141,
          "loss": 0.6622269749641418,
          "floss_loss": 0.00019842266920022666,
          "kl_loss": 0.06620285660028458,
          "lyap1_mean": -0.003708766307681799,
          "lyap1_max": 0.0021593002602458,
          "lyap_mean": -0.008121533319354057,
          "volume_mean": -0.008121533319354057,
          "volume_max": 0.006547410041093826,
          "frac_active": 0.25
        },
        {
          "step": 142,
          "loss": 4.107730865478516,
          "floss_loss": 0.0003342140116728842,
          "kl_loss": 0.4107396602630615,
          "lyap1_mean": -0.010209205560386181,
          "lyap1_max": 0.005009389948099852,
          "lyap_mean": -0.017001137137413025,
          "volume_mean": -0.017001137137413025,
          "volume_max": -0.015931114554405212,
          "frac_active": 0.0625
        },
        {
          "step": 143,
          "loss": 3.080552816390991,
          "floss_loss": 0.0003630401333793998,
          "kl_loss": 0.3080189824104309,
          "lyap1_mean": -0.007799321319907904,
          "lyap1_max": 0.019000131636857986,
          "lyap_mean": -0.01517602801322937,
          "volume_mean": -0.01517602801322937,
          "volume_max": -0.0011670943349599838,
          "frac_active": 0.0625
        },
        {
          "step": 144,
          "loss": 2.5679993629455566,
          "floss_loss": 0.0004717631090898067,
          "kl_loss": 0.25675275921821594,
          "lyap1_mean": -0.015178201720118523,
          "lyap1_max": 0.000447364232968539,
          "lyap_mean": -0.02069374918937683,
          "volume_mean": -0.02069374918937683,
          "volume_max": -0.013760579749941826,
          "frac_active": 0.0625
        },
        {
          "step": 145,
          "loss": 1.2615574598312378,
          "floss_loss": 0.00028162659145891666,
          "kl_loss": 0.12612758576869965,
          "lyap1_mean": -0.013562606647610664,
          "lyap1_max": -0.0039211115799844265,
          "lyap_mean": -0.015904229134321213,
          "volume_mean": -0.015904229134321213,
          "volume_max": -0.010321835987269878,
          "frac_active": 0.0
        },
        {
          "step": 146,
          "loss": 1.8744739294052124,
          "floss_loss": 0.00040919979801401496,
          "kl_loss": 0.18740646541118622,
          "lyap1_mean": -0.016108445823192596,
          "lyap1_max": -0.008919634856283665,
          "lyap_mean": -0.019614815711975098,
          "volume_mean": -0.019614815711975098,
          "volume_max": -0.014045999385416508,
          "frac_active": 0.0
        },
        {
          "step": 147,
          "loss": 0.882580041885376,
          "floss_loss": 0.0005474984645843506,
          "kl_loss": 0.08820325136184692,
          "lyap1_mean": -0.023141907528042793,
          "lyap1_max": -0.019401226192712784,
          "lyap_mean": -0.023003743961453438,
          "volume_mean": -0.023003743961453438,
          "volume_max": -0.01940836012363434,
          "frac_active": 0.0
        },
        {
          "step": 148,
          "loss": 0.19960767030715942,
          "floss_loss": 0.00031865923665463924,
          "kl_loss": 0.01992890052497387,
          "lyap1_mean": -0.01531585305929184,
          "lyap1_max": -0.011446070857346058,
          "lyap_mean": -0.017376942560076714,
          "volume_mean": -0.017376942560076714,
          "volume_max": -0.015415501780807972,
          "frac_active": 0.0
        },
        {
          "step": 149,
          "loss": 0.13119086623191833,
          "floss_loss": 0.00039858714444562793,
          "kl_loss": 0.013079226948320866,
          "lyap1_mean": -0.018978087231516838,
          "lyap1_max": -0.010975287295877934,
          "lyap_mean": -0.018828464671969414,
          "volume_mean": -0.018828464671969414,
          "volume_max": -0.010197602212429047,
          "frac_active": 0.0
        },
        {
          "step": 150,
          "loss": 0.5173028707504272,
          "floss_loss": 0.00031118019251152873,
          "kl_loss": 0.05169916898012161,
          "lyap1_mean": -0.004688246175646782,
          "lyap1_max": 0.005195344798266888,
          "lyap_mean": -0.01512853428721428,
          "volume_mean": -0.01512853428721428,
          "volume_max": -0.009390516206622124,
          "frac_active": 0.0625
        },
        {
          "step": 151,
          "loss": 3.228424072265625,
          "floss_loss": 0.00030223658541217446,
          "kl_loss": 0.32281216979026794,
          "lyap1_mean": 0.0036084912717342377,
          "lyap1_max": 0.03626297414302826,
          "lyap_mean": -0.009873239323496819,
          "volume_mean": -0.009873239323496819,
          "volume_max": -0.0060338289476931095,
          "frac_active": 0.1875
        },
        {
          "step": 152,
          "loss": 1.07891047000885,
          "floss_loss": 0.00026894608163274825,
          "kl_loss": 0.10786415636539459,
          "lyap1_mean": 0.00872554536908865,
          "lyap1_max": 0.03041379526257515,
          "lyap_mean": -0.0024505252949893475,
          "volume_mean": -0.0024505252949893475,
          "volume_max": 0.012041714042425156,
          "frac_active": 0.25
        },
        {
          "step": 153,
          "loss": 0.0016182379331439734,
          "floss_loss": 0.00023491185856983066,
          "kl_loss": 0.0001383326161885634,
          "lyap1_mean": 0.02364407852292061,
          "lyap1_max": 0.045147594064474106,
          "lyap_mean": 0.002356595126911998,
          "volume_mean": 0.002356595126911998,
          "volume_max": 0.005403350107371807,
          "frac_active": 0.5
        },
        {
          "step": 154,
          "loss": 2.35373854637146,
          "floss_loss": 0.00014048822049517184,
          "kl_loss": 0.2353598028421402,
          "lyap1_mean": 0.0025657431688159704,
          "lyap1_max": 0.02649015560746193,
          "lyap_mean": -0.0015399365220218897,
          "volume_mean": -0.0015399365220218897,
          "volume_max": 0.008201711811125278,
          "frac_active": 0.4375
        },
        {
          "step": 155,
          "loss": 1.442899465560913,
          "floss_loss": 0.00023243835312314332,
          "kl_loss": 0.14426669478416443,
          "lyap1_mean": 0.0036221067421138287,
          "lyap1_max": 0.028119975700974464,
          "lyap_mean": -0.007583036553114653,
          "volume_mean": -0.007583036553114653,
          "volume_max": 0.007205977104604244,
          "frac_active": 0.25
        },
        {
          "step": 156,
          "loss": 9.833359718322754,
          "floss_loss": 0.0001407539239153266,
          "kl_loss": 0.9833218455314636,
          "lyap1_mean": 0.00637501385062933,
          "lyap1_max": 0.02609674446284771,
          "lyap_mean": -0.0027681870851665735,
          "volume_mean": -0.0027681870851665735,
          "volume_max": 0.00408979644998908,
          "frac_active": 0.375
        },
        {
          "step": 157,
          "loss": 7.549330234527588,
          "floss_loss": 0.00015163468196988106,
          "kl_loss": 0.7549178600311279,
          "lyap1_mean": -0.003969340119510889,
          "lyap1_max": 0.0037687765434384346,
          "lyap_mean": -0.009200260043144226,
          "volume_mean": -0.009200260043144226,
          "volume_max": -0.0036238019820302725,
          "frac_active": 0.125
        },
        {
          "step": 158,
          "loss": 0.046543337404727936,
          "floss_loss": 0.00012568451347760856,
          "kl_loss": 0.004641765262931585,
          "lyap1_mean": -0.0062569864094257355,
          "lyap1_max": 0.004165593534708023,
          "lyap_mean": -0.009117850102484226,
          "volume_mean": -0.009117850102484226,
          "volume_max": -0.00675895344465971,
          "frac_active": 0.125
        },
        {
          "step": 159,
          "loss": 0.3353859782218933,
          "floss_loss": 0.00014870418817736208,
          "kl_loss": 0.03352372720837593,
          "lyap1_mean": -0.0012861673021689057,
          "lyap1_max": 0.011147096753120422,
          "lyap_mean": -0.006702962797135115,
          "volume_mean": -0.006702962797135115,
          "volume_max": 0.0014943857677280903,
          "frac_active": 0.25
        },
        {
          "step": 160,
          "loss": 0.515419065952301,
          "floss_loss": 0.00019297286053188145,
          "kl_loss": 0.05152260884642601,
          "lyap1_mean": 0.004880926106125116,
          "lyap1_max": 0.017428677529096603,
          "lyap_mean": -0.008613133803009987,
          "volume_mean": -0.008613133803009987,
          "volume_max": -0.004619181156158447,
          "frac_active": 0.1875
        },
        {
          "step": 161,
          "loss": 3.3495919704437256,
          "floss_loss": 0.00020416584447957575,
          "kl_loss": 0.33493879437446594,
          "lyap1_mean": 0.004086405970156193,
          "lyap1_max": 0.024065066128969193,
          "lyap_mean": -0.005925393663346767,
          "volume_mean": -0.005925393663346767,
          "volume_max": 0.003512723371386528,
          "frac_active": 0.3125
        },
        {
          "step": 162,
          "loss": 0.3385419249534607,
          "floss_loss": 0.0002031704061664641,
          "kl_loss": 0.03383387625217438,
          "lyap1_mean": -0.010571945458650589,
          "lyap1_max": -0.0031684665009379387,
          "lyap_mean": -0.013196993619203568,
          "volume_mean": -0.013196993619203568,
          "volume_max": -0.009347427636384964,
          "frac_active": 0.0625
        },
        {
          "step": 163,
          "loss": 0.1391514539718628,
          "floss_loss": 0.00015373367932625115,
          "kl_loss": 0.013899772427976131,
          "lyap1_mean": 0.0013585062697529793,
          "lyap1_max": 0.02976597473025322,
          "lyap_mean": -0.005311502609401941,
          "volume_mean": -0.005311502609401941,
          "volume_max": 0.008973249234259129,
          "frac_active": 0.25
        },
        {
          "step": 164,
          "loss": 0.12426503747701645,
          "floss_loss": 0.0001960160443559289,
          "kl_loss": 0.012406902387738228,
          "lyap1_mean": 0.0066542671993374825,
          "lyap1_max": 0.015178605914115906,
          "lyap_mean": 6.264413241297007e-05,
          "volume_mean": 6.264413241297007e-05,
          "volume_max": 0.017968596890568733,
          "frac_active": 0.5
        },
        {
          "step": 165,
          "loss": 0.08659705519676208,
          "floss_loss": 0.00037337630055844784,
          "kl_loss": 0.0086223678663373,
          "lyap1_mean": 0.025297529995441437,
          "lyap1_max": 0.0458773598074913,
          "lyap_mean": 0.005896066315472126,
          "volume_mean": 0.005896066315472126,
          "volume_max": 0.015768464654684067,
          "frac_active": 0.5
        },
        {
          "step": 166,
          "loss": 0.16897732019424438,
          "floss_loss": 0.00022861154866404831,
          "kl_loss": 0.016874870285391808,
          "lyap1_mean": 0.014106055721640587,
          "lyap1_max": 0.03009340725839138,
          "lyap_mean": 0.005043558776378632,
          "volume_mean": 0.005043558776378632,
          "volume_max": 0.01677572913467884,
          "frac_active": 0.625
        },
        {
          "step": 167,
          "loss": 5.784017086029053,
          "floss_loss": 0.00025160692166537046,
          "kl_loss": 0.5783765316009521,
          "lyap1_mean": 0.018550580367445946,
          "lyap1_max": 0.0424349270761013,
          "lyap_mean": 0.004645706154406071,
          "volume_mean": 0.004645706154406071,
          "volume_max": 0.01361758355051279,
          "frac_active": 0.5625
        },
        {
          "step": 168,
          "loss": 3.8655214309692383,
          "floss_loss": 0.00034013629192486405,
          "kl_loss": 0.38651812076568604,
          "lyap1_mean": 0.027678195387125015,
          "lyap1_max": 0.03376304730772972,
          "lyap_mean": 0.00483263935893774,
          "volume_mean": 0.00483263935893774,
          "volume_max": 0.014214744791388512,
          "frac_active": 0.5
        },
        {
          "step": 169,
          "loss": 0.004604365210980177,
          "floss_loss": 0.00047685200115665793,
          "kl_loss": 0.0004127513093408197,
          "lyap1_mean": 0.011995522305369377,
          "lyap1_max": 0.054820623248815536,
          "lyap_mean": 0.0034568863920867443,
          "volume_mean": 0.0034568863920867443,
          "volume_max": 0.031976450234651566,
          "frac_active": 0.5
        },
        {
          "step": 170,
          "loss": 12.837262153625488,
          "floss_loss": 0.00011909213208127767,
          "kl_loss": 1.2837142944335938,
          "lyap1_mean": 0.008181797340512276,
          "lyap1_max": 0.014902523718774319,
          "lyap_mean": -0.0038362268824130297,
          "volume_mean": -0.0038362268824130297,
          "volume_max": 0.0027135582640767097,
          "frac_active": 0.375
        },
        {
          "step": 171,
          "loss": 5.286047458648682,
          "floss_loss": 0.0003407223557587713,
          "kl_loss": 0.5285706520080566,
          "lyap1_mean": 0.008787965402007103,
          "lyap1_max": 0.03426750749349594,
          "lyap_mean": -0.009765278548002243,
          "volume_mean": -0.009765278548002243,
          "volume_max": 0.0004315068945288658,
          "frac_active": 0.25
        },
        {
          "step": 172,
          "loss": 2.266061305999756,
          "floss_loss": 0.00023625127505511045,
          "kl_loss": 0.22658249735832214,
          "lyap1_mean": 0.005786673631519079,
          "lyap1_max": 0.030120283365249634,
          "lyap_mean": -0.008157603442668915,
          "volume_mean": -0.008157603442668915,
          "volume_max": -0.0010194825008511543,
          "frac_active": 0.1875
        },
        {
          "step": 173,
          "loss": 2.5038998126983643,
          "floss_loss": 0.0002696329902391881,
          "kl_loss": 0.2503630220890045,
          "lyap1_mean": 0.0025778310373425484,
          "lyap1_max": 0.039457112550735474,
          "lyap_mean": -0.009123175404965878,
          "volume_mean": -0.009123175404965878,
          "volume_max": 0.0016104569658637047,
          "frac_active": 0.0625
        },
        {
          "step": 174,
          "loss": 0.03235761821269989,
          "floss_loss": 0.00014959990221541375,
          "kl_loss": 0.003220801707357168,
          "lyap1_mean": -0.004973637871444225,
          "lyap1_max": -0.002859807573258877,
          "lyap_mean": -0.010294433683156967,
          "volume_mean": -0.010294433683156967,
          "volume_max": -0.007065736688673496,
          "frac_active": 0.0625
        },
        {
          "step": 175,
          "loss": 4.817699432373047,
          "floss_loss": 0.00031222664983943105,
          "kl_loss": 0.48173871636390686,
          "lyap1_mean": -0.014531031250953674,
          "lyap1_max": -0.007781010586768389,
          "lyap_mean": -0.01685686595737934,
          "volume_mean": -0.01685686595737934,
          "volume_max": -0.010086425580084324,
          "frac_active": 0.0
        },
        {
          "step": 176,
          "loss": 0.5532257556915283,
          "floss_loss": 0.00026138158864341676,
          "kl_loss": 0.05529643967747688,
          "lyap1_mean": -0.015854060649871826,
          "lyap1_max": -0.008934194222092628,
          "lyap_mean": -0.015269303694367409,
          "volume_mean": -0.015269303694367409,
          "volume_max": -0.008258752524852753,
          "frac_active": 0.0
        },
        {
          "step": 177,
          "loss": 1.5381077527999878,
          "floss_loss": 0.00026282560429535806,
          "kl_loss": 0.15378448367118835,
          "lyap1_mean": -0.01095912978053093,
          "lyap1_max": 0.001556510222144425,
          "lyap_mean": -0.013891828246414661,
          "volume_mean": -0.013891828246414661,
          "volume_max": -0.004460058640688658,
          "frac_active": 0.125
        },
        {
          "step": 178,
          "loss": 1.4868532419204712,
          "floss_loss": 0.00027899874839931726,
          "kl_loss": 0.14865742623806,
          "lyap1_mean": -0.014523948542773724,
          "lyap1_max": -0.004584993701428175,
          "lyap_mean": -0.01621575653553009,
          "volume_mean": -0.01621575653553009,
          "volume_max": -0.013561349362134933,
          "frac_active": 0.0
        },
        {
          "step": 179,
          "loss": 1.5626991987228394,
          "floss_loss": 0.0004058904305566102,
          "kl_loss": 0.15622933208942413,
          "lyap1_mean": -0.012407796457409859,
          "lyap1_max": 0.0027552195824682713,
          "lyap_mean": -0.019005589187145233,
          "volume_mean": -0.019005589187145233,
          "volume_max": -0.01583676040172577,
          "frac_active": 0.0625
        },
        {
          "step": 180,
          "loss": 2.2214207649230957,
          "floss_loss": 0.00040946752415038645,
          "kl_loss": 0.2221011370420456,
          "lyap1_mean": -0.017557058483362198,
          "lyap1_max": -0.003633569460362196,
          "lyap_mean": -0.01870630867779255,
          "volume_mean": -0.01870630867779255,
          "volume_max": -0.008816726505756378,
          "frac_active": 0.0
        },
        {
          "step": 181,
          "loss": 1.8276844024658203,
          "floss_loss": 0.00019382202299311757,
          "kl_loss": 0.18274906277656555,
          "lyap1_mean": -0.010205725207924843,
          "lyap1_max": -0.0073569416999816895,
          "lyap_mean": -0.012834122404456139,
          "volume_mean": -0.012834122404456139,
          "volume_max": -0.011570537462830544,
          "frac_active": 0.0
        },
        {
          "step": 182,
          "loss": 3.102806329727173,
          "floss_loss": 0.0003991282428614795,
          "kl_loss": 0.3102407157421112,
          "lyap1_mean": -0.014955466613173485,
          "lyap1_max": -0.0013638943200930953,
          "lyap_mean": -0.01877531036734581,
          "volume_mean": -0.01877531036734581,
          "volume_max": -0.016675468534231186,
          "frac_active": 0.0
        },
        {
          "step": 183,
          "loss": 1.5936393737792969,
          "floss_loss": 0.00038092624163255095,
          "kl_loss": 0.15932585299015045,
          "lyap1_mean": -0.01873740926384926,
          "lyap1_max": -0.01078018918633461,
          "lyap_mean": -0.018003638833761215,
          "volume_mean": -0.018003638833761215,
          "volume_max": -0.012419622391462326,
          "frac_active": 0.0625
        },
        {
          "step": 184,
          "loss": 0.853434145450592,
          "floss_loss": 0.0002779299975372851,
          "kl_loss": 0.08531562238931656,
          "lyap1_mean": -0.014564046636223793,
          "lyap1_max": -0.008918439038097858,
          "lyap_mean": -0.016245821490883827,
          "volume_mean": -0.016245821490883827,
          "volume_max": -0.014135093428194523,
          "frac_active": 0.0
        },
        {
          "step": 185,
          "loss": 0.6004468202590942,
          "floss_loss": 0.00010045782255474478,
          "kl_loss": 0.06003463640809059,
          "lyap1_mean": -0.003332427702844143,
          "lyap1_max": 0.007347310893237591,
          "lyap_mean": -0.007331240922212601,
          "volume_mean": -0.007331240922212601,
          "volume_max": -0.0006558748427778482,
          "frac_active": 0.125
        },
        {
          "step": 186,
          "loss": 2.163241147994995,
          "floss_loss": 0.0002466239675413817,
          "kl_loss": 0.21629945933818817,
          "lyap1_mean": 0.01193749625235796,
          "lyap1_max": 0.035919032990932465,
          "lyap_mean": 0.00037882174365222454,
          "volume_mean": 0.00037882174365222454,
          "volume_max": 0.01217359397560358,
          "frac_active": 0.4375
        },
        {
          "step": 187,
          "loss": 3.233724594116211,
          "floss_loss": 0.0003852921654470265,
          "kl_loss": 0.3233339190483093,
          "lyap1_mean": 0.02259962633252144,
          "lyap1_max": 0.043537821620702744,
          "lyap_mean": 0.008030700497329235,
          "volume_mean": 0.008030700497329235,
          "volume_max": 0.0189815703779459,
          "frac_active": 0.6875
        },
        {
          "step": 188,
          "loss": 0.2358497679233551,
          "floss_loss": 0.0004448129911907017,
          "kl_loss": 0.023540494963526726,
          "lyap1_mean": 0.025822091847658157,
          "lyap1_max": 0.043272219598293304,
          "lyap_mean": 0.0092889703810215,
          "volume_mean": 0.0092889703810215,
          "volume_max": 0.022996634244918823,
          "frac_active": 0.6875
        },
        {
          "step": 189,
          "loss": 0.7649846076965332,
          "floss_loss": 0.0005948581965640187,
          "kl_loss": 0.07643897831439972,
          "lyap1_mean": 0.02485910803079605,
          "lyap1_max": 0.05819651857018471,
          "lyap_mean": 0.007051266264170408,
          "volume_mean": 0.007051266264170408,
          "volume_max": 0.025162089616060257,
          "frac_active": 0.625
        },
        {
          "step": 190,
          "loss": 0.014470936730504036,
          "floss_loss": 0.00037691433681175113,
          "kl_loss": 0.0014094022335484624,
          "lyap1_mean": 0.02886098623275757,
          "lyap1_max": 0.04339829459786415,
          "lyap_mean": 0.011562485247850418,
          "volume_mean": 0.011562485247850418,
          "volume_max": 0.019803691655397415,
          "frac_active": 0.75
        },
        {
          "step": 191,
          "loss": 1.1596901416778564,
          "floss_loss": 0.00046549990656785667,
          "kl_loss": 0.1159224584698677,
          "lyap1_mean": 0.026831846684217453,
          "lyap1_max": 0.04321448504924774,
          "lyap_mean": 0.01257295347750187,
          "volume_mean": 0.01257295347750187,
          "volume_max": 0.023867055773735046,
          "frac_active": 0.75
        },
        {
          "step": 192,
          "loss": 2.81767201423645,
          "floss_loss": 0.0012466134503483772,
          "kl_loss": 0.28164252638816833,
          "lyap1_mean": 0.04497796669602394,
          "lyap1_max": 0.1113494262099266,
          "lyap_mean": 0.01592213660478592,
          "volume_mean": 0.01592213660478592,
          "volume_max": 0.04866794869303703,
          "frac_active": 0.5625
        },
        {
          "step": 193,
          "loss": 0.018509943038225174,
          "floss_loss": 0.0005935144727118313,
          "kl_loss": 0.0017916427459567785,
          "lyap1_mean": 0.02845759317278862,
          "lyap1_max": 0.04597104713320732,
          "lyap_mean": 0.016114376485347748,
          "volume_mean": 0.016114376485347748,
          "volume_max": 0.030048632994294167,
          "frac_active": 0.6875
        },
        {
          "step": 194,
          "loss": 1.4913127422332764,
          "floss_loss": 0.0006515720742754638,
          "kl_loss": 0.14906612038612366,
          "lyap1_mean": 0.02492397092282772,
          "lyap1_max": 0.041997749358415604,
          "lyap_mean": 0.017874786630272865,
          "volume_mean": 0.017874786630272865,
          "volume_max": 0.02806529588997364,
          "frac_active": 0.8125
        },
        {
          "step": 195,
          "loss": 0.36324653029441833,
          "floss_loss": 0.0010489586275070906,
          "kl_loss": 0.0362197570502758,
          "lyap1_mean": 0.0407479926943779,
          "lyap1_max": 0.059814486652612686,
          "lyap_mean": 0.027088571339845657,
          "volume_mean": 0.027088571339845657,
          "volume_max": 0.04257243871688843,
          "frac_active": 0.9375
        },
        {
          "step": 196,
          "loss": 0.1972498893737793,
          "floss_loss": 0.0014281298499554396,
          "kl_loss": 0.019582176581025124,
          "lyap1_mean": 0.05611371994018555,
          "lyap1_max": 0.07674943655729294,
          "lyap_mean": 0.030299687758088112,
          "volume_mean": 0.030299687758088112,
          "volume_max": 0.040161311626434326,
          "frac_active": 0.875
        },
        {
          "step": 197,
          "loss": 1.8718363046646118,
          "floss_loss": 0.0008734746370464563,
          "kl_loss": 0.18709628283977509,
          "lyap1_mean": 0.0357004851102829,
          "lyap1_max": 0.09206568449735641,
          "lyap_mean": 0.014393385499715805,
          "volume_mean": 0.014393385499715805,
          "volume_max": 0.04445686936378479,
          "frac_active": 0.6875
        },
        {
          "step": 198,
          "loss": 0.7680749893188477,
          "floss_loss": 0.0002774876484181732,
          "kl_loss": 0.07677975296974182,
          "lyap1_mean": 0.015704579651355743,
          "lyap1_max": 0.0407920703291893,
          "lyap_mean": 0.007964217104017735,
          "volume_mean": 0.007964217104017735,
          "volume_max": 0.01851959340274334,
          "frac_active": 0.6875
        },
        {
          "step": 199,
          "loss": 5.161069393157959,
          "floss_loss": 0.00048213452100753784,
          "kl_loss": 0.5160587430000305,
          "lyap1_mean": 0.010169824585318565,
          "lyap1_max": 0.034441180527210236,
          "lyap_mean": 0.007940385490655899,
          "volume_mean": 0.007940385490655899,
          "volume_max": 0.030143067240715027,
          "frac_active": 0.625
        },
        {
          "step": 200,
          "loss": 0.5162805914878845,
          "floss_loss": 0.0001960603112820536,
          "kl_loss": 0.051608458161354065,
          "lyap1_mean": 0.002560691675171256,
          "lyap1_max": 0.01804419420659542,
          "lyap_mean": -0.0030060517601668835,
          "volume_mean": -0.0030060517601668835,
          "volume_max": 0.014490392059087753,
          "frac_active": 0.4375
        },
        {
          "step": 201,
          "loss": 2.2523834705352783,
          "floss_loss": 0.0004716968396678567,
          "kl_loss": 0.22519119083881378,
          "lyap1_mean": 0.0321585088968277,
          "lyap1_max": 0.061657946556806564,
          "lyap_mean": 0.0063131339848041534,
          "volume_mean": 0.0063131339848041534,
          "volume_max": 0.01057441160082817,
          "frac_active": 0.375
        },
        {
          "step": 202,
          "loss": 1.0016145706176758,
          "floss_loss": 0.00018991369870491326,
          "kl_loss": 0.1001424714922905,
          "lyap1_mean": 0.007119684014469385,
          "lyap1_max": 0.034636154770851135,
          "lyap_mean": 0.001350416918285191,
          "volume_mean": 0.001350416918285191,
          "volume_max": 0.014322696253657341,
          "frac_active": 0.6875
        },
        {
          "step": 203,
          "loss": 0.8802912831306458,
          "floss_loss": 0.0005804575048387051,
          "kl_loss": 0.08797108381986618,
          "lyap1_mean": 0.025865405797958374,
          "lyap1_max": 0.06964119523763657,
          "lyap_mean": -0.00044301978778094053,
          "volume_mean": -0.00044301978778094053,
          "volume_max": 0.01329639833420515,
          "frac_active": 0.25
        },
        {
          "step": 204,
          "loss": 0.12716694176197052,
          "floss_loss": 8.481197437504306e-05,
          "kl_loss": 0.012708212248980999,
          "lyap1_mean": 0.005306529812514782,
          "lyap1_max": 0.016949933022260666,
          "lyap_mean": 0.003133741905912757,
          "volume_mean": 0.003133741905912757,
          "volume_max": 0.010095421224832535,
          "frac_active": 0.5
        },
        {
          "step": 205,
          "loss": 0.4017416536808014,
          "floss_loss": 0.0001471916912123561,
          "kl_loss": 0.0401594452559948,
          "lyap1_mean": -0.004102039150893688,
          "lyap1_max": -7.175358041422442e-05,
          "lyap_mean": -0.009334946982562542,
          "volume_mean": -0.009334946982562542,
          "volume_max": -0.0035619796253740788,
          "frac_active": 0.0625
        },
        {
          "step": 206,
          "loss": 0.008836900815367699,
          "floss_loss": 0.00033800816163420677,
          "kl_loss": 0.0008498891838826239,
          "lyap1_mean": 0.016241014003753662,
          "lyap1_max": 0.03917184844613075,
          "lyap_mean": -0.005046492908149958,
          "volume_mean": -0.005046492908149958,
          "volume_max": 0.007430777885019779,
          "frac_active": 0.3125
        },
        {
          "step": 207,
          "loss": 0.5056202411651611,
          "floss_loss": 0.0003985474177170545,
          "kl_loss": 0.05052216723561287,
          "lyap1_mean": 0.001904848963022232,
          "lyap1_max": 0.04343605786561966,
          "lyap_mean": -0.007804404944181442,
          "volume_mean": -0.007804404944181442,
          "volume_max": 0.012739425525069237,
          "frac_active": 0.125
        },
        {
          "step": 208,
          "loss": 0.582323431968689,
          "floss_loss": 0.0003052027605008334,
          "kl_loss": 0.058201827108860016,
          "lyap1_mean": 0.0006430540233850479,
          "lyap1_max": 0.01451329980045557,
          "lyap_mean": -0.012402324005961418,
          "volume_mean": -0.012402324005961418,
          "volume_max": -0.004641357809305191,
          "frac_active": 0.125
        },
        {
          "step": 209,
          "loss": 0.009354308247566223,
          "floss_loss": 0.00047523778630420566,
          "kl_loss": 0.0008879070519469678,
          "lyap1_mean": -0.002272830344736576,
          "lyap1_max": 0.034022774547338486,
          "lyap_mean": -0.016141682863235474,
          "volume_mean": -0.016141682863235474,
          "volume_max": -0.003895713947713375,
          "frac_active": 0.0625
        },
        {
          "step": 210,
          "loss": 5.156907081604004,
          "floss_loss": 0.00021869684860575944,
          "kl_loss": 0.5156688094139099,
          "lyap1_mean": -0.0043230215087533,
          "lyap1_max": -0.001974965212866664,
          "lyap_mean": -0.01270236261188984,
          "volume_mean": -0.01270236261188984,
          "volume_max": -0.008638245984911919,
          "frac_active": 0.0
        },
        {
          "step": 211,
          "loss": 0.2633998394012451,
          "floss_loss": 0.0002467600570525974,
          "kl_loss": 0.02631530910730362,
          "lyap1_mean": -0.0045301178470253944,
          "lyap1_max": 0.011554885655641556,
          "lyap_mean": -0.010548126883804798,
          "volume_mean": -0.010548126883804798,
          "volume_max": -0.0007805947680026293,
          "frac_active": 0.125
        },
        {
          "step": 212,
          "loss": 0.0993739515542984,
          "floss_loss": 0.00026111959596164525,
          "kl_loss": 0.009911282919347286,
          "lyap1_mean": -0.002963298698887229,
          "lyap1_max": 0.010684428736567497,
          "lyap_mean": -0.012755421921610832,
          "volume_mean": -0.012755421921610832,
          "volume_max": -0.0052105397917330265,
          "frac_active": 0.0625
        },
        {
          "step": 213,
          "loss": 0.1642799973487854,
          "floss_loss": 0.0003444397880230099,
          "kl_loss": 0.01639355532824993,
          "lyap1_mean": 0.007133368402719498,
          "lyap1_max": 0.05398213490843773,
          "lyap_mean": -0.006489007733762264,
          "volume_mean": -0.006489007733762264,
          "volume_max": 0.004280935041606426,
          "frac_active": 0.25
        },
        {
          "step": 214,
          "loss": 2.4032552242279053,
          "floss_loss": 0.00019530716235749424,
          "kl_loss": 0.24030600488185883,
          "lyap1_mean": -0.007400921545922756,
          "lyap1_max": 0.007399403024464846,
          "lyap_mean": -0.010921476408839226,
          "volume_mean": -0.010921476408839226,
          "volume_max": -0.002913611475378275,
          "frac_active": 0.125
        },
        {
          "step": 215,
          "loss": 0.4853142499923706,
          "floss_loss": 0.00027071795193478465,
          "kl_loss": 0.04850435256958008,
          "lyap1_mean": -0.015831058844923973,
          "lyap1_max": -0.006021081004291773,
          "lyap_mean": -0.013126051053404808,
          "volume_mean": -0.013126051053404808,
          "volume_max": -0.008747175335884094,
          "frac_active": 0.0625
        },
        {
          "step": 216,
          "loss": 0.2532026767730713,
          "floss_loss": 0.0002388262510066852,
          "kl_loss": 0.02529638260602951,
          "lyap1_mean": -0.006070014555007219,
          "lyap1_max": 0.0003345476579852402,
          "lyap_mean": -0.012232834473252296,
          "volume_mean": -0.012232834473252296,
          "volume_max": -0.0063269007951021194,
          "frac_active": 0.1875
        },
        {
          "step": 217,
          "loss": 0.0027090576477348804,
          "floss_loss": 0.00047195007209666073,
          "kl_loss": 0.00022371076920535415,
          "lyap1_mean": -0.020505119115114212,
          "lyap1_max": -0.008891087025403976,
          "lyap_mean": -0.0206068754196167,
          "volume_mean": -0.0206068754196167,
          "volume_max": -0.01399943232536316,
          "frac_active": 0.0
        },
        {
          "step": 218,
          "loss": 0.7602964043617249,
          "floss_loss": 0.00031685736030340195,
          "kl_loss": 0.07599795609712601,
          "lyap1_mean": -0.011377345770597458,
          "lyap1_max": 0.005982508417218924,
          "lyap_mean": -0.015766285359859467,
          "volume_mean": -0.015766285359859467,
          "volume_max": -0.01022613886743784,
          "frac_active": 0.0625
        },
        {
          "step": 219,
          "loss": 0.4333614110946655,
          "floss_loss": 0.00031411461532115936,
          "kl_loss": 0.04330473020672798,
          "lyap1_mean": -0.0050755939446389675,
          "lyap1_max": 0.016293412074446678,
          "lyap_mean": -0.01512724906206131,
          "volume_mean": -0.01512724906206131,
          "volume_max": -0.009818082675337791,
          "frac_active": 0.0625
        },
        {
          "step": 220,
          "loss": 2.690887212753296,
          "floss_loss": 0.00029089153395034373,
          "kl_loss": 0.2690596282482147,
          "lyap1_mean": 0.006084215361624956,
          "lyap1_max": 0.016290903091430664,
          "lyap_mean": -0.010608121752738953,
          "volume_mean": -0.010608121752738953,
          "volume_max": -0.0004124552942812443,
          "frac_active": 0.25
        },
        {
          "step": 221,
          "loss": 0.14724799990653992,
          "floss_loss": 0.00042836819193325937,
          "kl_loss": 0.014681963250041008,
          "lyap1_mean": -0.010993912816047668,
          "lyap1_max": 0.012629788368940353,
          "lyap_mean": -0.015126774087548256,
          "volume_mean": -0.015126774087548256,
          "volume_max": -0.0006372900679707527,
          "frac_active": 0.1875
        },
        {
          "step": 222,
          "loss": 5.262965202331543,
          "floss_loss": 0.0002569832722656429,
          "kl_loss": 0.5262708067893982,
          "lyap1_mean": -0.01388099417090416,
          "lyap1_max": -0.0008859917288646102,
          "lyap_mean": -0.014259840361773968,
          "volume_mean": -0.014259840361773968,
          "volume_max": -0.008286818861961365,
          "frac_active": 0.0
        },
        {
          "step": 223,
          "loss": 0.17192231118679047,
          "floss_loss": 0.00038460202631540596,
          "kl_loss": 0.017153771594166756,
          "lyap1_mean": 0.0015151590341702104,
          "lyap1_max": 0.03203481808304787,
          "lyap_mean": -0.00616464065387845,
          "volume_mean": -0.00616464065387845,
          "volume_max": 0.018547384068369865,
          "frac_active": 0.375
        },
        {
          "step": 224,
          "loss": 0.19508351385593414,
          "floss_loss": 0.0005478216335177422,
          "kl_loss": 0.019453568384051323,
          "lyap1_mean": -0.013733172789216042,
          "lyap1_max": 0.008629785850644112,
          "lyap_mean": -0.021163109689950943,
          "volume_mean": -0.021163109689950943,
          "volume_max": -0.012878889217972755,
          "frac_active": 0.0625
        },
        {
          "step": 225,
          "loss": 0.0018127001821994781,
          "floss_loss": 0.0003430829383432865,
          "kl_loss": 0.0001469617272960022,
          "lyap1_mean": -0.008954419754445553,
          "lyap1_max": 0.0006958005833439529,
          "lyap_mean": -0.014517543837428093,
          "volume_mean": -0.014517543837428093,
          "volume_max": -0.006940935738384724,
          "frac_active": 0.125
        },
        {
          "step": 226,
          "loss": 0.08961482346057892,
          "floss_loss": 0.0006248624413274229,
          "kl_loss": 0.008898995816707611,
          "lyap1_mean": -0.020614881068468094,
          "lyap1_max": -0.018386494368314743,
          "lyap_mean": -0.024454165250062943,
          "volume_mean": -0.024454165250062943,
          "volume_max": -0.01854609325528145,
          "frac_active": 0.0
        },
        {
          "step": 227,
          "loss": 0.0991528332233429,
          "floss_loss": 0.00029354478465393186,
          "kl_loss": 0.009885928593575954,
          "lyap1_mean": -0.002713724272325635,
          "lyap1_max": 0.01749194785952568,
          "lyap_mean": -0.009770400822162628,
          "volume_mean": -0.009770400822162628,
          "volume_max": 0.00937245599925518,
          "frac_active": 0.3125
        },
        {
          "step": 228,
          "loss": 0.6375898122787476,
          "floss_loss": 0.00029166697640903294,
          "kl_loss": 0.06372981518507004,
          "lyap1_mean": -0.008419805206358433,
          "lyap1_max": 0.004811547696590424,
          "lyap_mean": -0.015489856712520123,
          "volume_mean": -0.015489856712520123,
          "volume_max": -0.00652408879250288,
          "frac_active": 0.0625
        },
        {
          "step": 229,
          "loss": 0.07172317802906036,
          "floss_loss": 0.0003327771555632353,
          "kl_loss": 0.007139039691537619,
          "lyap1_mean": -0.013001448474824429,
          "lyap1_max": -0.004590260796248913,
          "lyap_mean": -0.01560207549482584,
          "volume_mean": -0.01560207549482584,
          "volume_max": -0.003928788471966982,
          "frac_active": 0.0625
        },
        {
          "step": 230,
          "loss": 0.17402251064777374,
          "floss_loss": 0.00019575600163079798,
          "kl_loss": 0.017382675781846046,
          "lyap1_mean": -0.011163400486111641,
          "lyap1_max": -0.004231744445860386,
          "lyap_mean": -0.011869365349411964,
          "volume_mean": -0.011869365349411964,
          "volume_max": -0.0027949982322752476,
          "frac_active": 0.0
        },
        {
          "step": 231,
          "loss": 3.026756763458252,
          "floss_loss": 0.0004976834170520306,
          "kl_loss": 0.3026259243488312,
          "lyap1_mean": -0.00863666832447052,
          "lyap1_max": 0.0063600861467421055,
          "lyap_mean": -0.019531944766640663,
          "volume_mean": -0.019531944766640663,
          "volume_max": -0.01420586183667183,
          "frac_active": 0.125
        },
        {
          "step": 232,
          "loss": 0.21947741508483887,
          "floss_loss": 0.00032953836489468813,
          "kl_loss": 0.021914787590503693,
          "lyap1_mean": -0.007565552368760109,
          "lyap1_max": 0.009870683774352074,
          "lyap_mean": -0.013200119137763977,
          "volume_mean": -0.013200119137763977,
          "volume_max": -0.0044168345630168915,
          "frac_active": 0.125
        },
        {
          "step": 233,
          "loss": 0.0014004334807395935,
          "floss_loss": 0.0005102651193737984,
          "kl_loss": 8.901683759177104e-05,
          "lyap1_mean": -0.021292386576533318,
          "lyap1_max": -0.0145773496478796,
          "lyap_mean": -0.021579544991254807,
          "volume_mean": -0.021579544991254807,
          "volume_max": -0.012650460936129093,
          "frac_active": 0.0
        },
        {
          "step": 234,
          "loss": 3.126394033432007,
          "floss_loss": 0.0003422401496209204,
          "kl_loss": 0.31260520219802856,
          "lyap1_mean": -0.007909948006272316,
          "lyap1_max": 0.0015380578115582466,
          "lyap_mean": -0.015066270716488361,
          "volume_mean": -0.015066270716488361,
          "volume_max": -0.004823850933462381,
          "frac_active": 0.125
        },
        {
          "step": 235,
          "loss": 0.4592739939689636,
          "floss_loss": 0.00027472065994516015,
          "kl_loss": 0.045899927616119385,
          "lyap1_mean": 0.002733830362558365,
          "lyap1_max": 0.03047308325767517,
          "lyap_mean": -0.009650042280554771,
          "volume_mean": -0.009650042280554771,
          "volume_max": -0.0023162318393588066,
          "frac_active": 0.125
        },
        {
          "step": 236,
          "loss": 1.5488417148590088,
          "floss_loss": 0.00026421883376315236,
          "kl_loss": 0.15485775470733643,
          "lyap1_mean": 0.006621148437261581,
          "lyap1_max": 0.042566780000925064,
          "lyap_mean": -0.0068860407918691635,
          "volume_mean": -0.0068860407918691635,
          "volume_max": 0.0033480674028396606,
          "frac_active": 0.1875
        },
        {
          "step": 237,
          "loss": 1.24843168258667,
          "floss_loss": 0.0005806951667182148,
          "kl_loss": 0.12478510290384293,
          "lyap1_mean": 0.030178815126419067,
          "lyap1_max": 0.07022067159414291,
          "lyap_mean": 0.0013036716263741255,
          "volume_mean": 0.0013036716263741255,
          "volume_max": 0.01730736717581749,
          "frac_active": 0.3125
        },
        {
          "step": 238,
          "loss": 0.7821127772331238,
          "floss_loss": 0.00020246751955710351,
          "kl_loss": 0.07819102704524994,
          "lyap1_mean": -0.011096784844994545,
          "lyap1_max": -0.0016057657776400447,
          "lyap_mean": -0.01235363632440567,
          "volume_mean": -0.01235363632440567,
          "volume_max": -0.005572308786213398,
          "frac_active": 0.0625
        },
        {
          "step": 239,
          "loss": 2.177422523498535,
          "floss_loss": 0.00046215165639296174,
          "kl_loss": 0.21769605576992035,
          "lyap1_mean": -0.012173927389085293,
          "lyap1_max": 0.0007573330658487976,
          "lyap_mean": -0.017807612195611,
          "volume_mean": -0.017807612195611,
          "volume_max": -0.007842011749744415,
          "frac_active": 0.125
        },
        {
          "step": 240,
          "loss": 0.24069736897945404,
          "floss_loss": 0.0005423573893494904,
          "kl_loss": 0.024015501141548157,
          "lyap1_mean": -0.010775712318718433,
          "lyap1_max": 0.012954878620803356,
          "lyap_mean": -0.019196372479200363,
          "volume_mean": -0.019196372479200363,
          "volume_max": -0.003883764147758484,
          "frac_active": 0.0625
        },
        {
          "step": 241,
          "loss": 0.0009027498890645802,
          "floss_loss": 0.0004205991863273084,
          "kl_loss": 4.82150717289187e-05,
          "lyap1_mean": -0.01739262044429779,
          "lyap1_max": -0.009170599281787872,
          "lyap_mean": -0.019500911235809326,
          "volume_mean": -0.019500911235809326,
          "volume_max": -0.013235061429440975,
          "frac_active": 0.0
        },
        {
          "step": 242,
          "loss": 0.06750400364398956,
          "floss_loss": 0.00036087309126742184,
          "kl_loss": 0.006714312359690666,
          "lyap1_mean": 0.0082982974126935,
          "lyap1_max": 0.039349399507045746,
          "lyap_mean": -0.00866059772670269,
          "volume_mean": -0.00866059772670269,
          "volume_max": 0.0028386879712343216,
          "frac_active": 0.3125
        },
        {
          "step": 243,
          "loss": 0.26313498616218567,
          "floss_loss": 0.0006627875263802707,
          "kl_loss": 0.02624722197651863,
          "lyap1_mean": 0.008716394193470478,
          "lyap1_max": 0.05256395787000656,
          "lyap_mean": -0.013785529881715775,
          "volume_mean": -0.013785529881715775,
          "volume_max": -0.00032168254256248474,
          "frac_active": 0.1875
        },
        {
          "step": 244,
          "loss": 0.04493096470832825,
          "floss_loss": 0.0002866796567104757,
          "kl_loss": 0.004464428406208754,
          "lyap1_mean": 0.0029997776728123426,
          "lyap1_max": 0.017138320952653885,
          "lyap_mean": -0.011489175260066986,
          "volume_mean": -0.011489175260066986,
          "volume_max": -0.0024495082907378674,
          "frac_active": 0.125
        },
        {
          "step": 245,
          "loss": 0.2092752456665039,
          "floss_loss": 0.0003625264507718384,
          "kl_loss": 0.02089127153158188,
          "lyap1_mean": -0.009300557896494865,
          "lyap1_max": 0.0018221007194370031,
          "lyap_mean": -0.014801852405071259,
          "volume_mean": -0.014801852405071259,
          "volume_max": -0.005198866128921509,
          "frac_active": 0.1875
        },
        {
          "step": 246,
          "loss": 0.14950521290302277,
          "floss_loss": 0.00042373311589471996,
          "kl_loss": 0.014908148907124996,
          "lyap1_mean": -0.0025624679401516914,
          "lyap1_max": 0.013721004128456116,
          "lyap_mean": -0.015857268124818802,
          "volume_mean": -0.015857268124818802,
          "volume_max": -0.00863106269389391,
          "frac_active": 0.1875
        },
        {
          "step": 247,
          "loss": 0.2709568440914154,
          "floss_loss": 0.00035133626079186797,
          "kl_loss": 0.027060549706220627,
          "lyap1_mean": -0.008125473745167255,
          "lyap1_max": 0.0033673460129648447,
          "lyap_mean": -0.016248643398284912,
          "volume_mean": -0.016248643398284912,
          "volume_max": -0.008881742134690285,
          "frac_active": 0.125
        },
        {
          "step": 248,
          "loss": 7.0711212158203125,
          "floss_loss": 0.0004510906874202192,
          "kl_loss": 0.7070670127868652,
          "lyap1_mean": -0.0052548605017364025,
          "lyap1_max": 0.013464035466313362,
          "lyap_mean": -0.01754532754421234,
          "volume_mean": -0.01754532754421234,
          "volume_max": -0.011375639587640762,
          "frac_active": 0.0625
        },
        {
          "step": 249,
          "loss": 0.001413664547726512,
          "floss_loss": 0.0003654971078503877,
          "kl_loss": 0.00010481674689799547,
          "lyap1_mean": -0.018209900707006454,
          "lyap1_max": -0.01615847833454609,
          "lyap_mean": -0.017238039523363113,
          "volume_mean": -0.017238039523363113,
          "volume_max": -0.01138882152736187,
          "frac_active": 0.0625
        },
        {
          "step": 250,
          "loss": 0.6971708536148071,
          "floss_loss": 0.001644369913265109,
          "kl_loss": 0.06955264508724213,
          "lyap1_mean": 0.05627471208572388,
          "lyap1_max": 0.13203145563602448,
          "lyap_mean": 0.008265804499387741,
          "volume_mean": 0.008265804499387741,
          "volume_max": 0.0392245352268219,
          "frac_active": 0.4375
        },
        {
          "step": 251,
          "loss": 1.5493355989456177,
          "floss_loss": 0.0003196361067239195,
          "kl_loss": 0.15490159392356873,
          "lyap1_mean": -0.0071538109332323074,
          "lyap1_max": 0.008347169496119022,
          "lyap_mean": -0.015414373949170113,
          "volume_mean": -0.015414373949170113,
          "volume_max": -0.009945672936737537,
          "frac_active": 0.0625
        },
        {
          "step": 252,
          "loss": 0.12365932762622833,
          "floss_loss": 0.00021116549032740295,
          "kl_loss": 0.012344816699624062,
          "lyap1_mean": -0.007577718235552311,
          "lyap1_max": -0.0009668536949902773,
          "lyap_mean": -0.0135159557685256,
          "volume_mean": -0.0135159557685256,
          "volume_max": -0.013019186444580555,
          "frac_active": 0.0
        },
        {
          "step": 253,
          "loss": 3.0914065837860107,
          "floss_loss": 0.0005418931250460446,
          "kl_loss": 0.30908647179603577,
          "lyap1_mean": -0.015983138233423233,
          "lyap1_max": 0.006733214482665062,
          "lyap_mean": -0.021164514124393463,
          "volume_mean": -0.021164514124393463,
          "volume_max": -0.01549149677157402,
          "frac_active": 0.0625
        },
        {
          "step": 254,
          "loss": 0.021762438118457794,
          "floss_loss": 0.0005230641691014171,
          "kl_loss": 0.002123937476426363,
          "lyap1_mean": -0.019681468605995178,
          "lyap1_max": -0.006540221162140369,
          "lyap_mean": -0.02101568505167961,
          "volume_mean": -0.02101568505167961,
          "volume_max": -0.009761527180671692,
          "frac_active": 0.0
        },
        {
          "step": 255,
          "loss": 0.2821914851665497,
          "floss_loss": 0.00022777703998144716,
          "kl_loss": 0.028196370229125023,
          "lyap1_mean": -0.005766195710748434,
          "lyap1_max": 0.0046780589036643505,
          "lyap_mean": -0.012199534103274345,
          "volume_mean": -0.012199534103274345,
          "volume_max": -0.0053673130460083485,
          "frac_active": 0.0625
        },
        {
          "step": 256,
          "loss": 0.44068288803100586,
          "floss_loss": 0.00025783994351513684,
          "kl_loss": 0.04404250532388687,
          "lyap1_mean": -0.00748929800465703,
          "lyap1_max": 0.004349119495600462,
          "lyap_mean": -0.012055326253175735,
          "volume_mean": -0.012055326253175735,
          "volume_max": -0.005030306056141853,
          "frac_active": 0.125
        },
        {
          "step": 257,
          "loss": 0.0008820167277008295,
          "floss_loss": 0.00028905243379995227,
          "kl_loss": 5.92964279348962e-05,
          "lyap1_mean": 0.002031448297202587,
          "lyap1_max": 0.013501989655196667,
          "lyap_mean": -0.01041942834854126,
          "volume_mean": -0.01041942834854126,
          "volume_max": -0.0014578650007024407,
          "frac_active": 0.25
        },
        {
          "step": 258,
          "loss": 0.05182335153222084,
          "floss_loss": 0.00015164059004746377,
          "kl_loss": 0.0051671708934009075,
          "lyap1_mean": 0.009660344570875168,
          "lyap1_max": 0.016265859827399254,
          "lyap_mean": -7.32974149286747e-06,
          "volume_mean": -7.32974149286747e-06,
          "volume_max": 0.011804958805441856,
          "frac_active": 0.4375
        },
        {
          "step": 259,
          "loss": 0.08655573427677155,
          "floss_loss": 0.00035936234053224325,
          "kl_loss": 0.008619637228548527,
          "lyap1_mean": 0.012522362172603607,
          "lyap1_max": 0.05275523662567139,
          "lyap_mean": -0.004146983847022057,
          "volume_mean": -0.004146983847022057,
          "volume_max": 0.013761363923549652,
          "frac_active": 0.4375
        },
        {
          "step": 260,
          "loss": 0.17983263731002808,
          "floss_loss": 0.00016389373922720551,
          "kl_loss": 0.017966873943805695,
          "lyap1_mean": 0.002326288726180792,
          "lyap1_max": 0.014524656347930431,
          "lyap_mean": -0.006218159105628729,
          "volume_mean": -0.006218159105628729,
          "volume_max": 0.002630352508276701,
          "frac_active": 0.25
        },
        {
          "step": 261,
          "loss": 1.0015299320220947,
          "floss_loss": 0.00035577849484980106,
          "kl_loss": 0.10011741518974304,
          "lyap1_mean": -0.0048903366550803185,
          "lyap1_max": 0.03081700950860977,
          "lyap_mean": -0.012965830974280834,
          "volume_mean": -0.012965830974280834,
          "volume_max": 5.320040509104729e-05,
          "frac_active": 0.0625
        },
        {
          "step": 262,
          "loss": 0.1973179280757904,
          "floss_loss": 0.00017287340597249568,
          "kl_loss": 0.019714506343007088,
          "lyap1_mean": 0.009433130733668804,
          "lyap1_max": 0.019722692668437958,
          "lyap_mean": -0.003299119183793664,
          "volume_mean": -0.003299119183793664,
          "volume_max": 0.006623184308409691,
          "frac_active": 0.5
        },
        {
          "step": 263,
          "loss": 0.9040483236312866,
          "floss_loss": 0.00038045243127271533,
          "kl_loss": 0.09036678820848465,
          "lyap1_mean": -0.0051448531448841095,
          "lyap1_max": 0.011402594856917858,
          "lyap_mean": -0.013597952201962471,
          "volume_mean": -0.013597952201962471,
          "volume_max": 0.0028227970469743013,
          "frac_active": 0.25
        },
        {
          "step": 264,
          "loss": 0.34676674008369446,
          "floss_loss": 0.00033565470948815346,
          "kl_loss": 0.034643106162548065,
          "lyap1_mean": -0.009334051981568336,
          "lyap1_max": 0.0147949093952775,
          "lyap_mean": -0.015074157156050205,
          "volume_mean": -0.015074157156050205,
          "volume_max": -0.0014121057465672493,
          "frac_active": 0.0625
        },
        {
          "step": 265,
          "loss": 0.0025597289204597473,
          "floss_loss": 0.0003253020695410669,
          "kl_loss": 0.00022344267927110195,
          "lyap1_mean": -0.015019607730209827,
          "lyap1_max": -0.002621830441057682,
          "lyap_mean": -0.015904586762189865,
          "volume_mean": -0.015904586762189865,
          "volume_max": -0.002882228698581457,
          "frac_active": 0.0625
        },
        {
          "step": 266,
          "loss": 2.0966923236846924,
          "floss_loss": 0.0003051726089324802,
          "kl_loss": 0.20963871479034424,
          "lyap1_mean": -0.010709159076213837,
          "lyap1_max": 0.0014157019322738051,
          "lyap_mean": -0.01577567681670189,
          "volume_mean": -0.01577567681670189,
          "volume_max": -0.011395547538995743,
          "frac_active": 0.0625
        },
        {
          "step": 267,
          "loss": 0.40085697174072266,
          "floss_loss": 0.0004788083606399596,
          "kl_loss": 0.040037818253040314,
          "lyap1_mean": -0.012837680988013744,
          "lyap1_max": 0.006895024329423904,
          "lyap_mean": -0.015567958354949951,
          "volume_mean": -0.015567958354949951,
          "volume_max": 0.002465259749442339,
          "frac_active": 0.3125
        },
        {
          "step": 268,
          "loss": 0.09647148847579956,
          "floss_loss": 0.0003156989114359021,
          "kl_loss": 0.0096155796200037,
          "lyap1_mean": 0.007163393311202526,
          "lyap1_max": 0.01564275100827217,
          "lyap_mean": -0.010960678569972515,
          "volume_mean": -0.010960678569972515,
          "volume_max": -0.002989222062751651,
          "frac_active": 0.1875
        },
        {
          "step": 269,
          "loss": 8.411745071411133,
          "floss_loss": 0.00036966108018532395,
          "kl_loss": 0.8411374688148499,
          "lyap1_mean": -0.012341691181063652,
          "lyap1_max": 0.002038470935076475,
          "lyap_mean": -0.017184065654873848,
          "volume_mean": -0.017184065654873848,
          "volume_max": -0.012516709044575691,
          "frac_active": 0.125
        },
        {
          "step": 270,
          "loss": 2.5008134841918945,
          "floss_loss": 0.0004715078684967011,
          "kl_loss": 0.2500341832637787,
          "lyap1_mean": -0.013999980874359608,
          "lyap1_max": -0.0038156474474817514,
          "lyap_mean": -0.020458221435546875,
          "volume_mean": -0.020458221435546875,
          "volume_max": -0.016477186232805252,
          "frac_active": 0.0
        },
        {
          "step": 271,
          "loss": 1.9468048810958862,
          "floss_loss": 0.000519364548381418,
          "kl_loss": 0.1946285516023636,
          "lyap1_mean": -0.020973753184080124,
          "lyap1_max": -0.011859067715704441,
          "lyap_mean": -0.02209065854549408,
          "volume_mean": -0.02209065854549408,
          "volume_max": -0.015695810317993164,
          "frac_active": 0.0
        },
        {
          "step": 272,
          "loss": 0.7395974397659302,
          "floss_loss": 0.00045956362737342715,
          "kl_loss": 0.07391379028558731,
          "lyap1_mean": 0.0015735030174255371,
          "lyap1_max": 0.03775661438703537,
          "lyap_mean": -0.014226492494344711,
          "volume_mean": -0.014226492494344711,
          "volume_max": -0.009011982940137386,
          "frac_active": 0.0625
        },
        {
          "step": 273,
          "loss": 2.299062728881836,
          "floss_loss": 0.0005004105041734874,
          "kl_loss": 0.2298562377691269,
          "lyap1_mean": 0.01675177551805973,
          "lyap1_max": 0.03959707170724869,
          "lyap_mean": -0.00913497619330883,
          "volume_mean": -0.00913497619330883,
          "volume_max": 0.004870936274528503,
          "frac_active": 0.125
        },
        {
          "step": 274,
          "loss": 2.8672103881835938,
          "floss_loss": 0.00044782698387280107,
          "kl_loss": 0.2866762578487396,
          "lyap1_mean": -0.007611475884914398,
          "lyap1_max": 0.007473646197468042,
          "lyap_mean": -0.013125430792570114,
          "volume_mean": -0.013125430792570114,
          "volume_max": 0.0017228182405233383,
          "frac_active": 0.1875
        },
        {
          "step": 275,
          "loss": 1.9090499877929688,
          "floss_loss": 0.001460102153941989,
          "kl_loss": 0.19075898826122284,
          "lyap1_mean": 0.045841626822948456,
          "lyap1_max": 0.09799166023731232,
          "lyap_mean": -0.0022298023104667664,
          "volume_mean": -0.0022298023104667664,
          "volume_max": 0.008996360003948212,
          "frac_active": 0.1875
        },
        {
          "step": 276,
          "loss": 0.828082799911499,
          "floss_loss": 0.0003025828336831182,
          "kl_loss": 0.08277802169322968,
          "lyap1_mean": 0.01458941400051117,
          "lyap1_max": 0.0533766932785511,
          "lyap_mean": 0.0014552462380379438,
          "volume_mean": 0.0014552462380379438,
          "volume_max": 0.01626737229526043,
          "frac_active": 0.375
        },
        {
          "step": 277,
          "loss": 0.49182090163230896,
          "floss_loss": 0.00020969114848412573,
          "kl_loss": 0.049161121249198914,
          "lyap1_mean": 0.004622414708137512,
          "lyap1_max": 0.029892312362790108,
          "lyap_mean": -0.007799268700182438,
          "volume_mean": -0.007799268700182438,
          "volume_max": -0.005353763699531555,
          "frac_active": 0.1875
        },
        {
          "step": 278,
          "loss": 0.17128844559192657,
          "floss_loss": 0.00017301009211223572,
          "kl_loss": 0.01711154356598854,
          "lyap1_mean": -0.004591005854308605,
          "lyap1_max": 0.008929946459829807,
          "lyap_mean": -0.007553738076239824,
          "volume_mean": -0.007553738076239824,
          "volume_max": 0.00290668080560863,
          "frac_active": 0.25
        },
        {
          "step": 279,
          "loss": 0.5467201471328735,
          "floss_loss": 0.0004929418209940195,
          "kl_loss": 0.054622720927000046,
          "lyap1_mean": -0.007415862288326025,
          "lyap1_max": 0.030132362619042397,
          "lyap_mean": -0.014147692359983921,
          "volume_mean": -0.014147692359983921,
          "volume_max": 0.005049463827162981,
          "frac_active": 0.25
        },
        {
          "step": 280,
          "loss": 1.9990088939666748,
          "floss_loss": 0.0003040054871235043,
          "kl_loss": 0.1998704969882965,
          "lyap1_mean": 0.011449435725808144,
          "lyap1_max": 0.02527119778096676,
          "lyap_mean": 0.002128109335899353,
          "volume_mean": 0.002128109335899353,
          "volume_max": 0.020782118663191795,
          "frac_active": 0.4375
        },
        {
          "step": 281,
          "loss": 0.003812020644545555,
          "floss_loss": 0.0012474155519157648,
          "kl_loss": 0.00025646050926297903,
          "lyap1_mean": 0.03315112739801407,
          "lyap1_max": 0.0741884708404541,
          "lyap_mean": 0.011023908853530884,
          "volume_mean": 0.011023908853530884,
          "volume_max": 0.05061578005552292,
          "frac_active": 0.625
        },
        {
          "step": 282,
          "loss": 1.727525234222412,
          "floss_loss": 0.000685619655996561,
          "kl_loss": 0.1726839691400528,
          "lyap1_mean": 0.003388546407222748,
          "lyap1_max": 0.049122389405965805,
          "lyap_mean": -0.009340906515717506,
          "volume_mean": -0.009340906515717506,
          "volume_max": 0.012163901701569557,
          "frac_active": 0.3125
        },
        {
          "step": 283,
          "loss": 0.9873592853546143,
          "floss_loss": 0.0016628892626613379,
          "kl_loss": 0.09856963902711868,
          "lyap1_mean": 0.042623404413461685,
          "lyap1_max": 0.0750121995806694,
          "lyap_mean": 0.02408766932785511,
          "volume_mean": 0.02408766932785511,
          "volume_max": 0.04435654729604721,
          "frac_active": 0.75
        },
        {
          "step": 284,
          "loss": 0.0955655574798584,
          "floss_loss": 0.0005778230261057615,
          "kl_loss": 0.009498773142695427,
          "lyap1_mean": 0.03288447856903076,
          "lyap1_max": 0.05623944103717804,
          "lyap_mean": 0.014464054256677628,
          "volume_mean": 0.014464054256677628,
          "volume_max": 0.03366668522357941,
          "frac_active": 0.6875
        },
        {
          "step": 285,
          "loss": 0.4374602735042572,
          "floss_loss": 0.0005200197920203209,
          "kl_loss": 0.04369402676820755,
          "lyap1_mean": 0.037680674344301224,
          "lyap1_max": 0.04465894773602486,
          "lyap_mean": 0.010925388894975185,
          "volume_mean": 0.010925388894975185,
          "volume_max": 0.023971419781446457,
          "frac_active": 0.6875
        },
        {
          "step": 286,
          "loss": 0.0028117159381508827,
          "floss_loss": 0.00042652373667806387,
          "kl_loss": 0.00023851920559536666,
          "lyap1_mean": 0.025628775358200073,
          "lyap1_max": 0.042873281985521317,
          "lyap_mean": 0.005037332884967327,
          "volume_mean": 0.005037332884967327,
          "volume_max": 0.013359839096665382,
          "frac_active": 0.625
        },
        {
          "step": 287,
          "loss": 0.37446680665016174,
          "floss_loss": 0.0009418244590051472,
          "kl_loss": 0.03735249862074852,
          "lyap1_mean": 0.028550613671541214,
          "lyap1_max": 0.07659877836704254,
          "lyap_mean": 0.005180053412914276,
          "volume_mean": 0.005180053412914276,
          "volume_max": 0.04213636368513107,
          "frac_active": 0.375
        },
        {
          "step": 288,
          "loss": 1.153031826019287,
          "floss_loss": 0.0005059436080045998,
          "kl_loss": 0.11525259166955948,
          "lyap1_mean": 0.010661199688911438,
          "lyap1_max": 0.03029714711010456,
          "lyap_mean": -0.012235306203365326,
          "volume_mean": -0.012235306203365326,
          "volume_max": -0.004232797771692276,
          "frac_active": 0.1875
        },
        {
          "step": 289,
          "loss": 0.0018994970014318824,
          "floss_loss": 0.0004370681708678603,
          "kl_loss": 0.00014624287723563612,
          "lyap1_mean": -0.00013225036673247814,
          "lyap1_max": 0.01183027308434248,
          "lyap_mean": -0.016432035714387894,
          "volume_mean": -0.016432035714387894,
          "volume_max": -0.006053559482097626,
          "frac_active": 0.0625
        },
        {
          "step": 290,
          "loss": 0.31710171699523926,
          "floss_loss": 0.00042818300426006317,
          "kl_loss": 0.031667355448007584,
          "lyap1_mean": -0.00711458083242178,
          "lyap1_max": 0.004451264627277851,
          "lyap_mean": -0.01701655425131321,
          "volume_mean": -0.01701655425131321,
          "volume_max": -0.0095573291182518,
          "frac_active": 0.0625
        },
        {
          "step": 291,
          "loss": 0.07064683735370636,
          "floss_loss": 0.0006389610935002565,
          "kl_loss": 0.007000787183642387,
          "lyap1_mean": -0.006462774705141783,
          "lyap1_max": 0.0092605110257864,
          "lyap_mean": -0.021253181621432304,
          "volume_mean": -0.021253181621432304,
          "volume_max": -0.013294778764247894,
          "frac_active": 0.125
        },
        {
          "step": 292,
          "loss": 3.9372618198394775,
          "floss_loss": 0.00048669963143765926,
          "kl_loss": 0.3936775326728821,
          "lyap1_mean": -0.0011187883792445064,
          "lyap1_max": 0.011519713327288628,
          "lyap_mean": -0.01499093510210514,
          "volume_mean": -0.01499093510210514,
          "volume_max": 0.00016749859787523746,
          "frac_active": 0.25
        },
        {
          "step": 293,
          "loss": 0.08424505591392517,
          "floss_loss": 0.0006818092660978436,
          "kl_loss": 0.008356324397027493,
          "lyap1_mean": -0.004788504447788,
          "lyap1_max": 0.02239767089486122,
          "lyap_mean": -0.01968427002429962,
          "volume_mean": -0.01968427002429962,
          "volume_max": -0.008925261907279491,
          "frac_active": 0.125
        },
        {
          "step": 294,
          "loss": 9.315251350402832,
          "floss_loss": 0.0009906628401950002,
          "kl_loss": 0.9314260482788086,
          "lyap1_mean": -0.02810513786971569,
          "lyap1_max": -0.022202245891094208,
          "lyap_mean": -0.030820876359939575,
          "volume_mean": -0.030820876359939575,
          "volume_max": -0.02583906054496765,
          "frac_active": 0.0
        },
        {
          "step": 295,
          "loss": 0.3518885672092438,
          "floss_loss": 0.0007269177003763616,
          "kl_loss": 0.03511616587638855,
          "lyap1_mean": -0.022486701607704163,
          "lyap1_max": -0.012841248884797096,
          "lyap_mean": -0.024686578661203384,
          "volume_mean": -0.024686578661203384,
          "volume_max": -0.011797797866165638,
          "frac_active": 0.0
        },
        {
          "step": 296,
          "loss": 1.5760327577590942,
          "floss_loss": 0.00040737411472946405,
          "kl_loss": 0.15756253898143768,
          "lyap1_mean": -0.004895136691629887,
          "lyap1_max": 0.029456408694386482,
          "lyap_mean": -0.0150592140853405,
          "volume_mean": -0.0150592140853405,
          "volume_max": -0.008211705833673477,
          "frac_active": 0.0625
        },
        {
          "step": 297,
          "loss": 0.13998353481292725,
          "floss_loss": 0.0007255421951413155,
          "kl_loss": 0.01392580009996891,
          "lyap1_mean": -0.013290736824274063,
          "lyap1_max": 0.0035658031702041626,
          "lyap_mean": -0.02458002418279648,
          "volume_mean": -0.02458002418279648,
          "volume_max": -0.017792195081710815,
          "frac_active": 0.125
        },
        {
          "step": 298,
          "loss": 0.7287922501564026,
          "floss_loss": 0.0008967269095592201,
          "kl_loss": 0.07278954982757568,
          "lyap1_mean": -0.018265601247549057,
          "lyap1_max": 0.014715712517499924,
          "lyap_mean": -0.02725611999630928,
          "volume_mean": -0.02725611999630928,
          "volume_max": -0.015358617529273033,
          "frac_active": 0.0625
        },
        {
          "step": 299,
          "loss": 3.3833141326904297,
          "floss_loss": 0.0006515997811220586,
          "kl_loss": 0.3382662534713745,
          "lyap1_mean": -0.0069175357930362225,
          "lyap1_max": 0.039567116647958755,
          "lyap_mean": -0.019770801067352295,
          "volume_mean": -0.019770801067352295,
          "volume_max": -0.006579468958079815,
          "frac_active": 0.0625
        },
        {
          "step": 300,
          "loss": 0.1663520187139511,
          "floss_loss": 0.0006735221832059324,
          "kl_loss": 0.01656785048544407,
          "lyap1_mean": -0.02098240703344345,
          "lyap1_max": -0.014207536354660988,
          "lyap_mean": -0.02547995187342167,
          "volume_mean": -0.02547995187342167,
          "volume_max": -0.022506272420287132,
          "frac_active": 0.0
        },
        {
          "step": 301,
          "loss": 0.4022296369075775,
          "floss_loss": 0.0006311757024377584,
          "kl_loss": 0.04015984386205673,
          "lyap1_mean": -0.021639956161379814,
          "lyap1_max": -0.014774158596992493,
          "lyap_mean": -0.024341177195310593,
          "volume_mean": -0.024341177195310593,
          "volume_max": -0.016946502029895782,
          "frac_active": 0.0
        },
        {
          "step": 302,
          "loss": 0.015133854933083057,
          "floss_loss": 0.0005074344226159155,
          "kl_loss": 0.0014626419870182872,
          "lyap1_mean": -0.015613086521625519,
          "lyap1_max": 0.004072103649377823,
          "lyap_mean": -0.02001059427857399,
          "volume_mean": -0.02001059427857399,
          "volume_max": -0.010070517659187317,
          "frac_active": 0.125
        },
        {
          "step": 303,
          "loss": 0.2202448695898056,
          "floss_loss": 0.0003017054405063391,
          "kl_loss": 0.021994316950440407,
          "lyap1_mean": 0.0005455105565488338,
          "lyap1_max": 0.03162398934364319,
          "lyap_mean": -0.010651381686329842,
          "volume_mean": -0.010651381686329842,
          "volume_max": -0.0013491949066519737,
          "frac_active": 0.125
        },
        {
          "step": 304,
          "loss": 0.24548271298408508,
          "floss_loss": 0.0005055652582086623,
          "kl_loss": 0.024497713893651962,
          "lyap1_mean": -0.01870645396411419,
          "lyap1_max": -0.012630654498934746,
          "lyap_mean": -0.020880287513136864,
          "volume_mean": -0.020880287513136864,
          "volume_max": -0.014292198233306408,
          "frac_active": 0.0
        },
        {
          "step": 305,
          "loss": 0.0012085698544979095,
          "floss_loss": 0.0005382778472267091,
          "kl_loss": 6.702920654788613e-05,
          "lyap1_mean": 0.024054737761616707,
          "lyap1_max": 0.0819142758846283,
          "lyap_mean": 0.0044093141332268715,
          "volume_mean": 0.0044093141332268715,
          "volume_max": 0.01755928248167038,
          "frac_active": 0.4375
        },
        {
          "step": 306,
          "loss": 0.08152373135089874,
          "floss_loss": 0.0003439132997300476,
          "kl_loss": 0.008117982186377048,
          "lyap1_mean": -0.012883041985332966,
          "lyap1_max": 0.003472563112154603,
          "lyap_mean": -0.012026115320622921,
          "volume_mean": -0.012026115320622921,
          "volume_max": 0.0007352686952799559,
          "frac_active": 0.3125
        },
        {
          "step": 307,
          "loss": 0.0946098193526268,
          "floss_loss": 0.000645945081487298,
          "kl_loss": 0.0093963872641325,
          "lyap1_mean": 0.0010425408836454153,
          "lyap1_max": 0.04480870068073273,
          "lyap_mean": -0.015710830688476562,
          "volume_mean": -0.015710830688476562,
          "volume_max": 0.004054848104715347,
          "frac_active": 0.0625
        },
        {
          "step": 308,
          "loss": 0.13570082187652588,
          "floss_loss": 0.0004265003080945462,
          "kl_loss": 0.013527431525290012,
          "lyap1_mean": 0.028767859563231468,
          "lyap1_max": 0.0352645218372345,
          "lyap_mean": 0.0046255458146333694,
          "volume_mean": 0.0046255458146333694,
          "volume_max": 0.01004047505557537,
          "frac_active": 0.5625
        },
        {
          "step": 309,
          "loss": 0.11021983623504639,
          "floss_loss": 0.00020713778212666512,
          "kl_loss": 0.011001269333064556,
          "lyap1_mean": 0.011808007024228573,
          "lyap1_max": 0.024622833356261253,
          "lyap_mean": -0.005235846620053053,
          "volume_mean": -0.005235846620053053,
          "volume_max": 0.0011254195123910904,
          "frac_active": 0.3125
        },
        {
          "step": 310,
          "loss": 6.417064666748047,
          "floss_loss": 0.0003106202930212021,
          "kl_loss": 0.6416754126548767,
          "lyap1_mean": 0.03008432313799858,
          "lyap1_max": 0.03894728049635887,
          "lyap_mean": 0.0037693888880312443,
          "volume_mean": 0.0037693888880312443,
          "volume_max": 0.0102546913549304,
          "frac_active": 0.4375
        },
        {
          "step": 311,
          "loss": 0.24915091693401337,
          "floss_loss": 0.00035548978485167027,
          "kl_loss": 0.024879541248083115,
          "lyap1_mean": 0.01632741093635559,
          "lyap1_max": 0.03786955401301384,
          "lyap_mean": -0.006344938185065985,
          "volume_mean": -0.006344938185065985,
          "volume_max": 0.00728636234998703,
          "frac_active": 0.375
        },
        {
          "step": 312,
          "loss": 2.449979066848755,
          "floss_loss": 0.00024777904036454856,
          "kl_loss": 0.24497312307357788,
          "lyap1_mean": 0.01637374237179756,
          "lyap1_max": 0.03153303638100624,
          "lyap_mean": -0.002216123975813389,
          "volume_mean": -0.002216123975813389,
          "volume_max": 0.005672394298017025,
          "frac_active": 0.5
        },
        {
          "step": 313,
          "loss": 0.0032049608416855335,
          "floss_loss": 0.0005871095927432179,
          "kl_loss": 0.0002617851132526994,
          "lyap1_mean": 0.023511255159974098,
          "lyap1_max": 0.04527837410569191,
          "lyap_mean": 0.0020502752158790827,
          "volume_mean": 0.0020502752158790827,
          "volume_max": 0.03306692838668823,
          "frac_active": 0.4375
        },
        {
          "step": 314,
          "loss": 0.934696614742279,
          "floss_loss": 0.0002468271413818002,
          "kl_loss": 0.09344498068094254,
          "lyap1_mean": 0.01695392094552517,
          "lyap1_max": 0.025374194607138634,
          "lyap_mean": -0.004075550474226475,
          "volume_mean": -0.004075550474226475,
          "volume_max": 0.004107339773327112,
          "frac_active": 0.4375
        },
        {
          "step": 315,
          "loss": 0.6461278796195984,
          "floss_loss": 0.00025397652643732727,
          "kl_loss": 0.06458739191293716,
          "lyap1_mean": 0.0013034376315772533,
          "lyap1_max": 0.024195192381739616,
          "lyap_mean": -0.007674195803701878,
          "volume_mean": -0.007674195803701878,
          "volume_max": 0.004372327588498592,
          "frac_active": 0.1875
        },
        {
          "step": 316,
          "loss": 0.13947352766990662,
          "floss_loss": 0.0004861110937781632,
          "kl_loss": 0.013898742385208607,
          "lyap1_mean": -0.010133701376616955,
          "lyap1_max": 0.00487304013222456,
          "lyap_mean": -0.018210289999842644,
          "volume_mean": -0.018210289999842644,
          "volume_max": -0.009362930431962013,
          "frac_active": 0.125
        },
        {
          "step": 317,
          "loss": 0.1506795883178711,
          "floss_loss": 0.0004430451081134379,
          "kl_loss": 0.015023655258119106,
          "lyap1_mean": -0.014849288389086723,
          "lyap1_max": -0.00932723842561245,
          "lyap_mean": -0.018420539796352386,
          "volume_mean": -0.018420539796352386,
          "volume_max": -0.009918979369103909,
          "frac_active": 0.0625
        },
        {
          "step": 318,
          "loss": 0.006166895851492882,
          "floss_loss": 0.0003849340137094259,
          "kl_loss": 0.0005781961954198778,
          "lyap1_mean": -0.005960386712104082,
          "lyap1_max": 0.022136030718684196,
          "lyap_mean": -0.00897949282079935,
          "volume_mean": -0.00897949282079935,
          "volume_max": 0.017730139195919037,
          "frac_active": 0.3125
        },
        {
          "step": 319,
          "loss": 0.25411659479141235,
          "floss_loss": 0.00022328976774588227,
          "kl_loss": 0.025389332324266434,
          "lyap1_mean": -0.00713421031832695,
          "lyap1_max": 0.006365681067109108,
          "lyap_mean": -0.012382770888507366,
          "volume_mean": -0.012382770888507366,
          "volume_max": -0.006996702402830124,
          "frac_active": 0.0625
        },
        {
          "step": 320,
          "loss": 0.45429688692092896,
          "floss_loss": 0.0002075995726045221,
          "kl_loss": 0.04540892690420151,
          "lyap1_mean": -0.008585173636674881,
          "lyap1_max": -0.002827488584443927,
          "lyap_mean": -0.011584488674998283,
          "volume_mean": -0.011584488674998283,
          "volume_max": -0.003753634402528405,
          "frac_active": 0.0625
        },
        {
          "step": 321,
          "loss": 0.006476159673184156,
          "floss_loss": 0.0003880964941345155,
          "kl_loss": 0.0006088063237257302,
          "lyap1_mean": -0.01210339181125164,
          "lyap1_max": 0.007964080199599266,
          "lyap_mean": -0.016150809824466705,
          "volume_mean": -0.016150809824466705,
          "volume_max": -0.005090095102787018,
          "frac_active": 0.125
        },
        {
          "step": 322,
          "loss": 0.06282753497362137,
          "floss_loss": 0.0002968785120174289,
          "kl_loss": 0.006253066007047892,
          "lyap1_mean": -0.008707646280527115,
          "lyap1_max": -0.0040022218599915504,
          "lyap_mean": -0.015828104689717293,
          "volume_mean": -0.015828104689717293,
          "volume_max": -0.010893174447119236,
          "frac_active": 0.0
        },
        {
          "step": 323,
          "loss": 0.1496194750070572,
          "floss_loss": 0.0001985794515348971,
          "kl_loss": 0.014942090027034283,
          "lyap1_mean": -0.001537275966256857,
          "lyap1_max": 0.009811091236770153,
          "lyap_mean": -0.007397213950753212,
          "volume_mean": -0.007397213950753212,
          "volume_max": 0.0012277024798095226,
          "frac_active": 0.1875
        },
        {
          "step": 324,
          "loss": 0.09209880232810974,
          "floss_loss": 0.00015036892727948725,
          "kl_loss": 0.009194843471050262,
          "lyap1_mean": 0.008788013830780983,
          "lyap1_max": 0.022580690681934357,
          "lyap_mean": -0.004416442010551691,
          "volume_mean": -0.004416442010551691,
          "volume_max": 0.00289118941873312,
          "frac_active": 0.375
        },
        {
          "step": 325,
          "loss": 0.08085489273071289,
          "floss_loss": 0.00019784797041211277,
          "kl_loss": 0.008065704256296158,
          "lyap1_mean": -0.0009427075274288654,
          "lyap1_max": 0.015765061601996422,
          "lyap_mean": -0.0018592489650472999,
          "volume_mean": -0.0018592489650472999,
          "volume_max": 0.004512316547334194,
          "frac_active": 0.3125
        },
        {
          "step": 326,
          "loss": 0.27243468165397644,
          "floss_loss": 0.00036989146610721946,
          "kl_loss": 0.027206480503082275,
          "lyap1_mean": 0.023896899074316025,
          "lyap1_max": 0.038117632269859314,
          "lyap_mean": 0.002439405070617795,
          "volume_mean": 0.002439405070617795,
          "volume_max": 0.015077397227287292,
          "frac_active": 0.5625
        },
        {
          "step": 327,
          "loss": 0.3243463635444641,
          "floss_loss": 0.00037019397132098675,
          "kl_loss": 0.032397616654634476,
          "lyap1_mean": 0.025819992646574974,
          "lyap1_max": 0.0456860177218914,
          "lyap_mean": 0.007492519449442625,
          "volume_mean": 0.007492519449442625,
          "volume_max": 0.019059855490922928,
          "frac_active": 0.625
        },
        {
          "step": 328,
          "loss": 0.14959357678890228,
          "floss_loss": 0.0006750341271981597,
          "kl_loss": 0.014891854487359524,
          "lyap1_mean": 0.02867356315255165,
          "lyap1_max": 0.06954113394021988,
          "lyap_mean": -0.004442407749593258,
          "volume_mean": -0.004442407749593258,
          "volume_max": 0.0035232692025601864,
          "frac_active": 0.25
        },
        {
          "step": 329,
          "loss": 0.0007500644423998892,
          "floss_loss": 0.0006485058693215251,
          "kl_loss": 1.015585530694807e-05,
          "lyap1_mean": 0.0027875155210494995,
          "lyap1_max": 0.037714067846536636,
          "lyap_mean": -0.01281679142266512,
          "volume_mean": -0.01281679142266512,
          "volume_max": 0.0009917072020471096,
          "frac_active": 0.3125
        },
        {
          "step": 330,
          "loss": 6.546830654144287,
          "floss_loss": 0.0002674852730706334,
          "kl_loss": 0.6546562910079956,
          "lyap1_mean": 0.00996567215770483,
          "lyap1_max": 0.02755090221762657,
          "lyap_mean": -0.0026107318699359894,
          "volume_mean": -0.0026107318699359894,
          "volume_max": 0.018356308341026306,
          "frac_active": 0.375
        },
        {
          "step": 331,
          "loss": 0.8697050213813782,
          "floss_loss": 0.0001394652499584481,
          "kl_loss": 0.08695655316114426,
          "lyap1_mean": 0.018610280007123947,
          "lyap1_max": 0.02824554033577442,
          "lyap_mean": 0.004474634304642677,
          "volume_mean": 0.004474634304642677,
          "volume_max": 0.007900888100266457,
          "frac_active": 0.5625
        },
        {
          "step": 332,
          "loss": 0.11853790283203125,
          "floss_loss": 0.00016814490663819015,
          "kl_loss": 0.011836975812911987,
          "lyap1_mean": 0.013445832766592503,
          "lyap1_max": 0.018499422818422318,
          "lyap_mean": -0.003668183460831642,
          "volume_mean": -0.003668183460831642,
          "volume_max": 0.005269512999802828,
          "frac_active": 0.375
        },
        {
          "step": 333,
          "loss": 0.18749889731407166,
          "floss_loss": 0.00030005205189809203,
          "kl_loss": 0.018719885498285294,
          "lyap1_mean": 0.022839874029159546,
          "lyap1_max": 0.03935385122895241,
          "lyap_mean": 0.006306204944849014,
          "volume_mean": 0.006306204944849014,
          "volume_max": 0.01752385124564171,
          "frac_active": 0.625
        },
        {
          "step": 334,
          "loss": 0.0002621333405841142,
          "floss_loss": 0.00017968815518543124,
          "kl_loss": 8.244519449362997e-06,
          "lyap1_mean": 0.00035980413667857647,
          "lyap1_max": 0.01121581718325615,
          "lyap_mean": -0.0065720840357244015,
          "volume_mean": -0.0065720840357244015,
          "volume_max": 0.0037651625461876392,
          "frac_active": 0.25
        },
        {
          "step": 335,
          "loss": 0.14227458834648132,
          "floss_loss": 0.00044738774886354804,
          "kl_loss": 0.014182720333337784,
          "lyap1_mean": -0.008991943672299385,
          "lyap1_max": 0.004916493780910969,
          "lyap_mean": -0.014336830005049706,
          "volume_mean": -0.014336830005049706,
          "volume_max": 0.006989697925746441,
          "frac_active": 0.25
        },
        {
          "step": 336,
          "loss": 3.7640254497528076,
          "floss_loss": 0.0002505612501408905,
          "kl_loss": 0.37637749314308167,
          "lyap1_mean": -0.013060836121439934,
          "lyap1_max": 0.0004902651417069137,
          "lyap_mean": -0.0140693299472332,
          "volume_mean": -0.0140693299472332,
          "volume_max": -0.007812965661287308,
          "frac_active": 0.0625
        },
        {
          "step": 337,
          "loss": 0.002073140349239111,
          "floss_loss": 0.00021363793348427862,
          "kl_loss": 0.0001859502517618239,
          "lyap1_mean": 0.012968722730875015,
          "lyap1_max": 0.03303291276097298,
          "lyap_mean": -0.003398521803319454,
          "volume_mean": -0.003398521803319454,
          "volume_max": 0.004985978361219168,
          "frac_active": 0.375
        },
        {
          "step": 338,
          "loss": 2.5282726287841797,
          "floss_loss": 0.0003032132226508111,
          "kl_loss": 0.2527969479560852,
          "lyap1_mean": -0.01282426156103611,
          "lyap1_max": -0.0018180222250521183,
          "lyap_mean": -0.014940784312784672,
          "volume_mean": -0.014940784312784672,
          "volume_max": -0.0029447837732732296,
          "frac_active": 0.0625
        },
        {
          "step": 339,
          "loss": 2.9413669109344482,
          "floss_loss": 0.0005344932433217764,
          "kl_loss": 0.29408323764801025,
          "lyap1_mean": -0.018106210976839066,
          "lyap1_max": -0.005031026899814606,
          "lyap_mean": -0.02195313386619091,
          "volume_mean": -0.02195313386619091,
          "volume_max": -0.013489284552633762,
          "frac_active": 0.0
        },
        {
          "step": 340,
          "loss": 1.4482654333114624,
          "floss_loss": 0.0004852208658121526,
          "kl_loss": 0.1447780281305313,
          "lyap1_mean": -0.015744008123874664,
          "lyap1_max": 0.00024137429136317223,
          "lyap_mean": -0.01943329907953739,
          "volume_mean": -0.01943329907953739,
          "volume_max": -0.0032604984007775784,
          "frac_active": 0.125
        },
        {
          "step": 341,
          "loss": 0.3948765695095062,
          "floss_loss": 0.0005197651917114854,
          "kl_loss": 0.03943568095564842,
          "lyap1_mean": -0.018853656947612762,
          "lyap1_max": -0.003910645376890898,
          "lyap_mean": -0.020799458026885986,
          "volume_mean": -0.020799458026885986,
          "volume_max": -0.01107292901724577,
          "frac_active": 0.0
        },
        {
          "step": 342,
          "loss": 1.8440921306610107,
          "floss_loss": 0.0005191225209273398,
          "kl_loss": 0.18435730040073395,
          "lyap1_mean": -0.021382182836532593,
          "lyap1_max": -0.016486629843711853,
          "lyap_mean": -0.02156381867825985,
          "volume_mean": -0.02156381867825985,
          "volume_max": -0.011245201341807842,
          "frac_active": 0.0
        },
        {
          "step": 343,
          "loss": 0.38572657108306885,
          "floss_loss": 0.0006514264387078583,
          "kl_loss": 0.03850751370191574,
          "lyap1_mean": -0.022897057235240936,
          "lyap1_max": -0.0058757164515554905,
          "lyap_mean": -0.024262763559818268,
          "volume_mean": -0.024262763559818268,
          "volume_max": -0.013044355437159538,
          "frac_active": 0.0
        },
        {
          "step": 344,
          "loss": 0.31072887778282166,
          "floss_loss": 0.0006432038499042392,
          "kl_loss": 0.031008567661046982,
          "lyap1_mean": -0.018692854791879654,
          "lyap1_max": -0.00427578529343009,
          "lyap_mean": -0.02454054355621338,
          "volume_mean": -0.02454054355621338,
          "volume_max": -0.02049390599131584,
          "frac_active": 0.0
        },
        {
          "step": 345,
          "loss": 0.029063722118735313,
          "floss_loss": 0.0003480321611277759,
          "kl_loss": 0.002871569013223052,
          "lyap1_mean": -0.007825111970305443,
          "lyap1_max": 0.014468603767454624,
          "lyap_mean": -0.01586141437292099,
          "volume_mean": -0.01586141437292099,
          "volume_max": -0.008721900172531605,
          "frac_active": 0.0625
        },
        {
          "step": 346,
          "loss": 0.47062426805496216,
          "floss_loss": 0.00044109459850005805,
          "kl_loss": 0.04701831564307213,
          "lyap1_mean": -0.012417714111506939,
          "lyap1_max": -0.0028210727032274008,
          "lyap_mean": -0.019075283780694008,
          "volume_mean": -0.019075283780694008,
          "volume_max": -0.011965829879045486,
          "frac_active": 0.0
        },
        {
          "step": 347,
          "loss": 0.4735448360443115,
          "floss_loss": 0.000411498942412436,
          "kl_loss": 0.04731333255767822,
          "lyap1_mean": -0.0010471194982528687,
          "lyap1_max": 0.03316565975546837,
          "lyap_mean": -0.014156789518892765,
          "volume_mean": -0.014156789518892765,
          "volume_max": -0.007119455374777317,
          "frac_active": 0.125
        },
        {
          "step": 348,
          "loss": 0.09410218894481659,
          "floss_loss": 0.0001764332118909806,
          "kl_loss": 0.009392576292157173,
          "lyap1_mean": 0.008152378723025322,
          "lyap1_max": 0.027752961963415146,
          "lyap_mean": 0.0025760747957974672,
          "volume_mean": 0.0025760747957974672,
          "volume_max": 0.014562823809683323,
          "frac_active": 0.5625
        },
        {
          "step": 349,
          "loss": 0.19365863502025604,
          "floss_loss": 0.0003200627688784152,
          "kl_loss": 0.0193338580429554,
          "lyap1_mean": -0.01040149386972189,
          "lyap1_max": 0.001965330680832267,
          "lyap_mean": -0.014771423302590847,
          "volume_mean": -0.014771423302590847,
          "volume_max": -0.005444006994366646,
          "frac_active": 0.0625
        },
        {
          "step": 350,
          "loss": 0.0008997443947009742,
          "floss_loss": 0.0003173359436914325,
          "kl_loss": 5.824084291816689e-05,
          "lyap1_mean": 0.01172207947820425,
          "lyap1_max": 0.04519243165850639,
          "lyap_mean": -0.004581748507916927,
          "volume_mean": -0.004581748507916927,
          "volume_max": 0.004308505915105343,
          "frac_active": 0.25
        },
        {
          "step": 351,
          "loss": 1.3022854328155518,
          "floss_loss": 0.0002862914407160133,
          "kl_loss": 0.13019990921020508,
          "lyap1_mean": 0.01963244378566742,
          "lyap1_max": 0.040473826229572296,
          "lyap_mean": 0.004166006576269865,
          "volume_mean": 0.004166006576269865,
          "volume_max": 0.01571633666753769,
          "frac_active": 0.625
        },
        {
          "step": 352,
          "loss": 0.350763738155365,
          "floss_loss": 0.0004947956185787916,
          "kl_loss": 0.03502689301967621,
          "lyap1_mean": 0.01378158200532198,
          "lyap1_max": 0.04368428885936737,
          "lyap_mean": -0.006117612589150667,
          "volume_mean": -0.006117612589150667,
          "volume_max": 0.005830458831042051,
          "frac_active": 0.3125
        },
        {
          "step": 353,
          "loss": 4.223708152770996,
          "floss_loss": 0.0007757041603326797,
          "kl_loss": 0.4222932457923889,
          "lyap1_mean": 0.03196033835411072,
          "lyap1_max": 0.04468232020735741,
          "lyap_mean": 0.021579697728157043,
          "volume_mean": 0.021579697728157043,
          "volume_max": 0.026865830644965172,
          "frac_active": 0.875
        },
        {
          "step": 354,
          "loss": 0.13026276230812073,
          "floss_loss": 0.00041964664706029,
          "kl_loss": 0.01298431120812893,
          "lyap1_mean": 0.029537534341216087,
          "lyap1_max": 0.057853586971759796,
          "lyap_mean": 0.010932985693216324,
          "volume_mean": 0.010932985693216324,
          "volume_max": 0.020868675783276558,
          "frac_active": 0.6875
        },
        {
          "step": 355,
          "loss": 2.774689197540283,
          "floss_loss": 0.0003827719483524561,
          "kl_loss": 0.2774306535720825,
          "lyap1_mean": 0.02410571277141571,
          "lyap1_max": 0.03216579556465149,
          "lyap_mean": 0.010713609866797924,
          "volume_mean": 0.010713609866797924,
          "volume_max": 0.017160024493932724,
          "frac_active": 0.6875
        },
        {
          "step": 356,
          "loss": 0.059700388461351395,
          "floss_loss": 0.0002897379454225302,
          "kl_loss": 0.005941065028309822,
          "lyap1_mean": 0.023390423506498337,
          "lyap1_max": 0.03304276242852211,
          "lyap_mean": 0.008005989715456963,
          "volume_mean": 0.008005989715456963,
          "volume_max": 0.011963597498834133,
          "frac_active": 0.6875
        },
        {
          "step": 357,
          "loss": 0.1284935176372528,
          "floss_loss": 0.0005682689370587468,
          "kl_loss": 0.012792523950338364,
          "lyap1_mean": 0.03841286525130272,
          "lyap1_max": 0.04861438274383545,
          "lyap_mean": 0.014783678576350212,
          "volume_mean": 0.014783678576350212,
          "volume_max": 0.030408179387450218,
          "frac_active": 0.75
        },
        {
          "step": 358,
          "loss": 0.19612514972686768,
          "floss_loss": 0.0011169591452926397,
          "kl_loss": 0.019500818103551865,
          "lyap1_mean": 0.04062306135892868,
          "lyap1_max": 0.07697340101003647,
          "lyap_mean": 0.020373579114675522,
          "volume_mean": 0.020373579114675522,
          "volume_max": 0.04452165961265564,
          "frac_active": 0.75
        },
        {
          "step": 359,
          "loss": 2.7775073051452637,
          "floss_loss": 0.0006017142441123724,
          "kl_loss": 0.2776905596256256,
          "lyap1_mean": 0.037009987980127335,
          "lyap1_max": 0.07118253409862518,
          "lyap_mean": 0.012244977056980133,
          "volume_mean": 0.012244977056980133,
          "volume_max": 0.02256469428539276,
          "frac_active": 0.625
        },
        {
          "step": 360,
          "loss": 2.243283748626709,
          "floss_loss": 0.00035358540480956435,
          "kl_loss": 0.22429300844669342,
          "lyap1_mean": 0.02829575538635254,
          "lyap1_max": 0.041994187980890274,
          "lyap_mean": 0.007730516139417887,
          "volume_mean": 0.007730516139417887,
          "volume_max": 0.018637027591466904,
          "frac_active": 0.625
        },
        {
          "step": 361,
          "loss": 0.006403608247637749,
          "floss_loss": 0.00024389493046328425,
          "kl_loss": 0.0006159713375382125,
          "lyap1_mean": 0.008695971220731735,
          "lyap1_max": 0.02621261030435562,
          "lyap_mean": -0.0037406363990157843,
          "volume_mean": -0.0037406363990157843,
          "volume_max": 0.010186339728534222,
          "frac_active": 0.375
        },
        {
          "step": 362,
          "loss": 0.32123103737831116,
          "floss_loss": 0.0001276922266697511,
          "kl_loss": 0.03211033344268799,
          "lyap1_mean": 0.006241004914045334,
          "lyap1_max": 0.02488566003739834,
          "lyap_mean": -0.0031966972164809704,
          "volume_mean": -0.0031966972164809704,
          "volume_max": 0.003113809507340193,
          "frac_active": 0.25
        },
        {
          "step": 363,
          "loss": 2.311053991317749,
          "floss_loss": 0.00016352385864593089,
          "kl_loss": 0.2310890406370163,
          "lyap1_mean": 0.007480146363377571,
          "lyap1_max": 0.02891622669994831,
          "lyap_mean": -0.0015814832877367735,
          "volume_mean": -0.0015814832877367735,
          "volume_max": 0.011605450883507729,
          "frac_active": 0.4375
        },
        {
          "step": 364,
          "loss": 0.10671580582857132,
          "floss_loss": 0.0001447244721930474,
          "kl_loss": 0.010657107457518578,
          "lyap1_mean": 0.0010739339049905539,
          "lyap1_max": 0.012007501907646656,
          "lyap_mean": -0.008075349032878876,
          "volume_mean": -0.008075349032878876,
          "volume_max": -0.0024676115717738867,
          "frac_active": 0.125
        },
        {
          "step": 365,
          "loss": 4.107755184173584,
          "floss_loss": 0.0003865143225993961,
          "kl_loss": 0.4107368290424347,
          "lyap1_mean": -0.0040024761110544205,
          "lyap1_max": 0.0195472352206707,
          "lyap_mean": -0.010367447510361671,
          "volume_mean": -0.010367447510361671,
          "volume_max": 0.007051827851682901,
          "frac_active": 0.375
        },
        {
          "step": 366,
          "loss": 0.011465256102383137,
          "floss_loss": 0.0002517842804081738,
          "kl_loss": 0.0011213471880182624,
          "lyap1_mean": 0.00416651600971818,
          "lyap1_max": 0.020779719576239586,
          "lyap_mean": -0.008192695677280426,
          "volume_mean": -0.008192695677280426,
          "volume_max": 0.0007728142663836479,
          "frac_active": 0.25
        },
        {
          "step": 367,
          "loss": 0.3365292549133301,
          "floss_loss": 0.0002262637426611036,
          "kl_loss": 0.03363030031323433,
          "lyap1_mean": 0.01941431500017643,
          "lyap1_max": 0.03922078758478165,
          "lyap_mean": 0.005287647247314453,
          "volume_mean": 0.005287647247314453,
          "volume_max": 0.018486695364117622,
          "frac_active": 0.6875
        },
        {
          "step": 368,
          "loss": 0.25056976079940796,
          "floss_loss": 0.00021819052926730365,
          "kl_loss": 0.025035157799720764,
          "lyap1_mean": 0.010502291843295097,
          "lyap1_max": 0.024634884670376778,
          "lyap_mean": 0.0022365148179233074,
          "volume_mean": 0.0022365148179233074,
          "volume_max": 0.01861831732094288,
          "frac_active": 0.5
        },
        {
          "step": 369,
          "loss": 0.00691274693235755,
          "floss_loss": 0.0003612343571148813,
          "kl_loss": 0.0006551512633450329,
          "lyap1_mean": -0.009565779939293861,
          "lyap1_max": 0.0151915168389678,
          "lyap_mean": -0.010054689832031727,
          "volume_mean": -0.010054689832031727,
          "volume_max": 0.010633287020027637,
          "frac_active": 0.25
        },
        {
          "step": 370,
          "loss": 0.09234365820884705,
          "floss_loss": 0.00012427655747160316,
          "kl_loss": 0.00922193843871355,
          "lyap1_mean": 0.005621439777314663,
          "lyap1_max": 0.014755379408597946,
          "lyap_mean": -0.0043079061433672905,
          "volume_mean": -0.0043079061433672905,
          "volume_max": 0.0020885393023490906,
          "frac_active": 0.4375
        },
        {
          "step": 371,
          "loss": 0.17748264968395233,
          "floss_loss": 0.00018132112745661288,
          "kl_loss": 0.01773013360798359,
          "lyap1_mean": 0.013626652769744396,
          "lyap1_max": 0.02309083752334118,
          "lyap_mean": -0.004072873387485743,
          "volume_mean": -0.004072873387485743,
          "volume_max": 0.0002960080746561289,
          "frac_active": 0.3125
        },
        {
          "step": 372,
          "loss": 0.08651523292064667,
          "floss_loss": 0.0003181210486218333,
          "kl_loss": 0.008619711734354496,
          "lyap1_mean": 0.016354665160179138,
          "lyap1_max": 0.039686162024736404,
          "lyap_mean": 0.0018759830854833126,
          "volume_mean": 0.0018759830854833126,
          "volume_max": 0.02494579181075096,
          "frac_active": 0.4375
        },
        {
          "step": 373,
          "loss": 0.06699546426534653,
          "floss_loss": 0.00013108516577631235,
          "kl_loss": 0.006686437875032425,
          "lyap1_mean": 0.008572161197662354,
          "lyap1_max": 0.020608866587281227,
          "lyap_mean": 0.001113990438170731,
          "volume_mean": 0.001113990438170731,
          "volume_max": 0.007369921077042818,
          "frac_active": 0.625
        },
        {
          "step": 374,
          "loss": 0.10494966059923172,
          "floss_loss": 0.00040292146150022745,
          "kl_loss": 0.010454674251377583,
          "lyap1_mean": 0.010253450833261013,
          "lyap1_max": 0.04592513665556908,
          "lyap_mean": -0.004084567539393902,
          "volume_mean": -0.004084567539393902,
          "volume_max": 0.022967275232076645,
          "frac_active": 0.3125
        },
        {
          "step": 375,
          "loss": 0.2986485958099365,
          "floss_loss": 0.00018074680701829493,
          "kl_loss": 0.029846785590052605,
          "lyap1_mean": 0.003998792730271816,
          "lyap1_max": 0.029323263093829155,
          "lyap_mean": -0.004535144194960594,
          "volume_mean": -0.004535144194960594,
          "volume_max": 0.004326802212744951,
          "frac_active": 0.375
        },
        {
          "step": 376,
          "loss": 0.2273007184267044,
          "floss_loss": 0.0001615172077436,
          "kl_loss": 0.0227139201015234,
          "lyap1_mean": 0.0020633190870285034,
          "lyap1_max": 0.01086448971182108,
          "lyap_mean": -0.003833569586277008,
          "volume_mean": -0.003833569586277008,
          "volume_max": 0.0013202112168073654,
          "frac_active": 0.3125
        },
        {
          "step": 377,
          "loss": 0.001900451141409576,
          "floss_loss": 0.00039558042772114277,
          "kl_loss": 0.00015048707427922636,
          "lyap1_mean": 0.0007239609258249402,
          "lyap1_max": 0.03349853679537773,
          "lyap_mean": -0.009609011933207512,
          "volume_mean": -0.009609011933207512,
          "volume_max": 0.010869476944208145,
          "frac_active": 0.1875
        },
        {
          "step": 378,
          "loss": 0.410475492477417,
          "floss_loss": 0.00043699360685423017,
          "kl_loss": 0.04100384935736656,
          "lyap1_mean": 0.009394319728016853,
          "lyap1_max": 0.023068277165293694,
          "lyap_mean": -0.006569229532033205,
          "volume_mean": -0.006569229532033205,
          "volume_max": 0.01571340300142765,
          "frac_active": 0.375
        },
        {
          "step": 379,
          "loss": 0.251312255859375,
          "floss_loss": 0.0001848457322921604,
          "kl_loss": 0.025112740695476532,
          "lyap1_mean": 0.01563147082924843,
          "lyap1_max": 0.03534476086497307,
          "lyap_mean": 0.004377373028546572,
          "volume_mean": 0.004377373028546572,
          "volume_max": 0.01551862247288227,
          "frac_active": 0.625
        },
        {
          "step": 380,
          "loss": 4.173399448394775,
          "floss_loss": 0.00021686896798200905,
          "kl_loss": 0.4173182249069214,
          "lyap1_mean": 0.00014164531603455544,
          "lyap1_max": 0.01952224224805832,
          "lyap_mean": -0.008145475760102272,
          "volume_mean": -0.008145475760102272,
          "volume_max": 0.003435207298025489,
          "frac_active": 0.25
        },
        {
          "step": 381,
          "loss": 0.1819305419921875,
          "floss_loss": 0.0002523689763620496,
          "kl_loss": 0.01816781796514988,
          "lyap1_mean": -0.0088539132848382,
          "lyap1_max": 0.008315371349453926,
          "lyap_mean": -0.013660958968102932,
          "volume_mean": -0.013660958968102932,
          "volume_max": -0.00976755004376173,
          "frac_active": 0.0625
        },
        {
          "step": 382,
          "loss": 0.001524616964161396,
          "floss_loss": 0.00047651183558627963,
          "kl_loss": 0.0001048105041263625,
          "lyap1_mean": -0.0036724661476910114,
          "lyap1_max": 0.012649565935134888,
          "lyap_mean": -0.01884176954627037,
          "volume_mean": -0.01884176954627037,
          "volume_max": -0.01284076925367117,
          "frac_active": 0.125
        },
        {
          "step": 383,
          "loss": 3.7488577365875244,
          "floss_loss": 0.0004167869046796113,
          "kl_loss": 0.37484410405158997,
          "lyap1_mean": -0.018323874101042747,
          "lyap1_max": -0.005885383114218712,
          "lyap_mean": -0.018716860562562943,
          "volume_mean": -0.018716860562562943,
          "volume_max": -0.009985409677028656,
          "frac_active": 0.0
        },
        {
          "step": 384,
          "loss": 0.22780926525592804,
          "floss_loss": 0.000351303635397926,
          "kl_loss": 0.02274579554796219,
          "lyap1_mean": -0.013257448561489582,
          "lyap1_max": 0.003911609761416912,
          "lyap_mean": -0.01686543971300125,
          "volume_mean": -0.01686543971300125,
          "volume_max": -0.008298413828015327,
          "frac_active": 0.0625
        },
        {
          "step": 385,
          "loss": 0.01618281751871109,
          "floss_loss": 0.00034016085555776954,
          "kl_loss": 0.001584265730343759,
          "lyap1_mean": -0.014671369455754757,
          "lyap1_max": -0.0023371418938040733,
          "lyap_mean": -0.016626689583063126,
          "volume_mean": -0.016626689583063126,
          "volume_max": -0.006469376385211945,
          "frac_active": 0.0
        },
        {
          "step": 386,
          "loss": 0.2193528562784195,
          "floss_loss": 0.0003843421582132578,
          "kl_loss": 0.0218968503177166,
          "lyap1_mean": -0.017035678029060364,
          "lyap1_max": -0.008744870312511921,
          "lyap_mean": -0.017863577231764793,
          "volume_mean": -0.017863577231764793,
          "volume_max": -0.009363995864987373,
          "frac_active": 0.0625
        },
        {
          "step": 387,
          "loss": 0.8367233276367188,
          "floss_loss": 0.000468031270429492,
          "kl_loss": 0.08362553268671036,
          "lyap1_mean": -0.018501657992601395,
          "lyap1_max": -0.010453800670802593,
          "lyap_mean": -0.020597103983163834,
          "volume_mean": -0.020597103983163834,
          "volume_max": -0.013953188434243202,
          "frac_active": 0.0
        },
        {
          "step": 388,
          "loss": 0.039351820945739746,
          "floss_loss": 0.0004178389790467918,
          "kl_loss": 0.0038933982141315937,
          "lyap1_mean": -0.018668442964553833,
          "lyap1_max": -0.01044992171227932,
          "lyap_mean": -0.01883944496512413,
          "volume_mean": -0.01883944496512413,
          "volume_max": -0.010427745059132576,
          "frac_active": 0.0
        },
        {
          "step": 389,
          "loss": 0.11989939957857132,
          "floss_loss": 0.00016834291454870254,
          "kl_loss": 0.011973105370998383,
          "lyap1_mean": -0.004331226926296949,
          "lyap1_max": 0.0033335108309984207,
          "lyap_mean": -0.010793529450893402,
          "volume_mean": -0.010793529450893402,
          "volume_max": -0.0071039809845387936,
          "frac_active": 0.0625
        },
        {
          "step": 390,
          "loss": 0.13790763914585114,
          "floss_loss": 0.00024831402697600424,
          "kl_loss": 0.013765932992100716,
          "lyap1_mean": 0.0030081402510404587,
          "lyap1_max": 0.029596777632832527,
          "lyap_mean": -0.009386656805872917,
          "volume_mean": -0.009386656805872917,
          "volume_max": -0.005803317297250032,
          "frac_active": 0.125
        },
        {
          "step": 391,
          "loss": 2.949070453643799,
          "floss_loss": 0.00032573420321568847,
          "kl_loss": 0.29487448930740356,
          "lyap1_mean": 0.006881290581077337,
          "lyap1_max": 0.03861311078071594,
          "lyap_mean": -0.0056330058723688126,
          "volume_mean": -0.0056330058723688126,
          "volume_max": 0.008833744563162327,
          "frac_active": 0.25
        },
        {
          "step": 392,
          "loss": 0.3107900023460388,
          "floss_loss": 0.00019182005780749023,
          "kl_loss": 0.03105981834232807,
          "lyap1_mean": 0.01638081856071949,
          "lyap1_max": 0.03248545527458191,
          "lyap_mean": 0.005090177990496159,
          "volume_mean": 0.005090177990496159,
          "volume_max": 0.012440957129001617,
          "frac_active": 0.5625
        },
        {
          "step": 393,
          "loss": 2.4862418174743652,
          "floss_loss": 0.00044354290002956986,
          "kl_loss": 0.24857984483242035,
          "lyap1_mean": 0.024748845025897026,
          "lyap1_max": 0.0421496257185936,
          "lyap_mean": 0.007818164303898811,
          "volume_mean": 0.007818164303898811,
          "volume_max": 0.018714912235736847,
          "frac_active": 0.6875
        },
        {
          "step": 394,
          "loss": 1.0285265445709229,
          "floss_loss": 0.00030570727540180087,
          "kl_loss": 0.10282209515571594,
          "lyap1_mean": 0.018826158717274666,
          "lyap1_max": 0.05057110637426376,
          "lyap_mean": 0.004721520002931356,
          "volume_mean": 0.004721520002931356,
          "volume_max": 0.023108379915356636,
          "frac_active": 0.5
        },
        {
          "step": 395,
          "loss": 0.44239282608032227,
          "floss_loss": 0.00012510066153481603,
          "kl_loss": 0.04422677308320999,
          "lyap1_mean": 0.01232180092483759,
          "lyap1_max": 0.03464611619710922,
          "lyap_mean": 0.0021993620321154594,
          "volume_mean": 0.0021993620321154594,
          "volume_max": 0.0056795417331159115,
          "frac_active": 0.625
        },
        {
          "step": 396,
          "loss": 2.4974920749664307,
          "floss_loss": 0.00033469730988144875,
          "kl_loss": 0.24971573054790497,
          "lyap1_mean": 0.02002716064453125,
          "lyap1_max": 0.041682783514261246,
          "lyap_mean": 0.006440355442464352,
          "volume_mean": 0.006440355442464352,
          "volume_max": 0.02180803380906582,
          "frac_active": 0.5625
        },
        {
          "step": 397,
          "loss": 0.1343526542186737,
          "floss_loss": 0.0003319398092571646,
          "kl_loss": 0.013402070850133896,
          "lyap1_mean": 0.020334839820861816,
          "lyap1_max": 0.05153840407729149,
          "lyap_mean": 0.008370057679712772,
          "volume_mean": 0.008370057679712772,
          "volume_max": 0.024490369483828545,
          "frac_active": 0.6875
        },
        {
          "step": 398,
          "loss": 0.0008299086475744843,
          "floss_loss": 0.00030635189614258707,
          "kl_loss": 5.235567732597701e-05,
          "lyap1_mean": 0.022203881293535233,
          "lyap1_max": 0.049629099667072296,
          "lyap_mean": 0.006583043839782476,
          "volume_mean": 0.006583043839782476,
          "volume_max": 0.023004405200481415,
          "frac_active": 0.6875
        },
        {
          "step": 399,
          "loss": 0.1513918787240982,
          "floss_loss": 0.0002226200158474967,
          "kl_loss": 0.015116925351321697,
          "lyap1_mean": 0.019616151228547096,
          "lyap1_max": 0.03162597119808197,
          "lyap_mean": 0.009871996939182281,
          "volume_mean": 0.009871996939182281,
          "volume_max": 0.013538779690861702,
          "frac_active": 0.875
        },
        {
          "step": 400,
          "loss": 0.304582804441452,
          "floss_loss": 0.0002432279579807073,
          "kl_loss": 0.030433960258960724,
          "lyap1_mean": 0.024864597246050835,
          "lyap1_max": 0.03592010214924812,
          "lyap_mean": 0.008003833703696728,
          "volume_mean": 0.008003833703696728,
          "volume_max": 0.010131633840501308,
          "frac_active": 0.6875
        },
        {
          "step": 401,
          "loss": 0.0037087842356413603,
          "floss_loss": 0.00017768543330021203,
          "kl_loss": 0.0003531098773237318,
          "lyap1_mean": 0.011762616224586964,
          "lyap1_max": 0.04315206781029701,
          "lyap_mean": 0.0035086311399936676,
          "volume_mean": 0.0035086311399936676,
          "volume_max": 0.01668163575232029,
          "frac_active": 0.625
        },
        {
          "step": 402,
          "loss": 0.06232759729027748,
          "floss_loss": 0.000162579191965051,
          "kl_loss": 0.006216501817107201,
          "lyap1_mean": 0.006918829865753651,
          "lyap1_max": 0.02286522649228573,
          "lyap_mean": -0.001505398890003562,
          "volume_mean": -0.001505398890003562,
          "volume_max": 0.01403246633708477,
          "frac_active": 0.4375
        },
        {
          "step": 403,
          "loss": 0.1045682430267334,
          "floss_loss": 8.129583147820085e-05,
          "kl_loss": 0.010448695160448551,
          "lyap1_mean": 0.005037140101194382,
          "lyap1_max": 0.008938828483223915,
          "lyap_mean": -0.0010737846605479717,
          "volume_mean": -0.0010737846605479717,
          "volume_max": 0.009175836108624935,
          "frac_active": 0.4375
        },
        {
          "step": 404,
          "loss": 0.16400471329689026,
          "floss_loss": 0.0001513651222921908,
          "kl_loss": 0.016385335475206375,
          "lyap1_mean": 0.019385069608688354,
          "lyap1_max": 0.028784610331058502,
          "lyap_mean": 0.007934730499982834,
          "volume_mean": 0.007934730499982834,
          "volume_max": 0.01093529537320137,
          "frac_active": 0.75
        },
        {
          "step": 405,
          "loss": 1.9734795093536377,
          "floss_loss": 0.00018673294107429683,
          "kl_loss": 0.19732928276062012,
          "lyap1_mean": 0.012996496632695198,
          "lyap1_max": 0.01947382278740406,
          "lyap_mean": 0.005230402108281851,
          "volume_mean": 0.005230402108281851,
          "volume_max": 0.01145955827087164,
          "frac_active": 0.5
        },
        {
          "step": 406,
          "loss": 0.14854826033115387,
          "floss_loss": 0.00025870028184726834,
          "kl_loss": 0.014828955754637718,
          "lyap1_mean": 0.021936004981398582,
          "lyap1_max": 0.03920731693506241,
          "lyap_mean": 0.008111810311675072,
          "volume_mean": 0.008111810311675072,
          "volume_max": 0.013665654696524143,
          "frac_active": 0.6875
        },
        {
          "step": 407,
          "loss": 0.14826905727386475,
          "floss_loss": 0.00035516396746970713,
          "kl_loss": 0.014791388995945454,
          "lyap1_mean": 0.02256614714860916,
          "lyap1_max": 0.039693545550107956,
          "lyap_mean": 0.008724207989871502,
          "volume_mean": 0.008724207989871502,
          "volume_max": 0.017595777288079262,
          "frac_active": 0.6875
        },
        {
          "step": 408,
          "loss": 0.12804414331912994,
          "floss_loss": 0.00021861010463908315,
          "kl_loss": 0.01278255321085453,
          "lyap1_mean": 0.02295084111392498,
          "lyap1_max": 0.03607936203479767,
          "lyap_mean": 0.006754287518560886,
          "volume_mean": 0.006754287518560886,
          "volume_max": 0.008534172549843788,
          "frac_active": 0.5625
        },
        {
          "step": 409,
          "loss": 5.490972995758057,
          "floss_loss": 0.00017694283451419324,
          "kl_loss": 0.5490795969963074,
          "lyap1_mean": 0.012788075953722,
          "lyap1_max": 0.020685413852334023,
          "lyap_mean": 0.0011927912710234523,
          "volume_mean": 0.0011927912710234523,
          "volume_max": 0.009941261261701584,
          "frac_active": 0.5625
        },
        {
          "step": 410,
          "loss": 0.34674200415611267,
          "floss_loss": 0.00018315203487873077,
          "kl_loss": 0.03465588390827179,
          "lyap1_mean": 0.00787743553519249,
          "lyap1_max": 0.02965225838124752,
          "lyap_mean": -0.003140039276331663,
          "volume_mean": -0.003140039276331663,
          "volume_max": 0.004942396655678749,
          "frac_active": 0.3125
        },
        {
          "step": 411,
          "loss": 1.608351469039917,
          "floss_loss": 0.0002945389423985034,
          "kl_loss": 0.16080568730831146,
          "lyap1_mean": 0.023011386394500732,
          "lyap1_max": 0.04810144752264023,
          "lyap_mean": 0.003004311816766858,
          "volume_mean": 0.003004311816766858,
          "volume_max": 0.00794935692101717,
          "frac_active": 0.5625
        },
        {
          "step": 412,
          "loss": 0.04599814862012863,
          "floss_loss": 0.00044694688403978944,
          "kl_loss": 0.004555120132863522,
          "lyap1_mean": 0.021256905049085617,
          "lyap1_max": 0.04122104495763779,
          "lyap_mean": -0.0018492003437131643,
          "volume_mean": -0.0018492003437131643,
          "volume_max": 0.01705068349838257,
          "frac_active": 0.375
        },
        {
          "step": 413,
          "loss": 0.4093186855316162,
          "floss_loss": 0.0002416349743725732,
          "kl_loss": 0.04090770334005356,
          "lyap1_mean": 0.007907917723059654,
          "lyap1_max": 0.03169733285903931,
          "lyap_mean": -0.004662021063268185,
          "volume_mean": -0.004662021063268185,
          "volume_max": 0.0007740700384601951,
          "frac_active": 0.375
        },
        {
          "step": 414,
          "loss": 0.0006245963741093874,
          "floss_loss": 0.00023437978234142065,
          "kl_loss": 3.902165553881787e-05,
          "lyap1_mean": 0.01694880798459053,
          "lyap1_max": 0.023381846025586128,
          "lyap_mean": -0.0020996967796236277,
          "volume_mean": -0.0020996967796236277,
          "volume_max": 0.005020178854465485,
          "frac_active": 0.375
        },
        {
          "step": 415,
          "loss": 2.006249189376831,
          "floss_loss": 0.0003522164770402014,
          "kl_loss": 0.20058971643447876,
          "lyap1_mean": -0.012374717742204666,
          "lyap1_max": -0.00195515644736588,
          "lyap_mean": -0.015279831364750862,
          "volume_mean": -0.015279831364750862,
          "volume_max": -0.0009881656151264906,
          "frac_active": 0.125
        },
        {
          "step": 416,
          "loss": 0.19835104048252106,
          "floss_loss": 0.0006148091051727533,
          "kl_loss": 0.01977362297475338,
          "lyap1_mean": 0.0028130761347711086,
          "lyap1_max": 0.02956344187259674,
          "lyap_mean": -0.017594626173377037,
          "volume_mean": -0.017594626173377037,
          "volume_max": -0.007635454647243023,
          "frac_active": 0.125
        },
        {
          "step": 417,
          "loss": 0.0017010371666401625,
          "floss_loss": 0.00027987657813355327,
          "kl_loss": 0.00014211605594027787,
          "lyap1_mean": -0.0018004290759563446,
          "lyap1_max": 0.012093565426766872,
          "lyap_mean": -0.012070204131305218,
          "volume_mean": -0.012070204131305218,
          "volume_max": -0.003243480110540986,
          "frac_active": 0.1875
        },
        {
          "step": 418,
          "loss": 0.02658405713737011,
          "floss_loss": 0.0003425085451453924,
          "kl_loss": 0.0026241547893732786,
          "lyap1_mean": -0.01120699942111969,
          "lyap1_max": -0.0026569319888949394,
          "lyap_mean": -0.016817748546600342,
          "volume_mean": -0.016817748546600342,
          "volume_max": -0.01254272274672985,
          "frac_active": 0.0
        },
        {
          "step": 419,
          "loss": 0.06912841647863388,
          "floss_loss": 0.00040192832238972187,
          "kl_loss": 0.00687264883890748,
          "lyap1_mean": -0.010880416259169579,
          "lyap1_max": -0.0037646659184247255,
          "lyap_mean": -0.017247971147298813,
          "volume_mean": -0.017247971147298813,
          "volume_max": -0.010384958237409592,
          "frac_active": 0.0
        },
        {
          "step": 420,
          "loss": 0.04143635183572769,
          "floss_loss": 0.00026506814174354076,
          "kl_loss": 0.004117128439247608,
          "lyap1_mean": -0.011243616230785847,
          "lyap1_max": 0.0007869700784794986,
          "lyap_mean": -0.013950379565358162,
          "volume_mean": -0.013950379565358162,
          "volume_max": -0.004703338257968426,
          "frac_active": 0.0625
        },
        {
          "step": 421,
          "loss": 1.7287763357162476,
          "floss_loss": 0.00025759750860743225,
          "kl_loss": 0.17285187542438507,
          "lyap1_mean": -0.01060720719397068,
          "lyap1_max": 0.0007211208576336503,
          "lyap_mean": -0.014844452030956745,
          "volume_mean": -0.014844452030956745,
          "volume_max": -0.010394776239991188,
          "frac_active": 0.0625
        },
        {
          "step": 422,
          "loss": 0.12334036082029343,
          "floss_loss": 0.0003888806968461722,
          "kl_loss": 0.012295147404074669,
          "lyap1_mean": -0.016924358904361725,
          "lyap1_max": -0.014186588115990162,
          "lyap_mean": -0.018725087866187096,
          "volume_mean": -0.018725087866187096,
          "volume_max": -0.01343313418328762,
          "frac_active": 0.0
        },
        {
          "step": 423,
          "loss": 0.16360171139240265,
          "floss_loss": 0.00027573617990128696,
          "kl_loss": 0.0163325984030962,
          "lyap1_mean": -0.00805782712996006,
          "lyap1_max": 0.003003076883032918,
          "lyap_mean": -0.014605145901441574,
          "volume_mean": -0.014605145901441574,
          "volume_max": -0.009155422449111938,
          "frac_active": 0.0625
        },
        {
          "step": 424,
          "loss": 0.19745860993862152,
          "floss_loss": 0.00039608648512512445,
          "kl_loss": 0.019706252962350845,
          "lyap1_mean": 0.004258694127202034,
          "lyap1_max": 0.03756432980298996,
          "lyap_mean": -0.007808488793671131,
          "volume_mean": -0.007808488793671131,
          "volume_max": 0.01226163562387228,
          "frac_active": 0.4375
        },
        {
          "step": 425,
          "loss": 0.007018047384917736,
          "floss_loss": 0.00042823515832424164,
          "kl_loss": 0.0006589812110178173,
          "lyap1_mean": -0.011507855728268623,
          "lyap1_max": 0.0005644730990752578,
          "lyap_mean": -0.017910145223140717,
          "volume_mean": -0.017910145223140717,
          "volume_max": -0.00850619561970234,
          "frac_active": 0.0625
        },
        {
          "step": 426,
          "loss": 0.2798592150211334,
          "floss_loss": 0.0003371899947524071,
          "kl_loss": 0.027952203527092934,
          "lyap1_mean": -0.015049871057271957,
          "lyap1_max": -0.008241110481321812,
          "lyap_mean": -0.017155788838863373,
          "volume_mean": -0.017155788838863373,
          "volume_max": -0.011870804242789745,
          "frac_active": 0.0
        },
        {
          "step": 427,
          "loss": 0.288608580827713,
          "floss_loss": 0.0002544023736845702,
          "kl_loss": 0.028835417702794075,
          "lyap1_mean": -0.002395117422565818,
          "lyap1_max": 0.010348664596676826,
          "lyap_mean": -0.010682552121579647,
          "volume_mean": -0.010682552121579647,
          "volume_max": 0.001368035445921123,
          "frac_active": 0.1875
        },
        {
          "step": 428,
          "loss": 0.08738254010677338,
          "floss_loss": 0.00016041542403399944,
          "kl_loss": 0.008722212165594101,
          "lyap1_mean": -0.003443096298724413,
          "lyap1_max": 0.009155123494565487,
          "lyap_mean": -0.005862769205123186,
          "volume_mean": -0.005862769205123186,
          "volume_max": 0.003370815422385931,
          "frac_active": 0.3125
        },
        {
          "step": 429,
          "loss": 0.29608863592147827,
          "floss_loss": 0.00039115542313084006,
          "kl_loss": 0.029569748789072037,
          "lyap1_mean": -0.003818528028205037,
          "lyap1_max": 0.018726561218500137,
          "lyap_mean": -0.011283925734460354,
          "volume_mean": -0.011283925734460354,
          "volume_max": -0.0028793434612452984,
          "frac_active": 0.1875
        },
        {
          "step": 430,
          "loss": 0.0005219905870035291,
          "floss_loss": 0.0002784104144666344,
          "kl_loss": 2.4358014343306422e-05,
          "lyap1_mean": 0.006669499911367893,
          "lyap1_max": 0.03280552849173546,
          "lyap_mean": -0.006091741379350424,
          "volume_mean": -0.006091741379350424,
          "volume_max": 0.007815150544047356,
          "frac_active": 0.1875
        },
        {
          "step": 431,
          "loss": 0.17454436421394348,
          "floss_loss": 0.00019837339641526341,
          "kl_loss": 0.01743459887802601,
          "lyap1_mean": 0.01156003400683403,
          "lyap1_max": 0.030219709500670433,
          "lyap_mean": 0.0027522831223905087,
          "volume_mean": 0.0027522831223905087,
          "volume_max": 0.014517605304718018,
          "frac_active": 0.625
        },
        {
          "step": 432,
          "loss": 0.31799954175949097,
          "floss_loss": 0.00045730266720056534,
          "kl_loss": 0.03175422176718712,
          "lyap1_mean": 0.025648318231105804,
          "lyap1_max": 0.041734080761671066,
          "lyap_mean": 0.0029737064614892006,
          "volume_mean": 0.0029737064614892006,
          "volume_max": 0.008622407913208008,
          "frac_active": 0.4375
        },
        {
          "step": 433,
          "loss": 6.631036281585693,
          "floss_loss": 0.00051017728401348,
          "kl_loss": 0.6630526185035706,
          "lyap1_mean": 0.030423268675804138,
          "lyap1_max": 0.06687094271183014,
          "lyap_mean": 0.006059431005269289,
          "volume_mean": 0.006059431005269289,
          "volume_max": 0.016862891614437103,
          "frac_active": 0.5625
        },
        {
          "step": 434,
          "loss": 0.1778995841741562,
          "floss_loss": 0.0001780591846909374,
          "kl_loss": 0.01777215301990509,
          "lyap1_mean": 0.009234627708792686,
          "lyap1_max": 0.015469539910554886,
          "lyap_mean": -0.006079095881432295,
          "volume_mean": -0.006079095881432295,
          "volume_max": -0.0005796144250780344,
          "frac_active": 0.3125
        },
        {
          "step": 435,
          "loss": 0.046266328543424606,
          "floss_loss": 0.00015542781329713762,
          "kl_loss": 0.004611090291291475,
          "lyap1_mean": 0.0020302406046539545,
          "lyap1_max": 0.02160067670047283,
          "lyap_mean": -0.004927709233015776,
          "volume_mean": -0.004927709233015776,
          "volume_max": 0.008601506240665913,
          "frac_active": 0.3125
        },
        {
          "step": 436,
          "loss": 5.598674774169922,
          "floss_loss": 0.00040721328696236014,
          "kl_loss": 0.5598267316818237,
          "lyap1_mean": 0.03154309093952179,
          "lyap1_max": 0.036068495362997055,
          "lyap_mean": 0.010797692462801933,
          "volume_mean": 0.010797692462801933,
          "volume_max": 0.024229085072875023,
          "frac_active": 0.6875
        },
        {
          "step": 437,
          "loss": 0.11654610186815262,
          "floss_loss": 0.00012534564302768558,
          "kl_loss": 0.011642075143754482,
          "lyap1_mean": 0.00418679928407073,
          "lyap1_max": 0.024566708132624626,
          "lyap_mean": -0.0017573335207998753,
          "volume_mean": -0.0017573335207998753,
          "volume_max": 0.009879214689135551,
          "frac_active": 0.1875
        },
        {
          "step": 438,
          "loss": 0.14844292402267456,
          "floss_loss": 0.0004955328768119216,
          "kl_loss": 0.014794738031923771,
          "lyap1_mean": 0.026653505861759186,
          "lyap1_max": 0.05579404532909393,
          "lyap_mean": 0.0022236432414501905,
          "volume_mean": 0.0022236432414501905,
          "volume_max": 0.014695015735924244,
          "frac_active": 0.375
        },
        {
          "step": 439,
          "loss": 0.2755884528160095,
          "floss_loss": 0.0002560397260822356,
          "kl_loss": 0.027533242478966713,
          "lyap1_mean": 0.020030289888381958,
          "lyap1_max": 0.031869933009147644,
          "lyap_mean": 0.005643236916512251,
          "volume_mean": 0.005643236916512251,
          "volume_max": 0.011807075701653957,
          "frac_active": 0.625
        },
        {
          "step": 440,
          "loss": 7.625307083129883,
          "floss_loss": 0.00041659767157398164,
          "kl_loss": 0.7624890208244324,
          "lyap1_mean": -0.0031032508704811335,
          "lyap1_max": 0.01730632409453392,
          "lyap_mean": -0.011743362993001938,
          "volume_mean": -0.011743362993001938,
          "volume_max": 0.004429303575307131,
          "frac_active": 0.3125
        },
        {
          "step": 441,
          "loss": 0.015922924503684044,
          "floss_loss": 0.00022106118558440357,
          "kl_loss": 0.0015701863449066877,
          "lyap1_mean": 0.014452273026108742,
          "lyap1_max": 0.03422962501645088,
          "lyap_mean": 0.0021103662438690662,
          "volume_mean": 0.0021103662438690662,
          "volume_max": 0.007188480347394943,
          "frac_active": 0.5625
        },
        {
          "step": 442,
          "loss": 0.5912798047065735,
          "floss_loss": 0.00015167376841418445,
          "kl_loss": 0.05911280959844589,
          "lyap1_mean": -0.006924658082425594,
          "lyap1_max": 0.00350065971724689,
          "lyap_mean": -0.009485181421041489,
          "volume_mean": -0.009485181421041489,
          "volume_max": -0.00430419621989131,
          "frac_active": 0.1875
        },
        {
          "step": 443,
          "loss": 0.3316071927547455,
          "floss_loss": 0.0002664263593032956,
          "kl_loss": 0.03313407674431801,
          "lyap1_mean": 0.0038076662458479404,
          "lyap1_max": 0.027245812118053436,
          "lyap_mean": -0.0027060082647949457,
          "volume_mean": -0.0027060082647949457,
          "volume_max": 0.018240617588162422,
          "frac_active": 0.3125
        },
        {
          "step": 444,
          "loss": 0.0941464751958847,
          "floss_loss": 0.00022362064919434488,
          "kl_loss": 0.009392285719513893,
          "lyap1_mean": 0.006126056890934706,
          "lyap1_max": 0.04027436301112175,
          "lyap_mean": -0.0032883896492421627,
          "volume_mean": -0.0032883896492421627,
          "volume_max": 0.014920353889465332,
          "frac_active": 0.375
        },
        {
          "step": 445,
          "loss": 3.013890266418457,
          "floss_loss": 0.00020020872761961073,
          "kl_loss": 0.3013690114021301,
          "lyap1_mean": 0.0033146061468869448,
          "lyap1_max": 0.021829938516020775,
          "lyap_mean": -0.005342117510735989,
          "volume_mean": -0.005342117510735989,
          "volume_max": 0.0069806016981601715,
          "frac_active": 0.1875
        },
        {
          "step": 446,
          "loss": 2.044933557510376,
          "floss_loss": 0.0001291546504944563,
          "kl_loss": 0.20448043942451477,
          "lyap1_mean": 0.006609746254980564,
          "lyap1_max": 0.032680679112672806,
          "lyap_mean": -0.0014215223491191864,
          "volume_mean": -0.0014215223491191864,
          "volume_max": 0.010077185928821564,
          "frac_active": 0.3125
        },
        {
          "step": 447,
          "loss": 0.2800356447696686,
          "floss_loss": 0.0002592329983599484,
          "kl_loss": 0.02797764353454113,
          "lyap1_mean": -0.008836900815367699,
          "lyap1_max": 0.0008026588475331664,
          "lyap_mean": -0.013551452197134495,
          "volume_mean": -0.013551452197134495,
          "volume_max": -0.009285884909331799,
          "frac_active": 0.0625
        },
        {
          "step": 448,
          "loss": 0.548594057559967,
          "floss_loss": 8.149944187607616e-05,
          "kl_loss": 0.05485126003623009,
          "lyap1_mean": 0.005459641106426716,
          "lyap1_max": 0.01926698163151741,
          "lyap_mean": 0.003226247848942876,
          "volume_mean": 0.003226247848942876,
          "volume_max": 0.010424260050058365,
          "frac_active": 0.5
        },
        {
          "step": 449,
          "loss": 0.0032817362807691097,
          "floss_loss": 0.00011941877892240882,
          "kl_loss": 0.000316231744363904,
          "lyap1_mean": -0.00266454741358757,
          "lyap1_max": 0.00503253610804677,
          "lyap_mean": -0.00839069951325655,
          "volume_mean": -0.00839069951325655,
          "volume_max": -0.004515983629971743,
          "frac_active": 0.125
        },
        {
          "step": 450,
          "loss": 0.09717925637960434,
          "floss_loss": 0.00019730161875486374,
          "kl_loss": 0.00969819538295269,
          "lyap1_mean": 0.004405771382153034,
          "lyap1_max": 0.02535185031592846,
          "lyap_mean": -0.004929955117404461,
          "volume_mean": -0.004929955117404461,
          "volume_max": 0.0049377139657735825,
          "frac_active": 0.375
        },
        {
          "step": 451,
          "loss": 0.18485446274280548,
          "floss_loss": 0.00026053417241200805,
          "kl_loss": 0.018459392711520195,
          "lyap1_mean": 0.0005336068570613861,
          "lyap1_max": 0.0335516557097435,
          "lyap_mean": -0.0069068619050085545,
          "volume_mean": -0.0069068619050085545,
          "volume_max": 0.01320517249405384,
          "frac_active": 0.1875
        },
        {
          "step": 452,
          "loss": 0.21898338198661804,
          "floss_loss": 0.0002928309259004891,
          "kl_loss": 0.021869054064154625,
          "lyap1_mean": 0.02302306890487671,
          "lyap1_max": 0.04744051396846771,
          "lyap_mean": 0.004313600715249777,
          "volume_mean": 0.004313600715249777,
          "volume_max": 0.02329232543706894,
          "frac_active": 0.5
        },
        {
          "step": 453,
          "loss": 0.0903065875172615,
          "floss_loss": 0.00026312348200008273,
          "kl_loss": 0.00900434609502554,
          "lyap1_mean": 0.024542883038520813,
          "lyap1_max": 0.03287076577544212,
          "lyap_mean": 0.003097006818279624,
          "volume_mean": 0.003097006818279624,
          "volume_max": 0.0099060432985425,
          "frac_active": 0.5625
        },
        {
          "step": 454,
          "loss": 0.16274671256542206,
          "floss_loss": 0.00024903041776269674,
          "kl_loss": 0.016249768435955048,
          "lyap1_mean": 0.009492434561252594,
          "lyap1_max": 0.04366840794682503,
          "lyap_mean": -0.005035337992012501,
          "volume_mean": -0.005035337992012501,
          "volume_max": 0.005686847493052483,
          "frac_active": 0.375
        },
        {
          "step": 455,
          "loss": 7.154088020324707,
          "floss_loss": 0.0003003091551363468,
          "kl_loss": 0.7153787612915039,
          "lyap1_mean": -0.011576008051633835,
          "lyap1_max": 0.009161606431007385,
          "lyap_mean": -0.01356545276939869,
          "volume_mean": -0.01356545276939869,
          "volume_max": -0.0012112872209399939,
          "frac_active": 0.125
        },
        {
          "step": 456,
          "loss": 3.2554564476013184,
          "floss_loss": 0.00013567559653893113,
          "kl_loss": 0.32553207874298096,
          "lyap1_mean": 0.006284235045313835,
          "lyap1_max": 0.024535393342375755,
          "lyap_mean": -0.0020474805496633053,
          "volume_mean": -0.0020474805496633053,
          "volume_max": 0.009601269848644733,
          "frac_active": 0.5
        },
        {
          "step": 457,
          "loss": 0.003032988402992487,
          "floss_loss": 0.00024189951363950968,
          "kl_loss": 0.00027910887729376554,
          "lyap1_mean": -0.007842651568353176,
          "lyap1_max": 0.00955889094620943,
          "lyap_mean": -0.01279336865991354,
          "volume_mean": -0.01279336865991354,
          "volume_max": -0.0021080903243273497,
          "frac_active": 0.125
        },
        {
          "step": 458,
          "loss": 0.2694922983646393,
          "floss_loss": 0.0001870507257990539,
          "kl_loss": 0.02693052589893341,
          "lyap1_mean": 0.01811564341187477,
          "lyap1_max": 0.03914259746670723,
          "lyap_mean": 0.004197076894342899,
          "volume_mean": 0.004197076894342899,
          "volume_max": 0.014559157192707062,
          "frac_active": 0.625
        },
        {
          "step": 459,
          "loss": 0.3309269845485687,
          "floss_loss": 0.00048661595792509615,
          "kl_loss": 0.033044036477804184,
          "lyap1_mean": 0.028835352510213852,
          "lyap1_max": 0.06474945694208145,
          "lyap_mean": 0.003547247499227524,
          "volume_mean": 0.003547247499227524,
          "volume_max": 0.009724777191877365,
          "frac_active": 0.4375
        },
        {
          "step": 460,
          "loss": 0.12465006858110428,
          "floss_loss": 0.00023042489192448556,
          "kl_loss": 0.012441964820027351,
          "lyap1_mean": 0.01602761819958687,
          "lyap1_max": 0.04042739421129227,
          "lyap_mean": -0.0008120986167341471,
          "volume_mean": -0.0008120986167341471,
          "volume_max": 0.014995187520980835,
          "frac_active": 0.3125
        },
        {
          "step": 461,
          "loss": 0.18204821646213531,
          "floss_loss": 0.00023899564985185862,
          "kl_loss": 0.01818092167377472,
          "lyap1_mean": 0.004377148579806089,
          "lyap1_max": 0.020327290520071983,
          "lyap_mean": -0.00717917550355196,
          "volume_mean": -0.00717917550355196,
          "volume_max": 0.006204065401107073,
          "frac_active": 0.25
        },
        {
          "step": 462,
          "loss": 0.015433198772370815,
          "floss_loss": 0.000304835761198774,
          "kl_loss": 0.0015128363156691194,
          "lyap1_mean": -0.0118259247392416,
          "lyap1_max": 0.00193532626144588,
          "lyap_mean": -0.016021117568016052,
          "volume_mean": -0.016021117568016052,
          "volume_max": -0.00824922975152731,
          "frac_active": 0.0625
        },
        {
          "step": 463,
          "loss": 0.14659187197685242,
          "floss_loss": 0.00010650843614712358,
          "kl_loss": 0.01464853622019291,
          "lyap1_mean": 0.0010178294032812119,
          "lyap1_max": 0.01798098348081112,
          "lyap_mean": -0.0043690442107617855,
          "volume_mean": -0.0043690442107617855,
          "volume_max": 0.004001412075012922,
          "frac_active": 0.3125
        },
        {
          "step": 464,
          "loss": 0.134222611784935,
          "floss_loss": 0.000278928637271747,
          "kl_loss": 0.013394367881119251,
          "lyap1_mean": 0.007374526932835579,
          "lyap1_max": 0.046086400747299194,
          "lyap_mean": -0.00667647086083889,
          "volume_mean": -0.00667647086083889,
          "volume_max": -0.0028122621588408947,
          "frac_active": 0.1875
        },
        {
          "step": 465,
          "loss": 0.0013699643313884735,
          "floss_loss": 0.0001584815763635561,
          "kl_loss": 0.00012114828132325783,
          "lyap1_mean": 0.005749679170548916,
          "lyap1_max": 0.02608412317931652,
          "lyap_mean": -0.005012365058064461,
          "volume_mean": -0.005012365058064461,
          "volume_max": 0.0086373221129179,
          "frac_active": 0.375
        },
        {
          "step": 466,
          "loss": 0.04217716306447983,
          "floss_loss": 0.00023760012118145823,
          "kl_loss": 0.00419395649805665,
          "lyap1_mean": 0.018164295703172684,
          "lyap1_max": 0.024777589365839958,
          "lyap_mean": 0.00893261656165123,
          "volume_mean": 0.00893261656165123,
          "volume_max": 0.013065170496702194,
          "frac_active": 0.75
        },
        {
          "step": 467,
          "loss": 0.04565255343914032,
          "floss_loss": 0.0009828803595155478,
          "kl_loss": 0.004466967191547155,
          "lyap1_mean": 0.03150225803256035,
          "lyap1_max": 0.06432659178972244,
          "lyap_mean": 0.010621626861393452,
          "volume_mean": 0.010621626861393452,
          "volume_max": 0.029961470514535904,
          "frac_active": 0.6875
        },
        {
          "step": 468,
          "loss": 0.09435796737670898,
          "floss_loss": 0.0001376595173496753,
          "kl_loss": 0.009422031231224537,
          "lyap1_mean": 0.013086346909403801,
          "lyap1_max": 0.021006500348448753,
          "lyap_mean": -0.0015185553347691894,
          "volume_mean": -0.0015185553347691894,
          "volume_max": 0.0030926200561225414,
          "frac_active": 0.375
        },
        {
          "step": 469,
          "loss": 0.10647498816251755,
          "floss_loss": 0.0006925425841473043,
          "kl_loss": 0.010578244924545288,
          "lyap1_mean": 0.03286576271057129,
          "lyap1_max": 0.06548843532800674,
          "lyap_mean": 0.016331559047102928,
          "volume_mean": 0.016331559047102928,
          "volume_max": 0.03792036324739456,
          "frac_active": 0.625
        },
        {
          "step": 470,
          "loss": 0.10094868391752243,
          "floss_loss": 0.0008765991660766304,
          "kl_loss": 0.010007208213210106,
          "lyap1_mean": 0.0433272123336792,
          "lyap1_max": 0.07264625281095505,
          "lyap_mean": 0.021201975643634796,
          "volume_mean": 0.021201975643634796,
          "volume_max": 0.024981442838907242,
          "frac_active": 0.875
        },
        {
          "step": 471,
          "loss": 0.3019225299358368,
          "floss_loss": 0.00045724588562734425,
          "kl_loss": 0.030146528035402298,
          "lyap1_mean": 0.006142581347376108,
          "lyap1_max": 0.05021161958575249,
          "lyap_mean": -0.006082591135054827,
          "volume_mean": -0.006082591135054827,
          "volume_max": 0.01217847503721714,
          "frac_active": 0.375
        },
        {
          "step": 472,
          "loss": 1.8460142612457275,
          "floss_loss": 0.00044276457629166543,
          "kl_loss": 0.18455715477466583,
          "lyap1_mean": 0.014049670659005642,
          "lyap1_max": 0.04917874187231064,
          "lyap_mean": 0.005266012158244848,
          "volume_mean": 0.005266012158244848,
          "volume_max": 0.03550282120704651,
          "frac_active": 0.4375
        },
        {
          "step": 473,
          "loss": 0.0010222315322607756,
          "floss_loss": 0.00030048866756260395,
          "kl_loss": 7.217428355943412e-05,
          "lyap1_mean": -0.013996727764606476,
          "lyap1_max": -0.002052157185971737,
          "lyap_mean": -0.013692967593669891,
          "volume_mean": -0.013692967593669891,
          "volume_max": -0.005625738296657801,
          "frac_active": 0.0625
        },
        {
          "step": 474,
          "loss": 10.572900772094727,
          "floss_loss": 0.0009062147000804543,
          "kl_loss": 1.057199478149414,
          "lyap1_mean": 0.04616883397102356,
          "lyap1_max": 0.06579282134771347,
          "lyap_mean": 0.02066604420542717,
          "volume_mean": 0.02066604420542717,
          "volume_max": 0.03266642242670059,
          "frac_active": 0.75
        },
        {
          "step": 475,
          "loss": 7.680459022521973,
          "floss_loss": 0.0003645351971499622,
          "kl_loss": 0.7680094838142395,
          "lyap1_mean": 0.021786464378237724,
          "lyap1_max": 0.036198243498802185,
          "lyap_mean": 0.0025798403657972813,
          "volume_mean": 0.0025798403657972813,
          "volume_max": 0.01695333793759346,
          "frac_active": 0.4375
        },
        {
          "step": 476,
          "loss": 0.13005197048187256,
          "floss_loss": 0.00032397828181274235,
          "kl_loss": 0.012972798198461533,
          "lyap1_mean": 0.015700433403253555,
          "lyap1_max": 0.047554340213537216,
          "lyap_mean": 0.003961380571126938,
          "volume_mean": 0.003961380571126938,
          "volume_max": 0.014864210970699787,
          "frac_active": 0.5625
        },
        {
          "step": 477,
          "loss": 0.13157349824905396,
          "floss_loss": 0.0006263562827371061,
          "kl_loss": 0.013094713911414146,
          "lyap1_mean": 0.027286771684885025,
          "lyap1_max": 0.0774744525551796,
          "lyap_mean": 0.008444019593298435,
          "volume_mean": 0.008444019593298435,
          "volume_max": 0.03407767415046692,
          "frac_active": 0.5625
        },
        {
          "step": 478,
          "loss": 0.001478284946642816,
          "floss_loss": 0.00035947535070590675,
          "kl_loss": 0.00011188095231773332,
          "lyap1_mean": -0.0012453612871468067,
          "lyap1_max": 0.017077775672078133,
          "lyap_mean": -0.011203324422240257,
          "volume_mean": -0.011203324422240257,
          "volume_max": 0.007636386435478926,
          "frac_active": 0.25
        },
        {
          "step": 479,
          "loss": 0.17945484817028046,
          "floss_loss": 0.00013771744852419943,
          "kl_loss": 0.017931712791323662,
          "lyap1_mean": 0.01331391278654337,
          "lyap1_max": 0.022120703011751175,
          "lyap_mean": 0.0015256785554811358,
          "volume_mean": 0.0015256785554811358,
          "volume_max": 0.004914560355246067,
          "frac_active": 0.625
        },
        {
          "step": 480,
          "loss": 0.2744278907775879,
          "floss_loss": 0.0001609905157238245,
          "kl_loss": 0.027426689863204956,
          "lyap1_mean": 0.006569677963852882,
          "lyap1_max": 0.028106190264225006,
          "lyap_mean": -0.004540889989584684,
          "volume_mean": -0.004540889989584684,
          "volume_max": -0.00030660349875688553,
          "frac_active": 0.375
        },
        {
          "step": 481,
          "loss": 0.008633749559521675,
          "floss_loss": 0.0002736789465416223,
          "kl_loss": 0.0008360070060007274,
          "lyap1_mean": 0.003541299607604742,
          "lyap1_max": 0.04195054620504379,
          "lyap_mean": -0.0051455567590892315,
          "volume_mean": -0.0051455567590892315,
          "volume_max": 0.01531282439827919,
          "frac_active": 0.1875
        },
        {
          "step": 482,
          "loss": 0.08375050127506256,
          "floss_loss": 0.00013204112474340945,
          "kl_loss": 0.008361846208572388,
          "lyap1_mean": 0.005936242640018463,
          "lyap1_max": 0.013992298394441605,
          "lyap_mean": 0.0007257247343659401,
          "volume_mean": 0.0007257247343659401,
          "volume_max": 0.010056124068796635,
          "frac_active": 0.5625
        },
        {
          "step": 483,
          "loss": 0.13461461663246155,
          "floss_loss": 0.0002621430321596563,
          "kl_loss": 0.013435247354209423,
          "lyap1_mean": 0.01860469952225685,
          "lyap1_max": 0.037830375134944916,
          "lyap_mean": 0.004769923165440559,
          "volume_mean": 0.004769923165440559,
          "volume_max": 0.016945889219641685,
          "frac_active": 0.5
        },
        {
          "step": 484,
          "loss": 0.10441742837429047,
          "floss_loss": 7.018206815700978e-05,
          "kl_loss": 0.010434724390506744,
          "lyap1_mean": 0.007031090557575226,
          "lyap1_max": 0.018932823091745377,
          "lyap_mean": -0.0017442663665860891,
          "volume_mean": -0.0017442663665860891,
          "volume_max": 0.0006428873166441917,
          "frac_active": 0.375
        },
        {
          "step": 485,
          "loss": 0.08702456951141357,
          "floss_loss": 0.00017875045887194574,
          "kl_loss": 0.008684582076966763,
          "lyap1_mean": -0.0009322011610493064,
          "lyap1_max": 0.014406862668693066,
          "lyap_mean": -0.004145247861742973,
          "volume_mean": -0.004145247861742973,
          "volume_max": 0.00907126534730196,
          "frac_active": 0.3125
        },
        {
          "step": 486,
          "loss": 2.072516441345215,
          "floss_loss": 0.0002121582074323669,
          "kl_loss": 0.20723041892051697,
          "lyap1_mean": 0.019547680392861366,
          "lyap1_max": 0.04150954261422157,
          "lyap_mean": 0.0077478839084506035,
          "volume_mean": 0.0077478839084506035,
          "volume_max": 0.014482380822300911,
          "frac_active": 0.75
        },
        {
          "step": 487,
          "loss": 0.2229338437318802,
          "floss_loss": 0.00019852560944855213,
          "kl_loss": 0.022273531183600426,
          "lyap1_mean": 0.0070130000822246075,
          "lyap1_max": 0.01941533572971821,
          "lyap_mean": 0.0008676403667777777,
          "volume_mean": 0.0008676403667777777,
          "volume_max": 0.006984059698879719,
          "frac_active": 0.5
        },
        {
          "step": 488,
          "loss": 0.10450103133916855,
          "floss_loss": 0.0003515307907946408,
          "kl_loss": 0.010414949618279934,
          "lyap1_mean": 0.01904378831386566,
          "lyap1_max": 0.039676520973443985,
          "lyap_mean": 0.005038704723119736,
          "volume_mean": 0.005038704723119736,
          "volume_max": 0.015887603163719177,
          "frac_active": 0.4375
        },
        {
          "step": 489,
          "loss": 0.0012132017873227596,
          "floss_loss": 0.0002851790050044656,
          "kl_loss": 9.280228550778702e-05,
          "lyap1_mean": 0.0075697144493460655,
          "lyap1_max": 0.026168886572122574,
          "lyap_mean": -0.0016826777718961239,
          "volume_mean": -0.0016826777718961239,
          "volume_max": 0.014040417969226837,
          "frac_active": 0.375
        },
        {
          "step": 490,
          "loss": 0.26879799365997314,
          "floss_loss": 0.0002495533262845129,
          "kl_loss": 0.02685484290122986,
          "lyap1_mean": 0.014860117807984352,
          "lyap1_max": 0.03648652881383896,
          "lyap_mean": -0.00010229309555143118,
          "volume_mean": -0.00010229309555143118,
          "volume_max": 0.013493170961737633,
          "frac_active": 0.4375
        },
        {
          "step": 491,
          "loss": 0.26455262303352356,
          "floss_loss": 0.0005281970370560884,
          "kl_loss": 0.026402443647384644,
          "lyap1_mean": 0.015109483152627945,
          "lyap1_max": 0.05449560284614563,
          "lyap_mean": -0.003382935654371977,
          "volume_mean": -0.003382935654371977,
          "volume_max": 0.02635742351412773,
          "frac_active": 0.3125
        },
        {
          "step": 492,
          "loss": 0.04664076864719391,
          "floss_loss": 0.0004745455225929618,
          "kl_loss": 0.004616622347384691,
          "lyap1_mean": 0.03156362846493721,
          "lyap1_max": 0.055257707834243774,
          "lyap_mean": 0.005101838149130344,
          "volume_mean": 0.005101838149130344,
          "volume_max": 0.021741366013884544,
          "frac_active": 0.5
        },
        {
          "step": 493,
          "loss": 0.13423427939414978,
          "floss_loss": 0.00033282730146311224,
          "kl_loss": 0.013390145264565945,
          "lyap1_mean": 0.015366483479738235,
          "lyap1_max": 0.03590735048055649,
          "lyap_mean": 0.004706484265625477,
          "volume_mean": 0.004706484265625477,
          "volume_max": 0.021420752629637718,
          "frac_active": 0.5625
        },
        {
          "step": 494,
          "loss": 0.001042033196426928,
          "floss_loss": 0.0005727577372454107,
          "kl_loss": 4.692754737334326e-05,
          "lyap1_mean": 0.037203721702098846,
          "lyap1_max": 0.04550058767199516,
          "lyap_mean": 0.015393003821372986,
          "volume_mean": 0.015393003821372986,
          "volume_max": 0.02171074040234089,
          "frac_active": 0.75
        },
        {
          "step": 495,
          "loss": 5.453749179840088,
          "floss_loss": 0.0004655164375435561,
          "kl_loss": 0.5453283786773682,
          "lyap1_mean": 0.03202898055315018,
          "lyap1_max": 0.056847214698791504,
          "lyap_mean": 0.01144791953265667,
          "volume_mean": 0.01144791953265667,
          "volume_max": 0.029118942096829414,
          "frac_active": 0.6875
        },
        {
          "step": 496,
          "loss": 0.13573147356510162,
          "floss_loss": 0.000829450087621808,
          "kl_loss": 0.013490202836692333,
          "lyap1_mean": 0.03967728465795517,
          "lyap1_max": 0.061626702547073364,
          "lyap_mean": 0.02294556424021721,
          "volume_mean": 0.02294556424021721,
          "volume_max": 0.033232443034648895,
          "frac_active": 0.875
        },
        {
          "step": 497,
          "loss": 0.0015135423745959997,
          "floss_loss": 0.0007784360786899924,
          "kl_loss": 7.351063686655834e-05,
          "lyap1_mean": 0.036238379776477814,
          "lyap1_max": 0.055463872849941254,
          "lyap_mean": 0.02271820604801178,
          "volume_mean": 0.02271820604801178,
          "volume_max": 0.027550745755434036,
          "frac_active": 0.9375
        },
        {
          "step": 498,
          "loss": 4.566473960876465,
          "floss_loss": 0.0004489202401600778,
          "kl_loss": 0.4566025137901306,
          "lyap1_mean": 0.019283581525087357,
          "lyap1_max": 0.03274759650230408,
          "lyap_mean": 0.01200871728360653,
          "volume_mean": 0.01200871728360653,
          "volume_max": 0.027667731046676636,
          "frac_active": 0.75
        },
        {
          "step": 499,
          "loss": 0.04428115114569664,
          "floss_loss": 0.0002421797689748928,
          "kl_loss": 0.0044038970954716206,
          "lyap1_mean": 0.01577191799879074,
          "lyap1_max": 0.0393398143351078,
          "lyap_mean": 0.006421446334570646,
          "volume_mean": 0.006421446334570646,
          "volume_max": 0.017245052382349968,
          "frac_active": 0.6875
        }
      ]
    },
    {
      "episode": 1,
      "train_step": 500,
      "steps": [
        {
          "step": 0,
          "loss": 6.328017479972914e-05,
          "floss_loss": 6.328717427095398e-05,
          "kl_loss": -7.000026669423676e-10,
          "lyap1_mean": 0.003074982203543186,
          "lyap1_max": 0.009195102378726006,
          "lyap_mean": -0.0016518577467650175,
          "volume_mean": -0.0016518577467650175,
          "volume_max": 0.0047036693431437016,
          "frac_active": 0.375
        },
        {
          "step": 1,
          "loss": 9.270501136779785,
          "floss_loss": 0.00013977450726088136,
          "kl_loss": 0.9270360469818115,
          "lyap1_mean": 0.013156474567949772,
          "lyap1_max": 0.03167587146162987,
          "lyap_mean": 0.0069197057746350765,
          "volume_mean": 0.0069197057746350765,
          "volume_max": 0.010762471705675125,
          "frac_active": 0.8125
        },
        {
          "step": 2,
          "loss": 6.414191246032715,
          "floss_loss": 0.00013554701581597328,
          "kl_loss": 0.6414055824279785,
          "lyap1_mean": -0.0034442362375557423,
          "lyap1_max": 0.005223047453910112,
          "lyap_mean": -0.005911723710596561,
          "volume_mean": -0.005911723710596561,
          "volume_max": 0.006567391566932201,
          "frac_active": 0.3125
        },
        {
          "step": 3,
          "loss": 2.6271512508392334,
          "floss_loss": 0.0002028681628871709,
          "kl_loss": 0.2626948356628418,
          "lyap1_mean": 0.011256427504122257,
          "lyap1_max": 0.04479994624853134,
          "lyap_mean": -0.0010721328435465693,
          "volume_mean": -0.0010721328435465693,
          "volume_max": 0.00969192199409008,
          "frac_active": 0.3125
        },
        {
          "step": 4,
          "loss": 7.459099292755127,
          "floss_loss": 0.00022164577967487276,
          "kl_loss": 0.7458877563476562,
          "lyap1_mean": 0.016940847039222717,
          "lyap1_max": 0.04054994881153107,
          "lyap_mean": 0.0046873539686203,
          "volume_mean": 0.0046873539686203,
          "volume_max": 0.021113524213433266,
          "frac_active": 0.5625
        },
        {
          "step": 5,
          "loss": 4.98379373550415,
          "floss_loss": 0.0001295428373850882,
          "kl_loss": 0.4983663856983185,
          "lyap1_mean": 0.011645738035440445,
          "lyap1_max": 0.036745309829711914,
          "lyap_mean": 0.0023955386132001877,
          "volume_mean": 0.0023955386132001877,
          "volume_max": 0.012770745903253555,
          "frac_active": 0.4375
        },
        {
          "step": 6,
          "loss": 6.832474708557129,
          "floss_loss": 0.00020443886751309037,
          "kl_loss": 0.683227002620697,
          "lyap1_mean": 0.00680091418325901,
          "lyap1_max": 0.030937599018216133,
          "lyap_mean": -0.0002180817537009716,
          "volume_mean": -0.0002180817537009716,
          "volume_max": 0.011535515077412128,
          "frac_active": 0.25
        },
        {
          "step": 7,
          "loss": 4.73275089263916,
          "floss_loss": 8.228360820794478e-05,
          "kl_loss": 0.4732668399810791,
          "lyap1_mean": -0.005815383978188038,
          "lyap1_max": 0.00044081982923671603,
          "lyap_mean": -0.0067285532131791115,
          "volume_mean": -0.0067285532131791115,
          "volume_max": -0.0016710053896531463,
          "frac_active": 0.1875
        },
        {
          "step": 8,
          "loss": 4.751026153564453,
          "floss_loss": 0.0003907138016074896,
          "kl_loss": 0.4750635623931885,
          "lyap1_mean": 0.015386766754090786,
          "lyap1_max": 0.06771472841501236,
          "lyap_mean": 0.005728051066398621,
          "volume_mean": 0.005728051066398621,
          "volume_max": 0.020731458440423012,
          "frac_active": 0.5625
        },
        {
          "step": 9,
          "loss": 6.900600910186768,
          "floss_loss": 0.00012856149987783283,
          "kl_loss": 0.6900472044944763,
          "lyap1_mean": 0.004243594594299793,
          "lyap1_max": 0.014179104939103127,
          "lyap_mean": 0.003970960155129433,
          "volume_mean": 0.003970960155129433,
          "volume_max": 0.013934504240751266,
          "frac_active": 0.5
        },
        {
          "step": 10,
          "loss": 3.9082868099212646,
          "floss_loss": 0.00011874630581587553,
          "kl_loss": 0.3908168077468872,
          "lyap1_mean": -0.0030709546990692616,
          "lyap1_max": 0.005990639328956604,
          "lyap_mean": -0.005300281569361687,
          "volume_mean": -0.005300281569361687,
          "volume_max": 0.0044369325041770935,
          "frac_active": 0.3125
        },
        {
          "step": 11,
          "loss": 4.444663047790527,
          "floss_loss": 0.0003124929207842797,
          "kl_loss": 0.44443508982658386,
          "lyap1_mean": 0.008351103402674198,
          "lyap1_max": 0.04267052188515663,
          "lyap_mean": 0.004140119068324566,
          "volume_mean": 0.004140119068324566,
          "volume_max": 0.02996310591697693,
          "frac_active": 0.5
        },
        {
          "step": 12,
          "loss": 6.493005752563477,
          "floss_loss": 0.0003988321986980736,
          "kl_loss": 0.6492606997489929,
          "lyap1_mean": 0.029339740052819252,
          "lyap1_max": 0.059887297451496124,
          "lyap_mean": 0.013632014393806458,
          "volume_mean": 0.013632014393806458,
          "volume_max": 0.03021584078669548,
          "frac_active": 0.875
        },
        {
          "step": 13,
          "loss": 4.893409252166748,
          "floss_loss": 0.0001623644056962803,
          "kl_loss": 0.489324688911438,
          "lyap1_mean": 0.014534058049321175,
          "lyap1_max": 0.025248292833566666,
          "lyap_mean": 0.0041245361790061,
          "volume_mean": 0.0041245361790061,
          "volume_max": 0.012874297797679901,
          "frac_active": 0.6875
        },
        {
          "step": 14,
          "loss": 8.132925033569336,
          "floss_loss": 3.6728721170220524e-05,
          "kl_loss": 0.8132887482643127,
          "lyap1_mean": 0.005877241026610136,
          "lyap1_max": 0.013916915282607079,
          "lyap_mean": 0.0034052799455821514,
          "volume_mean": 0.0034052799455821514,
          "volume_max": 0.0055708568543195724,
          "frac_active": 0.6875
        },
        {
          "step": 15,
          "loss": 6.281820297241211,
          "floss_loss": 0.000507041928358376,
          "kl_loss": 0.6281313300132751,
          "lyap1_mean": 0.02771344780921936,
          "lyap1_max": 0.0382099486887455,
          "lyap_mean": 0.02000311017036438,
          "volume_mean": 0.02000311017036438,
          "volume_max": 0.025012288242578506,
          "frac_active": 1.0
        },
        {
          "step": 16,
          "loss": 8.37138557434082,
          "floss_loss": 0.00026344528305344284,
          "kl_loss": 0.8371122479438782,
          "lyap1_mean": 0.02044779062271118,
          "lyap1_max": 0.039306581020355225,
          "lyap_mean": 0.009332496672868729,
          "volume_mean": 0.009332496672868729,
          "volume_max": 0.016867946833372116,
          "frac_active": 0.8125
        },
        {
          "step": 17,
          "loss": 5.201272010803223,
          "floss_loss": 9.822649735724553e-05,
          "kl_loss": 0.5201174020767212,
          "lyap1_mean": 0.010069646872580051,
          "lyap1_max": 0.017020128667354584,
          "lyap_mean": 0.0022722138091921806,
          "volume_mean": 0.0022722138091921806,
          "volume_max": 0.012553377076983452,
          "frac_active": 0.5625
        },
        {
          "step": 18,
          "loss": 3.0941708087921143,
          "floss_loss": 0.00015793721831869334,
          "kl_loss": 0.3094013035297394,
          "lyap1_mean": 0.012327615171670914,
          "lyap1_max": 0.03784750774502754,
          "lyap_mean": 0.0008303516078740358,
          "volume_mean": 0.0008303516078740358,
          "volume_max": 0.007048604544252157,
          "frac_active": 0.4375
        },
        {
          "step": 19,
          "loss": 3.2638485431671143,
          "floss_loss": 0.00022024581267032772,
          "kl_loss": 0.32636281847953796,
          "lyap1_mean": 0.014531673863530159,
          "lyap1_max": 0.04858623445034027,
          "lyap_mean": 0.004964987747371197,
          "volume_mean": 0.004964987747371197,
          "volume_max": 0.02088802680373192,
          "frac_active": 0.5625
        },
        {
          "step": 20,
          "loss": 4.831172943115234,
          "floss_loss": 0.0002876124926842749,
          "kl_loss": 0.48308855295181274,
          "lyap1_mean": 0.0012114346027374268,
          "lyap1_max": 0.015580620616674423,
          "lyap_mean": -0.005717182997614145,
          "volume_mean": -0.005717182997614145,
          "volume_max": 0.007805037312209606,
          "frac_active": 0.4375
        },
        {
          "step": 21,
          "loss": 2.7474095821380615,
          "floss_loss": 9.805505396798253e-05,
          "kl_loss": 0.2747311592102051,
          "lyap1_mean": 0.004381884355098009,
          "lyap1_max": 0.01772357150912285,
          "lyap_mean": -0.004637612961232662,
          "volume_mean": -0.004637612961232662,
          "volume_max": 0.00012972624972462654,
          "frac_active": 0.25
        },
        {
          "step": 22,
          "loss": 3.8391566276550293,
          "floss_loss": 0.0003605576348491013,
          "kl_loss": 0.3838796019554138,
          "lyap1_mean": -0.011909889057278633,
          "lyap1_max": -0.0005815438344143331,
          "lyap_mean": -0.0163239985704422,
          "volume_mean": -0.0163239985704422,
          "volume_max": -0.006045383866876364,
          "frac_active": 0.0625
        },
        {
          "step": 23,
          "loss": 1.9332689046859741,
          "floss_loss": 0.0005162612069398165,
          "kl_loss": 0.19327525794506073,
          "lyap1_mean": -0.016048211604356766,
          "lyap1_max": 0.0032586243469268084,
          "lyap_mean": -0.020740263164043427,
          "volume_mean": -0.020740263164043427,
          "volume_max": -0.011089392937719822,
          "frac_active": 0.0625
        },
        {
          "step": 24,
          "loss": 1.868513584136963,
          "floss_loss": 0.00040132668800652027,
          "kl_loss": 0.18681122362613678,
          "lyap1_mean": -0.013049593195319176,
          "lyap1_max": 0.011732334271073341,
          "lyap_mean": -0.0174422524869442,
          "volume_mean": -0.0174422524869442,
          "volume_max": -0.011749413795769215,
          "frac_active": 0.0625
        },
        {
          "step": 25,
          "loss": 2.468503713607788,
          "floss_loss": 0.00037993292789906263,
          "kl_loss": 0.2468123733997345,
          "lyap1_mean": -0.003369454527273774,
          "lyap1_max": 0.02836998924612999,
          "lyap_mean": -0.012428659945726395,
          "volume_mean": -0.012428659945726395,
          "volume_max": 0.00851413793861866,
          "frac_active": 0.1875
        },
        {
          "step": 26,
          "loss": 2.2052500247955322,
          "floss_loss": 0.0001971642195712775,
          "kl_loss": 0.22050528228282928,
          "lyap1_mean": -0.00012525077909231186,
          "lyap1_max": 0.03249155730009079,
          "lyap_mean": -0.007112199440598488,
          "volume_mean": -0.007112199440598488,
          "volume_max": 0.0036149779334664345,
          "frac_active": 0.125
        },
        {
          "step": 27,
          "loss": 1.7576261758804321,
          "floss_loss": 0.00018450683273840696,
          "kl_loss": 0.1757441610097885,
          "lyap1_mean": 0.007466732989996672,
          "lyap1_max": 0.025453489273786545,
          "lyap_mean": -0.0051289089024066925,
          "volume_mean": -0.0051289089024066925,
          "volume_max": 0.005336785223335028,
          "frac_active": 0.375
        },
        {
          "step": 28,
          "loss": 2.4823100566864014,
          "floss_loss": 7.093630847521126e-05,
          "kl_loss": 0.2482239007949829,
          "lyap1_mean": 0.0012568288948386908,
          "lyap1_max": 0.010691037401556969,
          "lyap_mean": -0.000534741033334285,
          "volume_mean": -0.000534741033334285,
          "volume_max": 0.005573983769863844,
          "frac_active": 0.4375
        },
        {
          "step": 29,
          "loss": 2.6557154655456543,
          "floss_loss": 8.312337740790099e-05,
          "kl_loss": 0.2655632197856903,
          "lyap1_mean": -0.0018092450918629766,
          "lyap1_max": -0.00014042480324860662,
          "lyap_mean": -0.006447847932577133,
          "volume_mean": -0.006447847932577133,
          "volume_max": -0.0003905276535078883,
          "frac_active": 0.0625
        },
        {
          "step": 30,
          "loss": 2.8482635021209717,
          "floss_loss": 0.00023329004761762917,
          "kl_loss": 0.28480303287506104,
          "lyap1_mean": -0.0010696593672037125,
          "lyap1_max": 0.018658721819519997,
          "lyap_mean": -0.0084749236702919,
          "volume_mean": -0.0084749236702919,
          "volume_max": -0.0010690430644899607,
          "frac_active": 0.25
        },
        {
          "step": 31,
          "loss": 4.313543796539307,
          "floss_loss": 0.00039828300941735506,
          "kl_loss": 0.4313145875930786,
          "lyap1_mean": 0.007603127975016832,
          "lyap1_max": 0.03884423151612282,
          "lyap_mean": -0.001923329196870327,
          "volume_mean": -0.001923329196870327,
          "volume_max": 0.023918088525533676,
          "frac_active": 0.375
        },
        {
          "step": 32,
          "loss": 3.1125576496124268,
          "floss_loss": 0.00013806152855977416,
          "kl_loss": 0.3112419545650482,
          "lyap1_mean": 0.012232648208737373,
          "lyap1_max": 0.01815744675695896,
          "lyap_mean": 0.0028582150116562843,
          "volume_mean": 0.0028582150116562843,
          "volume_max": 0.012509537860751152,
          "frac_active": 0.625
        },
        {
          "step": 33,
          "loss": 2.5734686851501465,
          "floss_loss": 0.00034989294363185763,
          "kl_loss": 0.2573118805885315,
          "lyap1_mean": 0.0017334651201963425,
          "lyap1_max": 0.026081077754497528,
          "lyap_mean": -0.00703053455799818,
          "volume_mean": -0.00703053455799818,
          "volume_max": 0.011807533912360668,
          "frac_active": 0.3125
        },
        {
          "step": 34,
          "loss": 1.3112789392471313,
          "floss_loss": 0.00015683849051129073,
          "kl_loss": 0.131112203001976,
          "lyap1_mean": 0.01666153222322464,
          "lyap1_max": 0.0346088744699955,
          "lyap_mean": 0.00516038341447711,
          "volume_mean": 0.00516038341447711,
          "volume_max": 0.012025982141494751,
          "frac_active": 0.625
        },
        {
          "step": 35,
          "loss": 1.0591094493865967,
          "floss_loss": 0.0005661126342602074,
          "kl_loss": 0.105854332447052,
          "lyap1_mean": 0.008009595796465874,
          "lyap1_max": 0.044996775686740875,
          "lyap_mean": 0.001390466932207346,
          "volume_mean": 0.001390466932207346,
          "volume_max": 0.029416393488645554,
          "frac_active": 0.5
        },
        {
          "step": 36,
          "loss": 2.151500701904297,
          "floss_loss": 0.0004745524493046105,
          "kl_loss": 0.21510262787342072,
          "lyap1_mean": -0.00592763489112258,
          "lyap1_max": 0.014171618036925793,
          "lyap_mean": -0.01395285502076149,
          "volume_mean": -0.01395285502076149,
          "volume_max": 0.006822574883699417,
          "frac_active": 0.25
        },
        {
          "step": 37,
          "loss": 1.522001028060913,
          "floss_loss": 0.0003531168622430414,
          "kl_loss": 0.1521647870540619,
          "lyap1_mean": -0.007792357821017504,
          "lyap1_max": 0.0030635674484074116,
          "lyap_mean": -0.015197427943348885,
          "volume_mean": -0.015197427943348885,
          "volume_max": -0.002405900042504072,
          "frac_active": 0.125
        },
        {
          "step": 38,
          "loss": 1.3780581951141357,
          "floss_loss": 0.0004976982600055635,
          "kl_loss": 0.13775604963302612,
          "lyap1_mean": -0.015420226380228996,
          "lyap1_max": 0.0016291055362671614,
          "lyap_mean": -0.019505983218550682,
          "volume_mean": -0.019505983218550682,
          "volume_max": -0.009422685950994492,
          "frac_active": 0.0625
        },
        {
          "step": 39,
          "loss": 1.0015236139297485,
          "floss_loss": 0.00034750360646285117,
          "kl_loss": 0.10011761635541916,
          "lyap1_mean": -0.010344238951802254,
          "lyap1_max": 0.003783318679779768,
          "lyap_mean": -0.01688406802713871,
          "volume_mean": -0.01688406802713871,
          "volume_max": -0.009939340874552727,
          "frac_active": 0.0625
        },
        {
          "step": 40,
          "loss": 1.2406550645828247,
          "floss_loss": 0.00028587906854227185,
          "kl_loss": 0.12403692305088043,
          "lyap1_mean": 0.002628587419167161,
          "lyap1_max": 0.012763386592268944,
          "lyap_mean": -0.01128101721405983,
          "volume_mean": -0.01128101721405983,
          "volume_max": -0.002850920893251896,
          "frac_active": 0.1875
        },
        {
          "step": 41,
          "loss": 1.2369972467422485,
          "floss_loss": 0.0003296840295661241,
          "kl_loss": 0.12366675585508347,
          "lyap1_mean": 0.0012805270962417126,
          "lyap1_max": 0.0329323336482048,
          "lyap_mean": -0.007891812361776829,
          "volume_mean": -0.007891812361776829,
          "volume_max": 0.0030223634093999863,
          "frac_active": 0.25
        },
        {
          "step": 42,
          "loss": 0.931557834148407,
          "floss_loss": 0.00037406192859634757,
          "kl_loss": 0.09311837702989578,
          "lyap1_mean": -0.015066150575876236,
          "lyap1_max": -0.0038720951415598392,
          "lyap_mean": -0.017206426709890366,
          "volume_mean": -0.017206426709890366,
          "volume_max": -0.00808127410709858,
          "frac_active": 0.0
        },
        {
          "step": 43,
          "loss": 1.039440631866455,
          "floss_loss": 0.00023612660879734904,
          "kl_loss": 0.10392045229673386,
          "lyap1_mean": -0.0018984302878379822,
          "lyap1_max": 0.004670607857406139,
          "lyap_mean": -0.011337733827531338,
          "volume_mean": -0.011337733827531338,
          "volume_max": -0.003895227797329426,
          "frac_active": 0.125
        },
        {
          "step": 44,
          "loss": 1.7195035219192505,
          "floss_loss": 0.0002659743186086416,
          "kl_loss": 0.17192375659942627,
          "lyap1_mean": 0.00734610203653574,
          "lyap1_max": 0.01948082447052002,
          "lyap_mean": -0.007843539118766785,
          "volume_mean": -0.007843539118766785,
          "volume_max": 0.0025829803198575974,
          "frac_active": 0.25
        },
        {
          "step": 45,
          "loss": 2.337538242340088,
          "floss_loss": 0.00040895878919400275,
          "kl_loss": 0.23371294140815735,
          "lyap1_mean": -0.013511978089809418,
          "lyap1_max": 0.0006980617763474584,
          "lyap_mean": -0.015933237969875336,
          "volume_mean": -0.015933237969875336,
          "volume_max": -0.0019684068392962217,
          "frac_active": 0.25
        },
        {
          "step": 46,
          "loss": 2.0605716705322266,
          "floss_loss": 0.0004489832790568471,
          "kl_loss": 0.2060122787952423,
          "lyap1_mean": -0.012598340399563313,
          "lyap1_max": -0.002008403418585658,
          "lyap_mean": -0.019695019349455833,
          "volume_mean": -0.019695019349455833,
          "volume_max": -0.013296491466462612,
          "frac_active": 0.0
        },
        {
          "step": 47,
          "loss": 1.6233651638031006,
          "floss_loss": 0.0006501457537524402,
          "kl_loss": 0.16227149963378906,
          "lyap1_mean": -0.017168043181300163,
          "lyap1_max": 0.010746228508651257,
          "lyap_mean": -0.022986479103565216,
          "volume_mean": -0.022986479103565216,
          "volume_max": -0.006793830543756485,
          "frac_active": 0.0625
        },
        {
          "step": 48,
          "loss": 1.8089536428451538,
          "floss_loss": 7.84159274189733e-05,
          "kl_loss": 0.18088752031326294,
          "lyap1_mean": 0.004909857176244259,
          "lyap1_max": 0.020896030589938164,
          "lyap_mean": 7.021240890026093e-05,
          "volume_mean": 7.021240890026093e-05,
          "volume_max": 0.011202588677406311,
          "frac_active": 0.5625
        },
        {
          "step": 49,
          "loss": 1.7574869394302368,
          "floss_loss": 0.00028589670546352863,
          "kl_loss": 0.17572011053562164,
          "lyap1_mean": -0.0021928860805928707,
          "lyap1_max": 0.023308521136641502,
          "lyap_mean": -0.009579772129654884,
          "volume_mean": -0.009579772129654884,
          "volume_max": 0.003545103594660759,
          "frac_active": 0.125
        },
        {
          "step": 50,
          "loss": 1.2125149965286255,
          "floss_loss": 0.0005643600597977638,
          "kl_loss": 0.12119506299495697,
          "lyap1_mean": -0.01819346845149994,
          "lyap1_max": -0.008466055616736412,
          "lyap_mean": -0.022662758827209473,
          "volume_mean": -0.022662758827209473,
          "volume_max": -0.018580574542284012,
          "frac_active": 0.0
        },
        {
          "step": 51,
          "loss": 0.721222996711731,
          "floss_loss": 0.00032361919875256717,
          "kl_loss": 0.07208994030952454,
          "lyap1_mean": -0.0013606044230982661,
          "lyap1_max": 0.011587624438107014,
          "lyap_mean": -0.013389071449637413,
          "volume_mean": -0.013389071449637413,
          "volume_max": -0.0022888826206326485,
          "frac_active": 0.125
        },
        {
          "step": 52,
          "loss": 1.7036086320877075,
          "floss_loss": 0.0003993727150373161,
          "kl_loss": 0.17032092809677124,
          "lyap1_mean": -0.01778986304998398,
          "lyap1_max": -0.007275557145476341,
          "lyap_mean": -0.016712425276637077,
          "volume_mean": -0.016712425276637077,
          "volume_max": -0.00924181379377842,
          "frac_active": 0.0625
        },
        {
          "step": 53,
          "loss": 1.4536186456680298,
          "floss_loss": 0.00015279764193110168,
          "kl_loss": 0.14534658193588257,
          "lyap1_mean": 0.006442279554903507,
          "lyap1_max": 0.02678067609667778,
          "lyap_mean": -0.00507101695984602,
          "volume_mean": -0.00507101695984602,
          "volume_max": 0.006068829447031021,
          "frac_active": 0.25
        },
        {
          "step": 54,
          "loss": 1.0074764490127563,
          "floss_loss": 0.0005361692747101188,
          "kl_loss": 0.10069402307271957,
          "lyap1_mean": -0.015339154750108719,
          "lyap1_max": -0.004081742372363806,
          "lyap_mean": -0.020673900842666626,
          "volume_mean": -0.020673900842666626,
          "volume_max": -0.009059581905603409,
          "frac_active": 0.0
        },
        {
          "step": 55,
          "loss": 0.5131212472915649,
          "floss_loss": 0.0005332670407369733,
          "kl_loss": 0.05125879868865013,
          "lyap1_mean": -0.020335528999567032,
          "lyap1_max": -0.004371861461549997,
          "lyap_mean": -0.02119547314941883,
          "volume_mean": -0.02119547314941883,
          "volume_max": -0.014038052409887314,
          "frac_active": 0.0
        },
        {
          "step": 56,
          "loss": 0.8429496884346008,
          "floss_loss": 0.0003232797025702894,
          "kl_loss": 0.08426263928413391,
          "lyap1_mean": -0.014490395784378052,
          "lyap1_max": -0.007338810712099075,
          "lyap_mean": -0.016925984993577003,
          "volume_mean": -0.016925984993577003,
          "volume_max": -0.013490313664078712,
          "frac_active": 0.0
        },
        {
          "step": 57,
          "loss": 0.7632055878639221,
          "floss_loss": 0.0008586860494688153,
          "kl_loss": 0.07623469084501266,
          "lyap1_mean": -0.025805801153182983,
          "lyap1_max": -0.015073495917022228,
          "lyap_mean": -0.028698179870843887,
          "volume_mean": -0.028698179870843887,
          "volume_max": -0.02291906625032425,
          "frac_active": 0.0
        },
        {
          "step": 58,
          "loss": 0.29738306999206543,
          "floss_loss": 0.0006892665405757725,
          "kl_loss": 0.02966937981545925,
          "lyap1_mean": -0.022832540795207024,
          "lyap1_max": -0.012986881658434868,
          "lyap_mean": -0.02539214678108692,
          "volume_mean": -0.02539214678108692,
          "volume_max": -0.018313350155949593,
          "frac_active": 0.0
        },
        {
          "step": 59,
          "loss": 0.5452380180358887,
          "floss_loss": 0.0008429771405644715,
          "kl_loss": 0.05443950369954109,
          "lyap1_mean": -0.023014739155769348,
          "lyap1_max": -0.0132977906614542,
          "lyap_mean": -0.028153792023658752,
          "volume_mean": -0.028153792023658752,
          "volume_max": -0.020785193890333176,
          "frac_active": 0.0
        },
        {
          "step": 60,
          "loss": 1.0623615980148315,
          "floss_loss": 0.0004810639948118478,
          "kl_loss": 0.10618805885314941,
          "lyap1_mean": -0.016350096091628075,
          "lyap1_max": -0.00547960726544261,
          "lyap_mean": -0.018811162561178207,
          "volume_mean": -0.018811162561178207,
          "volume_max": -0.006097682286053896,
          "frac_active": 0.0625
        },
        {
          "step": 61,
          "loss": 1.260214924812317,
          "floss_loss": 0.0006405484164133668,
          "kl_loss": 0.1259574443101883,
          "lyap1_mean": -0.018976926803588867,
          "lyap1_max": -0.006086866836994886,
          "lyap_mean": -0.02429666928946972,
          "volume_mean": -0.02429666928946972,
          "volume_max": -0.02010185830295086,
          "frac_active": 0.0
        },
        {
          "step": 62,
          "loss": 1.3323794603347778,
          "floss_loss": 0.000826004019472748,
          "kl_loss": 0.13315534591674805,
          "lyap1_mean": -0.018198955804109573,
          "lyap1_max": 0.009347846731543541,
          "lyap_mean": -0.02684273011982441,
          "volume_mean": -0.02684273011982441,
          "volume_max": -0.01834038272500038,
          "frac_active": 0.0625
        },
        {
          "step": 63,
          "loss": 1.2920513153076172,
          "floss_loss": 0.0006716572679579258,
          "kl_loss": 0.12913796305656433,
          "lyap1_mean": -0.02089059166610241,
          "lyap1_max": -0.005518355406820774,
          "lyap_mean": -0.023605555295944214,
          "volume_mean": -0.023605555295944214,
          "volume_max": -0.007077369373291731,
          "frac_active": 0.0
        },
        {
          "step": 64,
          "loss": 1.9547759294509888,
          "floss_loss": 0.000841508386656642,
          "kl_loss": 0.19539344310760498,
          "lyap1_mean": -0.025101277977228165,
          "lyap1_max": -0.01978812739253044,
          "lyap_mean": -0.02862865850329399,
          "volume_mean": -0.02862865850329399,
          "volume_max": -0.024187594652175903,
          "frac_active": 0.0
        },
        {
          "step": 65,
          "loss": 1.6264963150024414,
          "floss_loss": 0.0010585603304207325,
          "kl_loss": 0.16254377365112305,
          "lyap1_mean": -0.027153925970196724,
          "lyap1_max": -0.014577385038137436,
          "lyap_mean": -0.03149637579917908,
          "volume_mean": -0.03149637579917908,
          "volume_max": -0.02332322858273983,
          "frac_active": 0.0
        },
        {
          "step": 66,
          "loss": 1.0190352201461792,
          "floss_loss": 0.00022585387341678143,
          "kl_loss": 0.10188093036413193,
          "lyap1_mean": -0.000571058364585042,
          "lyap1_max": 0.018083736300468445,
          "lyap_mean": -0.009252062067389488,
          "volume_mean": -0.009252062067389488,
          "volume_max": -0.0002432381152175367,
          "frac_active": 0.1875
        },
        {
          "step": 67,
          "loss": 0.9374391436576843,
          "floss_loss": 0.0006942871259525418,
          "kl_loss": 0.09367448836565018,
          "lyap1_mean": -0.021163813769817352,
          "lyap1_max": -0.007124762516468763,
          "lyap_mean": -0.025399446487426758,
          "volume_mean": -0.025399446487426758,
          "volume_max": -0.017936427146196365,
          "frac_active": 0.0
        },
        {
          "step": 68,
          "loss": 0.7163166999816895,
          "floss_loss": 0.0003949004167225212,
          "kl_loss": 0.07159218192100525,
          "lyap1_mean": -0.010823851451277733,
          "lyap1_max": 0.007196018006652594,
          "lyap_mean": -0.015249993652105331,
          "volume_mean": -0.015249993652105331,
          "volume_max": -0.0011750997509807348,
          "frac_active": 0.125
        },
        {
          "step": 69,
          "loss": 0.9182990789413452,
          "floss_loss": 0.0006618439219892025,
          "kl_loss": 0.09176371991634369,
          "lyap1_mean": -0.013047240674495697,
          "lyap1_max": 0.004247646313160658,
          "lyap_mean": -0.023431073874235153,
          "volume_mean": -0.023431073874235153,
          "volume_max": -0.01395677775144577,
          "frac_active": 0.0625
        },
        {
          "step": 70,
          "loss": 0.8058339357376099,
          "floss_loss": 0.0005264139035716653,
          "kl_loss": 0.08053074777126312,
          "lyap1_mean": -0.012670131400227547,
          "lyap1_max": 0.00113482098095119,
          "lyap_mean": -0.02091853879392147,
          "volume_mean": -0.02091853879392147,
          "volume_max": -0.012979150749742985,
          "frac_active": 0.0625
        },
        {
          "step": 71,
          "loss": 0.8218425512313843,
          "floss_loss": 0.0003610565618146211,
          "kl_loss": 0.08214814960956573,
          "lyap1_mean": -0.010736465454101562,
          "lyap1_max": 0.0020576012320816517,
          "lyap_mean": -0.01676926016807556,
          "volume_mean": -0.01676926016807556,
          "volume_max": -0.008151819929480553,
          "frac_active": 0.0625
        },
        {
          "step": 72,
          "loss": 0.9286670684814453,
          "floss_loss": 0.0007021598285064101,
          "kl_loss": 0.09279648959636688,
          "lyap1_mean": -0.02352522499859333,
          "lyap1_max": -0.003968952223658562,
          "lyap_mean": -0.025282008573412895,
          "volume_mean": -0.025282008573412895,
          "volume_max": -0.017214536666870117,
          "frac_active": 0.0
        },
        {
          "step": 73,
          "loss": 1.2569395303726196,
          "floss_loss": 0.00020405872783157974,
          "kl_loss": 0.1256735473871231,
          "lyap1_mean": -0.0038553299382328987,
          "lyap1_max": 0.009691386483609676,
          "lyap_mean": -0.00945349968969822,
          "volume_mean": -0.00945349968969822,
          "volume_max": 0.0050823874771595,
          "frac_active": 0.1875
        },
        {
          "step": 74,
          "loss": 0.5836705565452576,
          "floss_loss": 0.0003058395814150572,
          "kl_loss": 0.058336470276117325,
          "lyap1_mean": 0.010794445872306824,
          "lyap1_max": 0.028384024277329445,
          "lyap_mean": -0.003478748258203268,
          "volume_mean": -0.003478748258203268,
          "volume_max": 0.007494863588362932,
          "frac_active": 0.4375
        },
        {
          "step": 75,
          "loss": 0.7492558360099792,
          "floss_loss": 0.000506511889398098,
          "kl_loss": 0.07487493008375168,
          "lyap1_mean": -0.013683482073247433,
          "lyap1_max": -0.004291845485568047,
          "lyap_mean": -0.019652551040053368,
          "volume_mean": -0.019652551040053368,
          "volume_max": -0.00634824950248003,
          "frac_active": 0.0625
        },
        {
          "step": 76,
          "loss": 0.7713541388511658,
          "floss_loss": 0.0005588682834059,
          "kl_loss": 0.07707952708005905,
          "lyap1_mean": -0.004087822511792183,
          "lyap1_max": 0.025625573471188545,
          "lyap_mean": -0.019426744431257248,
          "volume_mean": -0.019426744431257248,
          "volume_max": -0.015284470282495022,
          "frac_active": 0.0625
        },
        {
          "step": 77,
          "loss": 0.5226860046386719,
          "floss_loss": 0.00023258622968569398,
          "kl_loss": 0.05224534496665001,
          "lyap1_mean": 0.0004601869732141495,
          "lyap1_max": 0.02678535133600235,
          "lyap_mean": -0.00823371671140194,
          "volume_mean": -0.00823371671140194,
          "volume_max": 0.006696031894534826,
          "frac_active": 0.1875
        },
        {
          "step": 78,
          "loss": 1.0325520038604736,
          "floss_loss": 0.0003566481464076787,
          "kl_loss": 0.10321953892707825,
          "lyap1_mean": -0.009828006848692894,
          "lyap1_max": 0.0070812939666211605,
          "lyap_mean": -0.015031995251774788,
          "volume_mean": -0.015031995251774788,
          "volume_max": -0.005505580920726061,
          "frac_active": 0.0625
        },
        {
          "step": 79,
          "loss": 0.6785794496536255,
          "floss_loss": 0.0002807493438012898,
          "kl_loss": 0.06782986968755722,
          "lyap1_mean": -0.007889612577855587,
          "lyap1_max": 0.008805696852505207,
          "lyap_mean": -0.011006031185388565,
          "volume_mean": -0.011006031185388565,
          "volume_max": 0.002263488946482539,
          "frac_active": 0.25
        },
        {
          "step": 80,
          "loss": 1.0284559726715088,
          "floss_loss": 0.0004437872557900846,
          "kl_loss": 0.10280121862888336,
          "lyap1_mean": -0.0107704047113657,
          "lyap1_max": 0.0086800092831254,
          "lyap_mean": -0.01545574702322483,
          "volume_mean": -0.01545574702322483,
          "volume_max": -0.0035600317642092705,
          "frac_active": 0.1875
        },
        {
          "step": 81,
          "loss": 1.1116999387741089,
          "floss_loss": 0.0006210455321706831,
          "kl_loss": 0.11110788583755493,
          "lyap1_mean": -0.011500367894768715,
          "lyap1_max": 0.0032185735180974007,
          "lyap_mean": -0.021468155086040497,
          "volume_mean": -0.021468155086040497,
          "volume_max": -0.012102979235351086,
          "frac_active": 0.125
        },
        {
          "step": 82,
          "loss": 0.7127127051353455,
          "floss_loss": 0.0005118162371218204,
          "kl_loss": 0.07122008502483368,
          "lyap1_mean": -0.014345202594995499,
          "lyap1_max": -0.006285311188548803,
          "lyap_mean": -0.0212014801800251,
          "volume_mean": -0.0212014801800251,
          "volume_max": -0.014269715175032616,
          "frac_active": 0.0
        },
        {
          "step": 83,
          "loss": 0.575710654258728,
          "floss_loss": 0.0004216599918436259,
          "kl_loss": 0.05752889811992645,
          "lyap1_mean": -0.013260139152407646,
          "lyap1_max": 0.005486899986863136,
          "lyap_mean": -0.01803699880838394,
          "volume_mean": -0.01803699880838394,
          "volume_max": -0.008847897872328758,
          "frac_active": 0.0625
        },
        {
          "step": 84,
          "loss": 0.948055624961853,
          "floss_loss": 0.0006329966126941144,
          "kl_loss": 0.09474226087331772,
          "lyap1_mean": -0.023604044690728188,
          "lyap1_max": -0.009494086727499962,
          "lyap_mean": -0.02347511611878872,
          "volume_mean": -0.02347511611878872,
          "volume_max": -0.01064751110970974,
          "frac_active": 0.0
        },
        {
          "step": 85,
          "loss": 0.7706918716430664,
          "floss_loss": 0.0004502861120272428,
          "kl_loss": 0.07702415436506271,
          "lyap1_mean": -0.01024160347878933,
          "lyap1_max": 0.004800410009920597,
          "lyap_mean": -0.015391466207802296,
          "volume_mean": -0.015391466207802296,
          "volume_max": 0.0006204029778018594,
          "frac_active": 0.125
        },
        {
          "step": 86,
          "loss": 0.5306857824325562,
          "floss_loss": 0.00040613242890685797,
          "kl_loss": 0.053027961403131485,
          "lyap1_mean": -0.01539706252515316,
          "lyap1_max": -0.0015513981925323606,
          "lyap_mean": -0.019013280048966408,
          "volume_mean": -0.019013280048966408,
          "volume_max": -0.013531853444874287,
          "frac_active": 0.0
        },
        {
          "step": 87,
          "loss": 0.8272759318351746,
          "floss_loss": 0.00034495649742893875,
          "kl_loss": 0.08269309997558594,
          "lyap1_mean": -0.006064810790121555,
          "lyap1_max": 0.01305369008332491,
          "lyap_mean": -0.012415044009685516,
          "volume_mean": -0.012415044009685516,
          "volume_max": 0.0007620123215019703,
          "frac_active": 0.25
        },
        {
          "step": 88,
          "loss": 0.7723661065101624,
          "floss_loss": 0.0008918943349272013,
          "kl_loss": 0.07714741677045822,
          "lyap1_mean": -0.015399331226944923,
          "lyap1_max": 0.027817394584417343,
          "lyap_mean": -0.025140423327684402,
          "volume_mean": -0.025140423327684402,
          "volume_max": -0.007544071413576603,
          "frac_active": 0.0625
        },
        {
          "step": 89,
          "loss": 0.7344542741775513,
          "floss_loss": 0.00035777597804553807,
          "kl_loss": 0.07340965420007706,
          "lyap1_mean": -0.0004619951359927654,
          "lyap1_max": 0.01714864932000637,
          "lyap_mean": -0.013903738930821419,
          "volume_mean": -0.013903738930821419,
          "volume_max": -0.0009035246912389994,
          "frac_active": 0.125
        },
        {
          "step": 90,
          "loss": 2.804069995880127,
          "floss_loss": 0.00044987109140492976,
          "kl_loss": 0.28036201000213623,
          "lyap1_mean": -0.007339513394981623,
          "lyap1_max": 0.015156084671616554,
          "lyap_mean": -0.01612752303481102,
          "volume_mean": -0.01612752303481102,
          "volume_max": -0.004000913351774216,
          "frac_active": 0.125
        },
        {
          "step": 91,
          "loss": 1.016316294670105,
          "floss_loss": 0.0006126679363660514,
          "kl_loss": 0.10157036781311035,
          "lyap1_mean": -0.012851115316152573,
          "lyap1_max": 0.021254131570458412,
          "lyap_mean": -0.01831253618001938,
          "volume_mean": -0.01831253618001938,
          "volume_max": 0.006064468063414097,
          "frac_active": 0.1875
        },
        {
          "step": 92,
          "loss": 1.8357770442962646,
          "floss_loss": 0.00019488175166770816,
          "kl_loss": 0.18355821073055267,
          "lyap1_mean": -0.008437958545982838,
          "lyap1_max": 0.00028564303647726774,
          "lyap_mean": -0.012510059401392937,
          "volume_mean": -0.012510059401392937,
          "volume_max": -0.010778211057186127,
          "frac_active": 0.0625
        },
        {
          "step": 93,
          "loss": 0.5756798386573792,
          "floss_loss": 0.0003615766472648829,
          "kl_loss": 0.05753182992339134,
          "lyap1_mean": -0.010940758511424065,
          "lyap1_max": 0.007213003933429718,
          "lyap_mean": -0.014953109435737133,
          "volume_mean": -0.014953109435737133,
          "volume_max": -0.003029164858162403,
          "frac_active": 0.1875
        },
        {
          "step": 94,
          "loss": 4.498968601226807,
          "floss_loss": 0.0005064625293016434,
          "kl_loss": 0.4498462378978729,
          "lyap1_mean": -0.014190828427672386,
          "lyap1_max": 0.0006152002024464309,
          "lyap_mean": -0.01998061127960682,
          "volume_mean": -0.01998061127960682,
          "volume_max": -0.01105822529643774,
          "frac_active": 0.0625
        },
        {
          "step": 95,
          "loss": 0.9319536685943604,
          "floss_loss": 0.0005786721594631672,
          "kl_loss": 0.09313749521970749,
          "lyap1_mean": -0.009385275654494762,
          "lyap1_max": 0.015272297896444798,
          "lyap_mean": -0.01865125074982643,
          "volume_mean": -0.01865125074982643,
          "volume_max": 0.0013416591100394726,
          "frac_active": 0.125
        },
        {
          "step": 96,
          "loss": 1.0450780391693115,
          "floss_loss": 0.0005210199160501361,
          "kl_loss": 0.10445570200681686,
          "lyap1_mean": -0.0015029115602374077,
          "lyap1_max": 0.041182346642017365,
          "lyap_mean": -0.015177085995674133,
          "volume_mean": -0.015177085995674133,
          "volume_max": 0.007267106790095568,
          "frac_active": 0.125
        },
        {
          "step": 97,
          "loss": 1.2910341024398804,
          "floss_loss": 0.0007339242729358375,
          "kl_loss": 0.1290300190448761,
          "lyap1_mean": -0.025761626660823822,
          "lyap1_max": -0.012143857777118683,
          "lyap_mean": -0.024052385240793228,
          "volume_mean": -0.024052385240793228,
          "volume_max": -0.005903709679841995,
          "frac_active": 0.0625
        },
        {
          "step": 98,
          "loss": 0.5249903798103333,
          "floss_loss": 0.0007150059100240469,
          "kl_loss": 0.052427537739276886,
          "lyap1_mean": -0.023096714168787003,
          "lyap1_max": -0.006807828322052956,
          "lyap_mean": -0.023821229115128517,
          "volume_mean": -0.023821229115128517,
          "volume_max": -0.007744366303086281,
          "frac_active": 0.0625
        },
        {
          "step": 99,
          "loss": 0.4748031198978424,
          "floss_loss": 0.0004416976880747825,
          "kl_loss": 0.04743614047765732,
          "lyap1_mean": -0.01600196771323681,
          "lyap1_max": -0.002641032449901104,
          "lyap_mean": -0.01956886425614357,
          "volume_mean": -0.01956886425614357,
          "volume_max": -0.012429251335561275,
          "frac_active": 0.0
        },
        {
          "step": 100,
          "loss": 0.47062408924102783,
          "floss_loss": 0.0006056316778995097,
          "kl_loss": 0.04700184613466263,
          "lyap1_mean": -0.015528758056461811,
          "lyap1_max": 0.00800007302314043,
          "lyap_mean": -0.02236129716038704,
          "volume_mean": -0.02236129716038704,
          "volume_max": -0.012093574739992619,
          "frac_active": 0.0625
        },
        {
          "step": 101,
          "loss": 0.6005672216415405,
          "floss_loss": 0.000395644165109843,
          "kl_loss": 0.06001715734601021,
          "lyap1_mean": -0.010120733641088009,
          "lyap1_max": 0.011137272231280804,
          "lyap_mean": -0.016950175166130066,
          "volume_mean": -0.016950175166130066,
          "volume_max": -0.007590869441628456,
          "frac_active": 0.0625
        },
        {
          "step": 102,
          "loss": 1.5095033645629883,
          "floss_loss": 0.0004394825082272291,
          "kl_loss": 0.15090638399124146,
          "lyap1_mean": -0.012333802878856659,
          "lyap1_max": -0.001680832589045167,
          "lyap_mean": -0.01800987310707569,
          "volume_mean": -0.01800987310707569,
          "volume_max": -0.008269092068076134,
          "frac_active": 0.0
        },
        {
          "step": 103,
          "loss": 0.6552807688713074,
          "floss_loss": 0.00035637704422697425,
          "kl_loss": 0.06549243628978729,
          "lyap1_mean": -0.012957843020558357,
          "lyap1_max": 0.001808243221603334,
          "lyap_mean": -0.01650165021419525,
          "volume_mean": -0.01650165021419525,
          "volume_max": -0.006210274063050747,
          "frac_active": 0.125
        },
        {
          "step": 104,
          "loss": 1.1018239259719849,
          "floss_loss": 0.0005907622398808599,
          "kl_loss": 0.11012331396341324,
          "lyap1_mean": -0.017539910972118378,
          "lyap1_max": 0.0050282906740903854,
          "lyap_mean": -0.02204589545726776,
          "volume_mean": -0.02204589545726776,
          "volume_max": -0.011608833447098732,
          "frac_active": 0.0625
        },
        {
          "step": 105,
          "loss": 1.2084417343139648,
          "floss_loss": 0.00031208409927785397,
          "kl_loss": 0.12081296741962433,
          "lyap1_mean": -0.015333245508372784,
          "lyap1_max": -0.0038476570043712854,
          "lyap_mean": -0.015855571255087852,
          "volume_mean": -0.015855571255087852,
          "volume_max": -0.010836878791451454,
          "frac_active": 0.0
        },
        {
          "step": 106,
          "loss": 0.44280001521110535,
          "floss_loss": 0.0004947375273332,
          "kl_loss": 0.04423052817583084,
          "lyap1_mean": -0.01858493499457836,
          "lyap1_max": -0.0070664850063622,
          "lyap_mean": -0.019219379872083664,
          "volume_mean": -0.019219379872083664,
          "volume_max": -0.006753767374902964,
          "frac_active": 0.0625
        },
        {
          "step": 107,
          "loss": 0.8180372714996338,
          "floss_loss": 0.0005435080965980887,
          "kl_loss": 0.08174937218427658,
          "lyap1_mean": -0.01604168303310871,
          "lyap1_max": -0.007525989785790443,
          "lyap_mean": -0.021415716037154198,
          "volume_mean": -0.021415716037154198,
          "volume_max": -0.011844095773994923,
          "frac_active": 0.0625
        },
        {
          "step": 108,
          "loss": 0.9941851496696472,
          "floss_loss": 0.0003572731220629066,
          "kl_loss": 0.09938278794288635,
          "lyap1_mean": -0.014435816556215286,
          "lyap1_max": -0.008267255499958992,
          "lyap_mean": -0.01817837357521057,
          "volume_mean": -0.01817837357521057,
          "volume_max": -0.015081636607646942,
          "frac_active": 0.0
        },
        {
          "step": 109,
          "loss": 0.3679830729961395,
          "floss_loss": 0.0005794291500933468,
          "kl_loss": 0.036740366369485855,
          "lyap1_mean": -0.01671035960316658,
          "lyap1_max": 0.0039918189868330956,
          "lyap_mean": -0.0219259113073349,
          "volume_mean": -0.0219259113073349,
          "volume_max": -0.014413547702133656,
          "frac_active": 0.0625
        },
        {
          "step": 110,
          "loss": 1.5796847343444824,
          "floss_loss": 0.00017242479952983558,
          "kl_loss": 0.1579512357711792,
          "lyap1_mean": 0.01754472404718399,
          "lyap1_max": 0.027539843693375587,
          "lyap_mean": -0.000738253933377564,
          "volume_mean": -0.000738253933377564,
          "volume_max": 0.0029449663124978542,
          "frac_active": 0.4375
        },
        {
          "step": 111,
          "loss": 0.5120090246200562,
          "floss_loss": 0.0002764336531981826,
          "kl_loss": 0.05117325857281685,
          "lyap1_mean": -0.009235687553882599,
          "lyap1_max": -0.0012273649917915463,
          "lyap_mean": -0.01316122617572546,
          "volume_mean": -0.01316122617572546,
          "volume_max": -0.007561469450592995,
          "frac_active": 0.0625
        },
        {
          "step": 112,
          "loss": 0.8498849272727966,
          "floss_loss": 0.00012096647697035223,
          "kl_loss": 0.08497639745473862,
          "lyap1_mean": 0.006299915257841349,
          "lyap1_max": 0.010490701533854008,
          "lyap_mean": -0.0035056809429079294,
          "volume_mean": -0.0035056809429079294,
          "volume_max": -0.0008986205793917179,
          "frac_active": 0.4375
        },
        {
          "step": 113,
          "loss": 1.5809836387634277,
          "floss_loss": 0.0001456542668165639,
          "kl_loss": 0.15808379650115967,
          "lyap1_mean": -0.004780872724950314,
          "lyap1_max": 0.007445840165019035,
          "lyap_mean": -0.009171376004815102,
          "volume_mean": -0.009171376004815102,
          "volume_max": -0.002921361941844225,
          "frac_active": 0.1875
        },
        {
          "step": 114,
          "loss": 0.8321717977523804,
          "floss_loss": 0.0001452178694307804,
          "kl_loss": 0.08320266008377075,
          "lyap1_mean": 0.00918249785900116,
          "lyap1_max": 0.022385291755199432,
          "lyap_mean": -0.003195773111656308,
          "volume_mean": -0.003195773111656308,
          "volume_max": 0.003764792811125517,
          "frac_active": 0.25
        },
        {
          "step": 115,
          "loss": 0.6596014499664307,
          "floss_loss": 0.00016930990386754274,
          "kl_loss": 0.06594321131706238,
          "lyap1_mean": 0.00010562257375568151,
          "lyap1_max": 0.010786555707454681,
          "lyap_mean": -0.007392234168946743,
          "volume_mean": -0.007392234168946743,
          "volume_max": -6.406521424651146e-05,
          "frac_active": 0.3125
        },
        {
          "step": 116,
          "loss": 0.595523476600647,
          "floss_loss": 0.0006197112961672246,
          "kl_loss": 0.059490375220775604,
          "lyap1_mean": -0.020457841455936432,
          "lyap1_max": -0.0095996567979455,
          "lyap_mean": -0.023403748869895935,
          "volume_mean": -0.023403748869895935,
          "volume_max": -0.015740934759378433,
          "frac_active": 0.0
        },
        {
          "step": 117,
          "loss": 0.9958536624908447,
          "floss_loss": 0.00022032031847629696,
          "kl_loss": 0.09956333786249161,
          "lyap1_mean": -0.005363278090953827,
          "lyap1_max": 0.00950083788484335,
          "lyap_mean": -0.010565709322690964,
          "volume_mean": -0.010565709322690964,
          "volume_max": 0.0010982556268572807,
          "frac_active": 0.1875
        },
        {
          "step": 118,
          "loss": 0.4356580078601837,
          "floss_loss": 0.0005941037088632584,
          "kl_loss": 0.04350639134645462,
          "lyap1_mean": -0.019745957106351852,
          "lyap1_max": -0.005893880967050791,
          "lyap_mean": -0.022684015333652496,
          "volume_mean": -0.022684015333652496,
          "volume_max": -0.014792423695325851,
          "frac_active": 0.0
        },
        {
          "step": 119,
          "loss": 0.47244027256965637,
          "floss_loss": 0.0008782899240031838,
          "kl_loss": 0.0471561960875988,
          "lyap1_mean": -0.023808380588889122,
          "lyap1_max": 0.00481125945225358,
          "lyap_mean": -0.026061588898301125,
          "volume_mean": -0.026061588898301125,
          "volume_max": -0.0026098957750946283,
          "frac_active": 0.0625
        },
        {
          "step": 120,
          "loss": 0.5384557843208313,
          "floss_loss": 0.0006471872329711914,
          "kl_loss": 0.05378086119890213,
          "lyap1_mean": -0.019361086189746857,
          "lyap1_max": -0.014084168709814548,
          "lyap_mean": -0.02498631924390793,
          "volume_mean": -0.02498631924390793,
          "volume_max": -0.02115437760949135,
          "frac_active": 0.0
        },
        {
          "step": 121,
          "loss": 0.9435727000236511,
          "floss_loss": 0.0008071716874837875,
          "kl_loss": 0.0942765548825264,
          "lyap1_mean": -0.02210748940706253,
          "lyap1_max": 0.0009552559931762516,
          "lyap_mean": -0.027017848566174507,
          "volume_mean": -0.027017848566174507,
          "volume_max": -0.017867371439933777,
          "frac_active": 0.0625
        },
        {
          "step": 122,
          "loss": 0.40232717990875244,
          "floss_loss": 0.0011911902111023664,
          "kl_loss": 0.040113598108291626,
          "lyap1_mean": -0.03312430530786514,
          "lyap1_max": -0.028562497347593307,
          "lyap_mean": -0.03422572836279869,
          "volume_mean": -0.03422572836279869,
          "volume_max": -0.028909437358379364,
          "frac_active": 0.0
        },
        {
          "step": 123,
          "loss": 0.48608142137527466,
          "floss_loss": 0.0008462589466944337,
          "kl_loss": 0.048523515462875366,
          "lyap1_mean": -0.024747006595134735,
          "lyap1_max": -0.013495363295078278,
          "lyap_mean": -0.024884041398763657,
          "volume_mean": -0.024884041398763657,
          "volume_max": -0.006564254406839609,
          "frac_active": 0.0625
        },
        {
          "step": 124,
          "loss": 0.7697076201438904,
          "floss_loss": 0.0008991668582893908,
          "kl_loss": 0.07688084244728088,
          "lyap1_mean": -0.027365490794181824,
          "lyap1_max": -0.0221658106893301,
          "lyap_mean": -0.029556749388575554,
          "volume_mean": -0.029556749388575554,
          "volume_max": -0.02690841443836689,
          "frac_active": 0.0
        },
        {
          "step": 125,
          "loss": 0.5652541518211365,
          "floss_loss": 0.0007212047930806875,
          "kl_loss": 0.056453291326761246,
          "lyap1_mean": -0.02433209866285324,
          "lyap1_max": -0.016377341002225876,
          "lyap_mean": -0.026074310764670372,
          "volume_mean": -0.026074310764670372,
          "volume_max": -0.02126901224255562,
          "frac_active": 0.0
        },
        {
          "step": 126,
          "loss": 1.164290428161621,
          "floss_loss": 0.00042229509563185275,
          "kl_loss": 0.11638681590557098,
          "lyap1_mean": -0.01586305722594261,
          "lyap1_max": 0.004659774247556925,
          "lyap_mean": -0.0178682804107666,
          "volume_mean": -0.0178682804107666,
          "volume_max": -0.0024828212335705757,
          "frac_active": 0.125
        },
        {
          "step": 127,
          "loss": 0.6847630739212036,
          "floss_loss": 0.0004760301089845598,
          "kl_loss": 0.06842871010303497,
          "lyap1_mean": -0.01658589579164982,
          "lyap1_max": -0.007063326425850391,
          "lyap_mean": -0.02098378725349903,
          "volume_mean": -0.02098378725349903,
          "volume_max": -0.019608259201049805,
          "frac_active": 0.0
        },
        {
          "step": 128,
          "loss": 1.071157455444336,
          "floss_loss": 0.0004479976778384298,
          "kl_loss": 0.10707094520330429,
          "lyap1_mean": -0.016329117119312286,
          "lyap1_max": -0.0032092623878270388,
          "lyap_mean": -0.01844077929854393,
          "volume_mean": -0.01844077929854393,
          "volume_max": -0.006827435456216335,
          "frac_active": 0.0
        },
        {
          "step": 129,
          "loss": 0.7536713480949402,
          "floss_loss": 0.00031074535218067467,
          "kl_loss": 0.07533606141805649,
          "lyap1_mean": -0.013365644961595535,
          "lyap1_max": -0.0025652144104242325,
          "lyap_mean": -0.014895189553499222,
          "volume_mean": -0.014895189553499222,
          "volume_max": -0.007384049706161022,
          "frac_active": 0.0
        },
        {
          "step": 130,
          "loss": 0.6820700764656067,
          "floss_loss": 0.0005520644481293857,
          "kl_loss": 0.0681518018245697,
          "lyap1_mean": 0.006755850277841091,
          "lyap1_max": 0.04644383490085602,
          "lyap_mean": -0.014437161386013031,
          "volume_mean": -0.014437161386013031,
          "volume_max": 0.0005547553300857544,
          "frac_active": 0.125
        },
        {
          "step": 131,
          "loss": 0.36295801401138306,
          "floss_loss": 0.00014837904018349946,
          "kl_loss": 0.03628096356987953,
          "lyap1_mean": 0.0035036620683968067,
          "lyap1_max": 0.020314902067184448,
          "lyap_mean": -0.007139761932194233,
          "volume_mean": -0.007139761932194233,
          "volume_max": -0.0015006072353571653,
          "frac_active": 0.1875
        },
        {
          "step": 132,
          "loss": 0.3805321753025055,
          "floss_loss": 0.0002646623470354825,
          "kl_loss": 0.03802675008773804,
          "lyap1_mean": 0.005697048269212246,
          "lyap1_max": 0.04498787596821785,
          "lyap_mean": -0.0034254686906933784,
          "volume_mean": -0.0034254686906933784,
          "volume_max": 0.019585613161325455,
          "frac_active": 0.25
        },
        {
          "step": 133,
          "loss": 0.8800635933876038,
          "floss_loss": 0.00024148212105501443,
          "kl_loss": 0.08798221498727798,
          "lyap1_mean": 0.01716766133904457,
          "lyap1_max": 0.04765094071626663,
          "lyap_mean": -0.0009711836464703083,
          "volume_mean": -0.0009711836464703083,
          "volume_max": 0.010086653754115105,
          "frac_active": 0.3125
        },
        {
          "step": 134,
          "loss": 0.5345351099967957,
          "floss_loss": 8.789791172603145e-05,
          "kl_loss": 0.053444717079401016,
          "lyap1_mean": 0.003542693331837654,
          "lyap1_max": 0.014572365209460258,
          "lyap_mean": -0.005085560493171215,
          "volume_mean": -0.005085560493171215,
          "volume_max": -0.00037632917519658804,
          "frac_active": 0.1875
        },
        {
          "step": 135,
          "loss": 0.43886181712150574,
          "floss_loss": 0.00012729677837342024,
          "kl_loss": 0.04387345165014267,
          "lyap1_mean": 0.005197330377995968,
          "lyap1_max": 0.017747744917869568,
          "lyap_mean": -0.004762069787830114,
          "volume_mean": -0.004762069787830114,
          "volume_max": 0.0009351891931146383,
          "frac_active": 0.3125
        },
        {
          "step": 136,
          "loss": 0.9254607558250427,
          "floss_loss": 6.492422107839957e-05,
          "kl_loss": 0.09253958612680435,
          "lyap1_mean": 0.0041289073415100574,
          "lyap1_max": 0.015973035246133804,
          "lyap_mean": 0.001701611210592091,
          "volume_mean": 0.001701611210592091,
          "volume_max": 0.007719089742749929,
          "frac_active": 0.5625
        },
        {
          "step": 137,
          "loss": 1.0121251344680786,
          "floss_loss": 4.4608896132558584e-05,
          "kl_loss": 0.10120805352926254,
          "lyap1_mean": 0.0024300154764205217,
          "lyap1_max": 0.010274993255734444,
          "lyap_mean": -0.002090319525450468,
          "volume_mean": -0.002090319525450468,
          "volume_max": 0.00044932332821190357,
          "frac_active": 0.375
        },
        {
          "step": 138,
          "loss": 0.7368667125701904,
          "floss_loss": 0.00013281730934977531,
          "kl_loss": 0.07367338985204697,
          "lyap1_mean": 0.01562579721212387,
          "lyap1_max": 0.028548114001750946,
          "lyap_mean": 0.004144377540796995,
          "volume_mean": 0.004144377540796995,
          "volume_max": 0.009434489533305168,
          "frac_active": 0.75
        },
        {
          "step": 139,
          "loss": 0.6670389771461487,
          "floss_loss": 0.00022155651822686195,
          "kl_loss": 0.06668174266815186,
          "lyap1_mean": -0.0068397363647818565,
          "lyap1_max": 0.003385084681212902,
          "lyap_mean": -0.011272183619439602,
          "volume_mean": -0.011272183619439602,
          "volume_max": -0.005384800955653191,
          "frac_active": 0.125
        },
        {
          "step": 140,
          "loss": 0.5649939775466919,
          "floss_loss": 0.0003837812691926956,
          "kl_loss": 0.056461017578840256,
          "lyap1_mean": -0.007853602059185505,
          "lyap1_max": 0.004816363099962473,
          "lyap_mean": -0.014508554711937904,
          "volume_mean": -0.014508554711937904,
          "volume_max": -0.0022863103076815605,
          "frac_active": 0.125
        },
        {
          "step": 141,
          "loss": 0.33933916687965393,
          "floss_loss": 0.00015776982763782144,
          "kl_loss": 0.03391813859343529,
          "lyap1_mean": -0.0040147919207811356,
          "lyap1_max": 0.007990394718945026,
          "lyap_mean": -0.010288688354194164,
          "volume_mean": -0.010288688354194164,
          "volume_max": -0.0038583348505198956,
          "frac_active": 0.0625
        },
        {
          "step": 142,
          "loss": 1.9175171852111816,
          "floss_loss": 0.00015828899631742388,
          "kl_loss": 0.1917358934879303,
          "lyap1_mean": -0.0022741009015589952,
          "lyap1_max": 0.0070230052806437016,
          "lyap_mean": -0.009749882854521275,
          "volume_mean": -0.009749882854521275,
          "volume_max": -0.005760794971138239,
          "frac_active": 0.1875
        },
        {
          "step": 143,
          "loss": 0.2803388833999634,
          "floss_loss": 0.00022851793619338423,
          "kl_loss": 0.028011037036776543,
          "lyap1_mean": -0.00042051891796290874,
          "lyap1_max": 0.022925542667508125,
          "lyap_mean": -0.009698138572275639,
          "volume_mean": -0.009698138572275639,
          "volume_max": -0.0018796236254274845,
          "frac_active": 0.125
        },
        {
          "step": 144,
          "loss": 1.0912359952926636,
          "floss_loss": 0.0003900344599969685,
          "kl_loss": 0.1090845912694931,
          "lyap1_mean": -0.007324954494833946,
          "lyap1_max": 0.021073533222079277,
          "lyap_mean": -0.010650457814335823,
          "volume_mean": -0.010650457814335823,
          "volume_max": 0.012585760094225407,
          "frac_active": 0.3125
        },
        {
          "step": 145,
          "loss": 1.1889941692352295,
          "floss_loss": 0.0005136301042512059,
          "kl_loss": 0.1188480481505394,
          "lyap1_mean": -0.015789519995450974,
          "lyap1_max": -0.004852278158068657,
          "lyap_mean": -0.02159750647842884,
          "volume_mean": -0.02159750647842884,
          "volume_max": -0.015858856961131096,
          "frac_active": 0.0
        },
        {
          "step": 146,
          "loss": 0.4806904196739197,
          "floss_loss": 0.0002997406118083745,
          "kl_loss": 0.048039067536592484,
          "lyap1_mean": -0.002393187955021858,
          "lyap1_max": 0.00945367943495512,
          "lyap_mean": -0.012972080148756504,
          "volume_mean": -0.012972080148756504,
          "volume_max": -0.007728218100965023,
          "frac_active": 0.1875
        },
        {
          "step": 147,
          "loss": 0.7936227917671204,
          "floss_loss": 0.0006892166566103697,
          "kl_loss": 0.07929335534572601,
          "lyap1_mean": -0.023042630404233932,
          "lyap1_max": -0.005059453658759594,
          "lyap_mean": -0.02373163215816021,
          "volume_mean": -0.02373163215816021,
          "volume_max": -0.005939273163676262,
          "frac_active": 0.0625
        },
        {
          "step": 148,
          "loss": 0.9108121991157532,
          "floss_loss": 0.00046905624913051724,
          "kl_loss": 0.09103431552648544,
          "lyap1_mean": -0.016904883086681366,
          "lyap1_max": 0.0012944962363690138,
          "lyap_mean": -0.01951425150036812,
          "volume_mean": -0.01951425150036812,
          "volume_max": -0.008554462343454361,
          "frac_active": 0.0625
        },
        {
          "step": 149,
          "loss": 0.8508822321891785,
          "floss_loss": 0.0004798368609044701,
          "kl_loss": 0.08504024147987366,
          "lyap1_mean": -0.014669137075543404,
          "lyap1_max": 0.00788182020187378,
          "lyap_mean": -0.019997095689177513,
          "volume_mean": -0.019997095689177513,
          "volume_max": -0.010110986419022083,
          "frac_active": 0.0625
        },
        {
          "step": 150,
          "loss": 0.4473345875740051,
          "floss_loss": 0.0006923663313500583,
          "kl_loss": 0.04466422274708748,
          "lyap1_mean": -0.022321414202451706,
          "lyap1_max": -0.016705641523003578,
          "lyap_mean": -0.025998635217547417,
          "volume_mean": -0.025998635217547417,
          "volume_max": -0.022941268980503082,
          "frac_active": 0.0
        },
        {
          "step": 151,
          "loss": 0.33880817890167236,
          "floss_loss": 0.000595199060626328,
          "kl_loss": 0.03382129594683647,
          "lyap1_mean": -0.010438844561576843,
          "lyap1_max": -0.0004609079915098846,
          "lyap_mean": -0.022645127028226852,
          "volume_mean": -0.022645127028226852,
          "volume_max": -0.017766229808330536,
          "frac_active": 0.0
        },
        {
          "step": 152,
          "loss": 0.587918758392334,
          "floss_loss": 0.0006499967421405017,
          "kl_loss": 0.058726876974105835,
          "lyap1_mean": -0.015143260359764099,
          "lyap1_max": 0.00899270735681057,
          "lyap_mean": -0.02303471788764,
          "volume_mean": -0.02303471788764,
          "volume_max": -0.0153259988874197,
          "frac_active": 0.0625
        },
        {
          "step": 153,
          "loss": 0.7088509202003479,
          "floss_loss": 0.000807223841547966,
          "kl_loss": 0.07080437242984772,
          "lyap1_mean": -0.021993327885866165,
          "lyap1_max": -0.004368113353848457,
          "lyap_mean": -0.027089495211839676,
          "volume_mean": -0.027089495211839676,
          "volume_max": -0.01964445970952511,
          "frac_active": 0.0
        },
        {
          "step": 154,
          "loss": 0.27992093563079834,
          "floss_loss": 0.0003765925648622215,
          "kl_loss": 0.027954434975981712,
          "lyap1_mean": -0.01542356051504612,
          "lyap1_max": -0.01279935147613287,
          "lyap_mean": -0.018313221633434296,
          "volume_mean": -0.018313221633434296,
          "volume_max": -0.012501629069447517,
          "frac_active": 0.0
        },
        {
          "step": 155,
          "loss": 0.5191826820373535,
          "floss_loss": 0.0007873941212892532,
          "kl_loss": 0.05183953046798706,
          "lyap1_mean": -0.0255502350628376,
          "lyap1_max": -0.017659492790699005,
          "lyap_mean": -0.02760256826877594,
          "volume_mean": -0.02760256826877594,
          "volume_max": -0.02418484166264534,
          "frac_active": 0.0
        },
        {
          "step": 156,
          "loss": 0.6687209606170654,
          "floss_loss": 0.0007486999966204166,
          "kl_loss": 0.06679722666740417,
          "lyap1_mean": -0.026543796062469482,
          "lyap1_max": -0.024373583495616913,
          "lyap_mean": -0.027282554656267166,
          "volume_mean": -0.027282554656267166,
          "volume_max": -0.026399735361337662,
          "frac_active": 0.0
        },
        {
          "step": 157,
          "loss": 0.6563464999198914,
          "floss_loss": 0.0007372575346380472,
          "kl_loss": 0.06556092202663422,
          "lyap1_mean": -0.023449193686246872,
          "lyap1_max": -0.018111633136868477,
          "lyap_mean": -0.026857972145080566,
          "volume_mean": -0.026857972145080566,
          "volume_max": -0.024425622075796127,
          "frac_active": 0.0
        },
        {
          "step": 158,
          "loss": 2.185591459274292,
          "floss_loss": 0.0006644816603511572,
          "kl_loss": 0.21849270164966583,
          "lyap1_mean": -0.026112522929906845,
          "lyap1_max": -0.010666808113455772,
          "lyap_mean": -0.024911049753427505,
          "volume_mean": -0.024911049753427505,
          "volume_max": -0.020716018974781036,
          "frac_active": 0.0
        },
        {
          "step": 159,
          "loss": 0.41618940234184265,
          "floss_loss": 0.0005727955140173435,
          "kl_loss": 0.041561659425497055,
          "lyap1_mean": -0.020258521661162376,
          "lyap1_max": -0.009901916608214378,
          "lyap_mean": -0.02227175608277321,
          "volume_mean": -0.02227175608277321,
          "volume_max": -0.008397834375500679,
          "frac_active": 0.0
        },
        {
          "step": 160,
          "loss": 0.9019971489906311,
          "floss_loss": 0.0003448095521889627,
          "kl_loss": 0.09016523510217667,
          "lyap1_mean": -0.017419882118701935,
          "lyap1_max": -0.013669587671756744,
          "lyap_mean": -0.017137164250016212,
          "volume_mean": -0.017137164250016212,
          "volume_max": -0.008744452148675919,
          "frac_active": 0.0625
        },
        {
          "step": 161,
          "loss": 0.8104014992713928,
          "floss_loss": 0.0005077913519926369,
          "kl_loss": 0.08098937571048737,
          "lyap1_mean": -0.01894434355199337,
          "lyap1_max": -0.00027967244386672974,
          "lyap_mean": -0.020326469093561172,
          "volume_mean": -0.020326469093561172,
          "volume_max": -0.004758399445563555,
          "frac_active": 0.0
        },
        {
          "step": 162,
          "loss": 0.3840024471282959,
          "floss_loss": 0.0005988263292238116,
          "kl_loss": 0.03834036365151405,
          "lyap1_mean": -0.01894463039934635,
          "lyap1_max": -0.017240114510059357,
          "lyap_mean": -0.023449745029211044,
          "volume_mean": -0.023449745029211044,
          "volume_max": -0.01651502028107643,
          "frac_active": 0.0
        },
        {
          "step": 163,
          "loss": 0.27872809767723083,
          "floss_loss": 0.0005437018116936088,
          "kl_loss": 0.02781843952834606,
          "lyap1_mean": -0.015283813700079918,
          "lyap1_max": 0.005652043968439102,
          "lyap_mean": -0.0203968808054924,
          "volume_mean": -0.0203968808054924,
          "volume_max": -0.005679820664227009,
          "frac_active": 0.0625
        },
        {
          "step": 164,
          "loss": 0.3537500500679016,
          "floss_loss": 0.0003517870500218123,
          "kl_loss": 0.035339824855327606,
          "lyap1_mean": -0.006055520381778479,
          "lyap1_max": 0.010897683911025524,
          "lyap_mean": -0.014740701764822006,
          "volume_mean": -0.014740701764822006,
          "volume_max": -0.006747634615749121,
          "frac_active": 0.125
        },
        {
          "step": 165,
          "loss": 2.841775894165039,
          "floss_loss": 0.0006778223323635757,
          "kl_loss": 0.28410980105400085,
          "lyap1_mean": -0.021375102922320366,
          "lyap1_max": -0.0051819621585309505,
          "lyap_mean": -0.024382520467042923,
          "volume_mean": -0.024382520467042923,
          "volume_max": -0.011660538613796234,
          "frac_active": 0.0
        },
        {
          "step": 166,
          "loss": 0.3813595771789551,
          "floss_loss": 0.0006180671043694019,
          "kl_loss": 0.038074150681495667,
          "lyap1_mean": -0.019534064456820488,
          "lyap1_max": -0.008934318087995052,
          "lyap_mean": -0.0231236033141613,
          "volume_mean": -0.0231236033141613,
          "volume_max": -0.017258383333683014,
          "frac_active": 0.0
        },
        {
          "step": 167,
          "loss": 0.34910720586776733,
          "floss_loss": 0.0007488114642910659,
          "kl_loss": 0.03483583778142929,
          "lyap1_mean": -0.0226249061524868,
          "lyap1_max": -0.009261436760425568,
          "lyap_mean": -0.02653423696756363,
          "volume_mean": -0.02653423696756363,
          "volume_max": -0.019393036141991615,
          "frac_active": 0.0
        },
        {
          "step": 168,
          "loss": 0.7457318305969238,
          "floss_loss": 0.0005670406972058117,
          "kl_loss": 0.07451648265123367,
          "lyap1_mean": -0.014848526567220688,
          "lyap1_max": 0.0005859497468918562,
          "lyap_mean": -0.020157955586910248,
          "volume_mean": -0.020157955586910248,
          "volume_max": -0.006714499089866877,
          "frac_active": 0.125
        },
        {
          "step": 169,
          "loss": 0.8210684061050415,
          "floss_loss": 0.00023891004093457013,
          "kl_loss": 0.08208294957876205,
          "lyap1_mean": -0.012281931936740875,
          "lyap1_max": 0.00041588657768443227,
          "lyap_mean": -0.013979388400912285,
          "volume_mean": -0.013979388400912285,
          "volume_max": -0.008627005852758884,
          "frac_active": 0.0625
        },
        {
          "step": 170,
          "loss": 0.3963629901409149,
          "floss_loss": 0.0003370645281393081,
          "kl_loss": 0.03960259258747101,
          "lyap1_mean": -0.015932941809296608,
          "lyap1_max": -0.005624609068036079,
          "lyap_mean": -0.016060981899499893,
          "volume_mean": -0.016060981899499893,
          "volume_max": -0.008797413669526577,
          "frac_active": 0.0625
        },
        {
          "step": 171,
          "loss": 0.36689719557762146,
          "floss_loss": 0.0005321885109879076,
          "kl_loss": 0.03663650155067444,
          "lyap1_mean": -0.00729221710935235,
          "lyap1_max": 0.03439183160662651,
          "lyap_mean": -0.014861728996038437,
          "volume_mean": -0.014861728996038437,
          "volume_max": 0.01028766855597496,
          "frac_active": 0.1875
        },
        {
          "step": 172,
          "loss": 0.5510153770446777,
          "floss_loss": 0.0002500773989595473,
          "kl_loss": 0.05507652834057808,
          "lyap1_mean": -0.006251214537769556,
          "lyap1_max": 0.003681120229884982,
          "lyap_mean": -0.01268488634377718,
          "volume_mean": -0.01268488634377718,
          "volume_max": -0.0030051255598664284,
          "frac_active": 0.125
        },
        {
          "step": 173,
          "loss": 0.32953783869743347,
          "floss_loss": 0.0004751498345285654,
          "kl_loss": 0.032906271517276764,
          "lyap1_mean": -0.01624877378344536,
          "lyap1_max": -0.007006638217717409,
          "lyap_mean": -0.020354751497507095,
          "volume_mean": -0.020354751497507095,
          "volume_max": -0.010189171880483627,
          "frac_active": 0.0
        },
        {
          "step": 174,
          "loss": 0.7691666483879089,
          "floss_loss": 0.0003349880571477115,
          "kl_loss": 0.07688316702842712,
          "lyap1_mean": -0.014554964378476143,
          "lyap1_max": -0.011284700594842434,
          "lyap_mean": -0.017287369817495346,
          "volume_mean": -0.017287369817495346,
          "volume_max": -0.011471664533019066,
          "frac_active": 0.0
        },
        {
          "step": 175,
          "loss": 0.32864174246788025,
          "floss_loss": 0.00038332247640937567,
          "kl_loss": 0.03282584249973297,
          "lyap1_mean": -0.014469603076577187,
          "lyap1_max": -0.003920210991054773,
          "lyap_mean": -0.01825859397649765,
          "volume_mean": -0.01825859397649765,
          "volume_max": -0.008187885396182537,
          "frac_active": 0.0
        },
        {
          "step": 176,
          "loss": 0.6055659651756287,
          "floss_loss": 0.0003424065944273025,
          "kl_loss": 0.060522351413965225,
          "lyap1_mean": -0.013157559558749199,
          "lyap1_max": -0.0004412172711454332,
          "lyap_mean": -0.01740466058254242,
          "volume_mean": -0.01740466058254242,
          "volume_max": -0.014108959585428238,
          "frac_active": 0.0
        },
        {
          "step": 177,
          "loss": 0.5683355927467346,
          "floss_loss": 0.0002653522533364594,
          "kl_loss": 0.0568070225417614,
          "lyap1_mean": -0.004184297285974026,
          "lyap1_max": 0.007428671699017286,
          "lyap_mean": -0.012514210306107998,
          "volume_mean": -0.012514210306107998,
          "volume_max": -0.004197183530777693,
          "frac_active": 0.125
        },
        {
          "step": 178,
          "loss": 0.42397668957710266,
          "floss_loss": 0.00042982609011232853,
          "kl_loss": 0.042354684323072433,
          "lyap1_mean": -0.018653297796845436,
          "lyap1_max": -0.010015619918704033,
          "lyap_mean": -0.01946527510881424,
          "volume_mean": -0.01946527510881424,
          "volume_max": -0.014385528862476349,
          "frac_active": 0.0
        },
        {
          "step": 179,
          "loss": 0.41554906964302063,
          "floss_loss": 0.0007837451994419098,
          "kl_loss": 0.0414765328168869,
          "lyap1_mean": -0.026415443047881126,
          "lyap1_max": -0.019148685038089752,
          "lyap_mean": -0.02740546315908432,
          "volume_mean": -0.02740546315908432,
          "volume_max": -0.02150164544582367,
          "frac_active": 0.0
        },
        {
          "step": 180,
          "loss": 0.527564525604248,
          "floss_loss": 0.0003470654773991555,
          "kl_loss": 0.052721742540597916,
          "lyap1_mean": -0.010494625195860863,
          "lyap1_max": 0.004021264147013426,
          "lyap_mean": -0.015865419059991837,
          "volume_mean": -0.015865419059991837,
          "volume_max": -0.008520573377609253,
          "frac_active": 0.0625
        },
        {
          "step": 181,
          "loss": 0.45603302121162415,
          "floss_loss": 0.0007024211809039116,
          "kl_loss": 0.045533061027526855,
          "lyap1_mean": -0.01674843393266201,
          "lyap1_max": 0.004341989289969206,
          "lyap_mean": -0.023245636373758316,
          "volume_mean": -0.023245636373758316,
          "volume_max": -0.01245587132871151,
          "frac_active": 0.0625
        },
        {
          "step": 182,
          "loss": 0.42526257038116455,
          "floss_loss": 0.0002955251547973603,
          "kl_loss": 0.0424967035651207,
          "lyap1_mean": -0.011342855170369148,
          "lyap1_max": 0.004488232545554638,
          "lyap_mean": -0.013331420719623566,
          "volume_mean": -0.013331420719623566,
          "volume_max": 0.0008408944122493267,
          "frac_active": 0.125
        },
        {
          "step": 183,
          "loss": 0.5560704469680786,
          "floss_loss": 0.00033871206687763333,
          "kl_loss": 0.05557316914200783,
          "lyap1_mean": -0.01229871902614832,
          "lyap1_max": 0.0007594386697746813,
          "lyap_mean": -0.016245562583208084,
          "volume_mean": -0.016245562583208084,
          "volume_max": -0.005159848369657993,
          "frac_active": 0.0625
        },
        {
          "step": 184,
          "loss": 0.6686874628067017,
          "floss_loss": 0.0003226297558285296,
          "kl_loss": 0.06683648377656937,
          "lyap1_mean": -0.017877640202641487,
          "lyap1_max": -0.011536544188857079,
          "lyap_mean": -0.016986940056085587,
          "volume_mean": -0.016986940056085587,
          "volume_max": -0.01532653532922268,
          "frac_active": 0.0
        },
        {
          "step": 185,
          "loss": 0.5673356652259827,
          "floss_loss": 0.00048617576248943806,
          "kl_loss": 0.0566849485039711,
          "lyap1_mean": -0.02376622147858143,
          "lyap1_max": -0.020062705501914024,
          "lyap_mean": -0.021708741784095764,
          "volume_mean": -0.021708741784095764,
          "volume_max": -0.018647007644176483,
          "frac_active": 0.0
        },
        {
          "step": 186,
          "loss": 0.4292152225971222,
          "floss_loss": 0.00031118851620703936,
          "kl_loss": 0.04289040341973305,
          "lyap1_mean": -0.017124434933066368,
          "lyap1_max": -0.011787899769842625,
          "lyap_mean": -0.016307178884744644,
          "volume_mean": -0.016307178884744644,
          "volume_max": -0.012127840891480446,
          "frac_active": 0.0625
        },
        {
          "step": 187,
          "loss": 0.9453495740890503,
          "floss_loss": 0.00022611340682487935,
          "kl_loss": 0.09451234340667725,
          "lyap1_mean": -0.008398381993174553,
          "lyap1_max": 0.0012733450857922435,
          "lyap_mean": -0.011952264234423637,
          "volume_mean": -0.011952264234423637,
          "volume_max": -0.0012591336853802204,
          "frac_active": 0.125
        },
        {
          "step": 188,
          "loss": 0.8520810008049011,
          "floss_loss": 0.00045387825230136514,
          "kl_loss": 0.08516271412372589,
          "lyap1_mean": -0.0035068541765213013,
          "lyap1_max": 0.021336084231734276,
          "lyap_mean": -0.015933698043227196,
          "volume_mean": -0.015933698043227196,
          "volume_max": -0.006982534658163786,
          "frac_active": 0.0625
        },
        {
          "step": 189,
          "loss": 0.2524643838405609,
          "floss_loss": 0.00023329569376073778,
          "kl_loss": 0.025223108008503914,
          "lyap1_mean": -0.01166757382452488,
          "lyap1_max": -0.006072565447539091,
          "lyap_mean": -0.012554855085909367,
          "volume_mean": -0.012554855085909367,
          "volume_max": -0.0039038192480802536,
          "frac_active": 0.0625
        },
        {
          "step": 190,
          "loss": 0.8352966904640198,
          "floss_loss": 0.0007557171629741788,
          "kl_loss": 0.08345409482717514,
          "lyap1_mean": -0.021634459495544434,
          "lyap1_max": -0.008320814929902554,
          "lyap_mean": -0.02665710635483265,
          "volume_mean": -0.02665710635483265,
          "volume_max": -0.021180761978030205,
          "frac_active": 0.0
        },
        {
          "step": 191,
          "loss": 0.37888750433921814,
          "floss_loss": 0.00040832103695720434,
          "kl_loss": 0.03784791752696037,
          "lyap1_mean": -0.01989281177520752,
          "lyap1_max": -0.01070843543857336,
          "lyap_mean": -0.018572714179754257,
          "volume_mean": -0.018572714179754257,
          "volume_max": -0.007161685265600681,
          "frac_active": 0.0
        },
        {
          "step": 192,
          "loss": 0.6177254319190979,
          "floss_loss": 0.0005886603030376136,
          "kl_loss": 0.061713676899671555,
          "lyap1_mean": -0.01572207361459732,
          "lyap1_max": 0.00715542770922184,
          "lyap_mean": -0.022053861990571022,
          "volume_mean": -0.022053861990571022,
          "volume_max": -0.014471904374659061,
          "frac_active": 0.0625
        },
        {
          "step": 193,
          "loss": 0.5658448934555054,
          "floss_loss": 0.00032008340349420905,
          "kl_loss": 0.05655248090624809,
          "lyap1_mean": -0.00860390905290842,
          "lyap1_max": 0.007962142117321491,
          "lyap_mean": -0.016346190124750137,
          "volume_mean": -0.016346190124750137,
          "volume_max": -0.010411999188363552,
          "frac_active": 0.0625
        },
        {
          "step": 194,
          "loss": 0.8366377949714661,
          "floss_loss": 0.00015358738892246038,
          "kl_loss": 0.08364842087030411,
          "lyap1_mean": -0.0009849285706877708,
          "lyap1_max": 0.011754323728382587,
          "lyap_mean": -0.008939219638705254,
          "volume_mean": -0.008939219638705254,
          "volume_max": -0.0037155109457671642,
          "frac_active": 0.1875
        },
        {
          "step": 195,
          "loss": 0.5084970593452454,
          "floss_loss": 0.0003120516485068947,
          "kl_loss": 0.05081850290298462,
          "lyap1_mean": -0.00797757226973772,
          "lyap1_max": 0.0042958869598805904,
          "lyap_mean": -0.014849722385406494,
          "volume_mean": -0.014849722385406494,
          "volume_max": -0.008498040027916431,
          "frac_active": 0.0625
        },
        {
          "step": 196,
          "loss": 0.4149430990219116,
          "floss_loss": 0.00031619088258594275,
          "kl_loss": 0.04146268963813782,
          "lyap1_mean": -0.011367438361048698,
          "lyap1_max": -0.00590104516595602,
          "lyap_mean": -0.016071490943431854,
          "volume_mean": -0.016071490943431854,
          "volume_max": -0.008985667489469051,
          "frac_active": 0.0
        },
        {
          "step": 197,
          "loss": 1.0323013067245483,
          "floss_loss": 0.00020745620713569224,
          "kl_loss": 0.10320939123630524,
          "lyap1_mean": -0.006910121068358421,
          "lyap1_max": -0.0020202009472995996,
          "lyap_mean": -0.012736824341118336,
          "volume_mean": -0.012736824341118336,
          "volume_max": -0.006542125251144171,
          "frac_active": 0.0
        },
        {
          "step": 198,
          "loss": 0.5373399257659912,
          "floss_loss": 0.00032496917992830276,
          "kl_loss": 0.0537014938890934,
          "lyap1_mean": 0.015279720537364483,
          "lyap1_max": 0.05111674964427948,
          "lyap_mean": 0.0008288912940770388,
          "volume_mean": 0.0008288912940770388,
          "volume_max": 0.015731411054730415,
          "frac_active": 0.375
        },
        {
          "step": 199,
          "loss": 0.410533607006073,
          "floss_loss": 0.00021251723228488117,
          "kl_loss": 0.041032109409570694,
          "lyap1_mean": -0.0031811033841222525,
          "lyap1_max": 0.011463379487395287,
          "lyap_mean": -0.010675648227334023,
          "volume_mean": -0.010675648227334023,
          "volume_max": -0.004221007693558931,
          "frac_active": 0.125
        },
        {
          "step": 200,
          "loss": 0.8727344274520874,
          "floss_loss": 0.00020767521345987916,
          "kl_loss": 0.087252676486969,
          "lyap1_mean": 0.0014613077510148287,
          "lyap1_max": 0.031427450478076935,
          "lyap_mean": -0.00491366907954216,
          "volume_mean": -0.00491366907954216,
          "volume_max": 0.00969646219164133,
          "frac_active": 0.3125
        },
        {
          "step": 201,
          "loss": 1.1234354972839355,
          "floss_loss": 0.0002614677941892296,
          "kl_loss": 0.11231740564107895,
          "lyap1_mean": 0.0027144872583448887,
          "lyap1_max": 0.026142699643969536,
          "lyap_mean": -0.006796086672693491,
          "volume_mean": -0.006796086672693491,
          "volume_max": 0.006543661002069712,
          "frac_active": 0.375
        },
        {
          "step": 202,
          "loss": 0.4147040843963623,
          "floss_loss": 0.00017583032604306936,
          "kl_loss": 0.04145282506942749,
          "lyap1_mean": -0.0001906518591567874,
          "lyap1_max": 0.016653213649988174,
          "lyap_mean": -0.007893244735896587,
          "volume_mean": -0.007893244735896587,
          "volume_max": -0.0015756459906697273,
          "frac_active": 0.1875
        },
        {
          "step": 203,
          "loss": 0.42231056094169617,
          "floss_loss": 0.00021784129785373807,
          "kl_loss": 0.04220927134156227,
          "lyap1_mean": -0.01139024831354618,
          "lyap1_max": -0.004839026369154453,
          "lyap_mean": -0.013160496018826962,
          "volume_mean": -0.013160496018826962,
          "volume_max": -0.008450835943222046,
          "frac_active": 0.0
        },
        {
          "step": 204,
          "loss": 0.47264939546585083,
          "floss_loss": 0.00011358683696016669,
          "kl_loss": 0.04725358262658119,
          "lyap1_mean": -0.0009192499564960599,
          "lyap1_max": 0.007940387353301048,
          "lyap_mean": -0.007248686160892248,
          "volume_mean": -0.007248686160892248,
          "volume_max": -0.0038028513081371784,
          "frac_active": 0.125
        },
        {
          "step": 205,
          "loss": 0.2633659243583679,
          "floss_loss": 0.00026885257102549076,
          "kl_loss": 0.026309708133339882,
          "lyap1_mean": -0.007307774852961302,
          "lyap1_max": 0.011378984898328781,
          "lyap_mean": -0.01239674910902977,
          "volume_mean": -0.01239674910902977,
          "volume_max": -3.221002407371998e-05,
          "frac_active": 0.125
        },
        {
          "step": 206,
          "loss": 0.6184861063957214,
          "floss_loss": 0.0006279260851442814,
          "kl_loss": 0.06178581714630127,
          "lyap1_mean": -0.017911307513713837,
          "lyap1_max": -0.01172688975930214,
          "lyap_mean": -0.02382238581776619,
          "volume_mean": -0.02382238581776619,
          "volume_max": -0.018433989956974983,
          "frac_active": 0.0
        },
        {
          "step": 207,
          "loss": 0.3239881992340088,
          "floss_loss": 0.0004556744243018329,
          "kl_loss": 0.03235325217247009,
          "lyap1_mean": -0.005278642289340496,
          "lyap1_max": 0.015004772692918777,
          "lyap_mean": -0.018378816545009613,
          "volume_mean": -0.018378816545009613,
          "volume_max": -0.009556153789162636,
          "frac_active": 0.0625
        },
        {
          "step": 208,
          "loss": 0.5735421180725098,
          "floss_loss": 0.0005865978891961277,
          "kl_loss": 0.057295553386211395,
          "lyap1_mean": -0.022780366241931915,
          "lyap1_max": -0.013934945687651634,
          "lyap_mean": -0.023852672427892685,
          "volume_mean": -0.023852672427892685,
          "volume_max": -0.018887504935264587,
          "frac_active": 0.0
        },
        {
          "step": 209,
          "loss": 0.8083720803260803,
          "floss_loss": 0.0005362945958040655,
          "kl_loss": 0.08078357577323914,
          "lyap1_mean": -0.01933741196990013,
          "lyap1_max": -0.008015918545424938,
          "lyap_mean": -0.02044125646352768,
          "volume_mean": -0.02044125646352768,
          "volume_max": -0.010956520214676857,
          "frac_active": 0.0
        },
        {
          "step": 210,
          "loss": 0.737837016582489,
          "floss_loss": 0.0009910749504342675,
          "kl_loss": 0.07368459552526474,
          "lyap1_mean": -0.027029898017644882,
          "lyap1_max": -0.021727202460169792,
          "lyap_mean": -0.031188923865556717,
          "volume_mean": -0.031188923865556717,
          "volume_max": -0.02865484543144703,
          "frac_active": 0.0
        },
        {
          "step": 211,
          "loss": 0.624527633190155,
          "floss_loss": 0.0006861921865493059,
          "kl_loss": 0.06238414719700813,
          "lyap1_mean": -0.0254967100918293,
          "lyap1_max": -0.016829444095492363,
          "lyap_mean": -0.02423972636461258,
          "volume_mean": -0.02423972636461258,
          "volume_max": -0.012032248079776764,
          "frac_active": 0.0625
        },
        {
          "step": 212,
          "loss": 0.5665832161903381,
          "floss_loss": 0.0007358808070421219,
          "kl_loss": 0.056584734469652176,
          "lyap1_mean": -0.023487381637096405,
          "lyap1_max": -0.02069821208715439,
          "lyap_mean": -0.0257769376039505,
          "volume_mean": -0.0257769376039505,
          "volume_max": -0.020886255428195,
          "frac_active": 0.0
        },
        {
          "step": 213,
          "loss": 0.807663083076477,
          "floss_loss": 0.0006733164191246033,
          "kl_loss": 0.08069898188114166,
          "lyap1_mean": -0.021042635664343834,
          "lyap1_max": -0.012921370565891266,
          "lyap_mean": -0.025459790602326393,
          "volume_mean": -0.025459790602326393,
          "volume_max": -0.02069452591240406,
          "frac_active": 0.0
        },
        {
          "step": 214,
          "loss": 0.38435426354408264,
          "floss_loss": 0.0004054142627865076,
          "kl_loss": 0.03839488700032234,
          "lyap1_mean": 0.010382188484072685,
          "lyap1_max": 0.0356815904378891,
          "lyap_mean": -0.00706417765468359,
          "volume_mean": -0.00706417765468359,
          "volume_max": 0.009431368671357632,
          "frac_active": 0.375
        },
        {
          "step": 215,
          "loss": 0.507698655128479,
          "floss_loss": 0.0005473032360896468,
          "kl_loss": 0.0507151335477829,
          "lyap1_mean": -0.008994926698505878,
          "lyap1_max": 0.011077217757701874,
          "lyap_mean": -0.016469454392790794,
          "volume_mean": -0.016469454392790794,
          "volume_max": 0.005282701924443245,
          "frac_active": 0.1875
        },
        {
          "step": 216,
          "loss": 0.7265815734863281,
          "floss_loss": 0.00043519417522475123,
          "kl_loss": 0.0726146399974823,
          "lyap1_mean": -0.0014434931799769402,
          "lyap1_max": 0.02065921761095524,
          "lyap_mean": -0.010590294376015663,
          "volume_mean": -0.010590294376015663,
          "volume_max": 6.426125764846802e-08,
          "frac_active": 0.3125
        },
        {
          "step": 217,
          "loss": 0.5558964610099792,
          "floss_loss": 0.0003152589197270572,
          "kl_loss": 0.05555812269449234,
          "lyap1_mean": -0.010479467920958996,
          "lyap1_max": 0.0089579988270998,
          "lyap_mean": -0.014914073050022125,
          "volume_mean": -0.014914073050022125,
          "volume_max": -0.006873432546854019,
          "frac_active": 0.0625
        },
        {
          "step": 218,
          "loss": 0.45372268557548523,
          "floss_loss": 0.0001528685534140095,
          "kl_loss": 0.045356981456279755,
          "lyap1_mean": -0.0014541198033839464,
          "lyap1_max": 0.006304036360234022,
          "lyap_mean": -0.008891956880688667,
          "volume_mean": -0.008891956880688667,
          "volume_max": -0.004663438536226749,
          "frac_active": 0.1875
        },
        {
          "step": 219,
          "loss": 0.7047814130783081,
          "floss_loss": 0.0002146870392607525,
          "kl_loss": 0.07045666873455048,
          "lyap1_mean": -0.00024356786161661148,
          "lyap1_max": 0.014910911209881306,
          "lyap_mean": -0.007930811494588852,
          "volume_mean": -0.007930811494588852,
          "volume_max": 0.0034120744094252586,
          "frac_active": 0.1875
        },
        {
          "step": 220,
          "loss": 0.5183896422386169,
          "floss_loss": 0.00014319244655780494,
          "kl_loss": 0.051824647933244705,
          "lyap1_mean": 0.011737879365682602,
          "lyap1_max": 0.03043030947446823,
          "lyap_mean": 0.0028168465942144394,
          "volume_mean": 0.0028168465942144394,
          "volume_max": 0.01447538286447525,
          "frac_active": 0.5
        },
        {
          "step": 221,
          "loss": 0.30077624320983887,
          "floss_loss": 0.00022595665359403938,
          "kl_loss": 0.030055029317736626,
          "lyap1_mean": 0.005646314471960068,
          "lyap1_max": 0.032690681517124176,
          "lyap_mean": 0.0026256227865815163,
          "volume_mean": 0.0026256227865815163,
          "volume_max": 0.022206291556358337,
          "frac_active": 0.5
        },
        {
          "step": 222,
          "loss": 4.598633766174316,
          "floss_loss": 0.0001647235912969336,
          "kl_loss": 0.4598469138145447,
          "lyap1_mean": 0.008608940988779068,
          "lyap1_max": 0.022882793098688126,
          "lyap_mean": 0.0013110623694956303,
          "volume_mean": 0.0013110623694956303,
          "volume_max": 0.01528577134013176,
          "frac_active": 0.4375
        },
        {
          "step": 223,
          "loss": 0.3789307177066803,
          "floss_loss": 0.0001805116917239502,
          "kl_loss": 0.03787501901388168,
          "lyap1_mean": 0.003127926029264927,
          "lyap1_max": 0.022866226732730865,
          "lyap_mean": -0.005049550905823708,
          "volume_mean": -0.005049550905823708,
          "volume_max": 0.011774308979511261,
          "frac_active": 0.3125
        },
        {
          "step": 224,
          "loss": 0.5542330145835876,
          "floss_loss": 0.00013233521895017475,
          "kl_loss": 0.05541006848216057,
          "lyap1_mean": 0.01256510242819786,
          "lyap1_max": 0.030092911794781685,
          "lyap_mean": 0.0020300820469856262,
          "volume_mean": 0.0020300820469856262,
          "volume_max": 0.007811452262103558,
          "frac_active": 0.5625
        },
        {
          "step": 225,
          "loss": 0.6251938343048096,
          "floss_loss": 0.0003437895211391151,
          "kl_loss": 0.06248500198125839,
          "lyap1_mean": 0.02225644886493683,
          "lyap1_max": 0.055319711565971375,
          "lyap_mean": 0.007359285373240709,
          "volume_mean": 0.007359285373240709,
          "volume_max": 0.026123516261577606,
          "frac_active": 0.5625
        },
        {
          "step": 226,
          "loss": 0.4269891381263733,
          "floss_loss": 0.000255105784162879,
          "kl_loss": 0.04267340153455734,
          "lyap1_mean": -0.0007587966974824667,
          "lyap1_max": 0.013351707719266415,
          "lyap_mean": -0.008518397808074951,
          "volume_mean": -0.008518397808074951,
          "volume_max": 0.006028028670698404,
          "frac_active": 0.25
        },
        {
          "step": 227,
          "loss": 0.42032456398010254,
          "floss_loss": 0.00015710049774497747,
          "kl_loss": 0.04201674833893776,
          "lyap1_mean": 0.01164926216006279,
          "lyap1_max": 0.022052232176065445,
          "lyap_mean": -0.0006499119335785508,
          "volume_mean": -0.0006499119335785508,
          "volume_max": 0.004217283334583044,
          "frac_active": 0.4375
        },
        {
          "step": 228,
          "loss": 0.4027048945426941,
          "floss_loss": 0.0002069507900159806,
          "kl_loss": 0.040249794721603394,
          "lyap1_mean": -0.008182558231055737,
          "lyap1_max": 0.0023201322183012962,
          "lyap_mean": -0.010559847578406334,
          "volume_mean": -0.010559847578406334,
          "volume_max": -0.0040884194895625114,
          "frac_active": 0.1875
        },
        {
          "step": 229,
          "loss": 0.5704668760299683,
          "floss_loss": 0.0002802862145472318,
          "kl_loss": 0.05701866373419762,
          "lyap1_mean": -0.007838426157832146,
          "lyap1_max": 0.004234404303133488,
          "lyap_mean": -0.013857070356607437,
          "volume_mean": -0.013857070356607437,
          "volume_max": -0.005965007469058037,
          "frac_active": 0.0625
        },
        {
          "step": 230,
          "loss": 0.37352901697158813,
          "floss_loss": 0.0003831502690445632,
          "kl_loss": 0.03731458634138107,
          "lyap1_mean": -0.015330775640904903,
          "lyap1_max": -0.00839723739773035,
          "lyap_mean": -0.01857972890138626,
          "volume_mean": -0.01857972890138626,
          "volume_max": -0.012505301274359226,
          "frac_active": 0.0
        },
        {
          "step": 231,
          "loss": 0.3681299090385437,
          "floss_loss": 0.0007787183276377618,
          "kl_loss": 0.03673512116074562,
          "lyap1_mean": -0.02380434051156044,
          "lyap1_max": -0.01410490833222866,
          "lyap_mean": -0.027086813002824783,
          "volume_mean": -0.027086813002824783,
          "volume_max": -0.02004084177315235,
          "frac_active": 0.0
        },
        {
          "step": 232,
          "loss": 0.6190040111541748,
          "floss_loss": 0.00036822870606556535,
          "kl_loss": 0.061863575130701065,
          "lyap1_mean": -0.01130304392427206,
          "lyap1_max": 0.0014659600565209985,
          "lyap_mean": -0.01660291664302349,
          "volume_mean": -0.01660291664302349,
          "volume_max": -0.009461496956646442,
          "frac_active": 0.125
        },
        {
          "step": 233,
          "loss": 0.5469374656677246,
          "floss_loss": 0.000476911140140146,
          "kl_loss": 0.05464605614542961,
          "lyap1_mean": -0.013588523492217064,
          "lyap1_max": 0.0027979875449091196,
          "lyap_mean": -0.01815001666545868,
          "volume_mean": -0.01815001666545868,
          "volume_max": 9.21962782740593e-06,
          "frac_active": 0.125
        },
        {
          "step": 234,
          "loss": 0.2726699411869049,
          "floss_loss": 0.0007310087094083428,
          "kl_loss": 0.027193890884518623,
          "lyap1_mean": -0.0269919466227293,
          "lyap1_max": -0.023385802283883095,
          "lyap_mean": -0.026822110638022423,
          "volume_mean": -0.026822110638022423,
          "volume_max": -0.02314959652721882,
          "frac_active": 0.0
        },
        {
          "step": 235,
          "loss": 0.5204905867576599,
          "floss_loss": 0.0002963519364129752,
          "kl_loss": 0.052019424736499786,
          "lyap1_mean": -0.008681226521730423,
          "lyap1_max": 0.0077421702444553375,
          "lyap_mean": -0.014763345941901207,
          "volume_mean": -0.014763345941901207,
          "volume_max": -0.0082384729757905,
          "frac_active": 0.0625
        },
        {
          "step": 236,
          "loss": 0.595925509929657,
          "floss_loss": 0.00040217547211796045,
          "kl_loss": 0.05955233424901962,
          "lyap1_mean": -0.01865197718143463,
          "lyap1_max": -0.01479513943195343,
          "lyap_mean": -0.016961809247732162,
          "volume_mean": -0.016961809247732162,
          "volume_max": -0.009946309030056,
          "frac_active": 0.0625
        },
        {
          "step": 237,
          "loss": 0.2421724498271942,
          "floss_loss": 0.00044197426177561283,
          "kl_loss": 0.02417304739356041,
          "lyap1_mean": -0.013717612251639366,
          "lyap1_max": -0.0050289868377149105,
          "lyap_mean": -0.019365739077329636,
          "volume_mean": -0.019365739077329636,
          "volume_max": -0.009345237165689468,
          "frac_active": 0.0
        },
        {
          "step": 238,
          "loss": 0.5987616777420044,
          "floss_loss": 0.00031196424970403314,
          "kl_loss": 0.059844970703125,
          "lyap1_mean": -0.00860711932182312,
          "lyap1_max": 0.0017424431862309575,
          "lyap_mean": -0.015275934711098671,
          "volume_mean": -0.015275934711098671,
          "volume_max": -0.009654637426137924,
          "frac_active": 0.0625
        },
        {
          "step": 239,
          "loss": 0.2926461100578308,
          "floss_loss": 0.0001349819649476558,
          "kl_loss": 0.029251113533973694,
          "lyap1_mean": -0.0024733515456318855,
          "lyap1_max": 0.023563703522086143,
          "lyap_mean": -0.005221829749643803,
          "volume_mean": -0.005221829749643803,
          "volume_max": 0.008475083857774734,
          "frac_active": 0.3125
        },
        {
          "step": 240,
          "loss": 0.4961407780647278,
          "floss_loss": 0.00011032887414330617,
          "kl_loss": 0.049603044986724854,
          "lyap1_mean": 0.0031090781558305025,
          "lyap1_max": 0.008141855709254742,
          "lyap_mean": -0.005602612625807524,
          "volume_mean": -0.005602612625807524,
          "volume_max": 0.003934412728995085,
          "frac_active": 0.375
        },
        {
          "step": 241,
          "loss": 0.5299934148788452,
          "floss_loss": 0.0004357039579190314,
          "kl_loss": 0.05295576900243759,
          "lyap1_mean": -0.007394263055175543,
          "lyap1_max": 0.014389310963451862,
          "lyap_mean": -0.013008764013648033,
          "volume_mean": -0.013008764013648033,
          "volume_max": 0.004138394258916378,
          "frac_active": 0.3125
        },
        {
          "step": 242,
          "loss": 0.3789125382900238,
          "floss_loss": 0.00013236558879725635,
          "kl_loss": 0.037878017872571945,
          "lyap1_mean": -0.007563078310340643,
          "lyap1_max": -0.004685473628342152,
          "lyap_mean": -0.008941857144236565,
          "volume_mean": -0.008941857144236565,
          "volume_max": -0.0016241016564890742,
          "frac_active": 0.125
        },
        {
          "step": 243,
          "loss": 0.3251449167728424,
          "floss_loss": 0.00011523704597493634,
          "kl_loss": 0.03250296786427498,
          "lyap1_mean": 0.004415631294250488,
          "lyap1_max": 0.015473960898816586,
          "lyap_mean": -0.0019114603055641055,
          "volume_mean": -0.0019114603055641055,
          "volume_max": 0.009527302347123623,
          "frac_active": 0.5
        },
        {
          "step": 244,
          "loss": 0.47299233078956604,
          "floss_loss": 0.00010250665218336508,
          "kl_loss": 0.04728898033499718,
          "lyap1_mean": -0.0011282041668891907,
          "lyap1_max": 0.006546097341924906,
          "lyap_mean": -0.006512971129268408,
          "volume_mean": -0.006512971129268408,
          "volume_max": -8.671311661601067e-05,
          "frac_active": 0.25
        },
        {
          "step": 245,
          "loss": 0.5294609665870667,
          "floss_loss": 0.00017738118185661733,
          "kl_loss": 0.05292835831642151,
          "lyap1_mean": 0.012034106999635696,
          "lyap1_max": 0.03108815662562847,
          "lyap_mean": 0.004033798351883888,
          "volume_mean": 0.004033798351883888,
          "volume_max": 0.01934845559298992,
          "frac_active": 0.5625
        },
        {
          "step": 246,
          "loss": 0.3350800573825836,
          "floss_loss": 7.600912067573518e-05,
          "kl_loss": 0.03350040689110756,
          "lyap1_mean": 0.0084209144115448,
          "lyap1_max": 0.023029277101159096,
          "lyap_mean": 0.000920889200642705,
          "volume_mean": 0.000920889200642705,
          "volume_max": 0.009001696482300758,
          "frac_active": 0.5
        },
        {
          "step": 247,
          "loss": 0.31853625178337097,
          "floss_loss": 8.848001016303897e-05,
          "kl_loss": 0.03184477612376213,
          "lyap1_mean": 0.0010132926981896162,
          "lyap1_max": 0.01332905888557434,
          "lyap_mean": -0.0009185983217321336,
          "volume_mean": -0.0009185983217321336,
          "volume_max": 0.010037536732852459,
          "frac_active": 0.5625
        },
        {
          "step": 248,
          "loss": 0.5210399031639099,
          "floss_loss": 0.00011428612924646586,
          "kl_loss": 0.05209256336092949,
          "lyap1_mean": -0.006165837869048119,
          "lyap1_max": -0.0011139626149088144,
          "lyap_mean": -0.009252365678548813,
          "volume_mean": -0.009252365678548813,
          "volume_max": -0.005862954538315535,
          "frac_active": 0.0
        },
        {
          "step": 249,
          "loss": 0.3758316934108734,
          "floss_loss": 0.00015429632912855595,
          "kl_loss": 0.03756774216890335,
          "lyap1_mean": 0.016170762479305267,
          "lyap1_max": 0.03486226499080658,
          "lyap_mean": 0.0013244609581306577,
          "volume_mean": 0.0013244609581306577,
          "volume_max": 0.007410106714814901,
          "frac_active": 0.4375
        },
        {
          "step": 250,
          "loss": 0.19221948087215424,
          "floss_loss": 0.00021337068756110966,
          "kl_loss": 0.019200611859560013,
          "lyap1_mean": -0.007889511995017529,
          "lyap1_max": 0.0045081027783453465,
          "lyap_mean": -0.013306157663464546,
          "volume_mean": -0.013306157663464546,
          "volume_max": -0.008105725049972534,
          "frac_active": 0.0625
        },
        {
          "step": 251,
          "loss": 0.3897864818572998,
          "floss_loss": 0.00018207862740382552,
          "kl_loss": 0.03896043822169304,
          "lyap1_mean": -0.0034990436397492886,
          "lyap1_max": 0.023212408646941185,
          "lyap_mean": -0.005018895026296377,
          "volume_mean": -0.005018895026296377,
          "volume_max": 0.01309230551123619,
          "frac_active": 0.25
        },
        {
          "step": 252,
          "loss": 0.3532037138938904,
          "floss_loss": 0.00039252161514014006,
          "kl_loss": 0.035281118005514145,
          "lyap1_mean": -0.012861092574894428,
          "lyap1_max": 0.0053575229831039906,
          "lyap_mean": -0.01672333851456642,
          "volume_mean": -0.01672333851456642,
          "volume_max": -0.007356178481131792,
          "frac_active": 0.0625
        },
        {
          "step": 253,
          "loss": 0.2376842349767685,
          "floss_loss": 0.00011479474778752774,
          "kl_loss": 0.02375694364309311,
          "lyap1_mean": -0.009051081724464893,
          "lyap1_max": -0.00422380305826664,
          "lyap_mean": -0.00915120355784893,
          "volume_mean": -0.00915120355784893,
          "volume_max": -0.005225153174251318,
          "frac_active": 0.0625
        },
        {
          "step": 254,
          "loss": 0.6471133828163147,
          "floss_loss": 0.00035378706525079906,
          "kl_loss": 0.0646759569644928,
          "lyap1_mean": -0.014996053650975227,
          "lyap1_max": -0.006221454124897718,
          "lyap_mean": -0.016209634020924568,
          "volume_mean": -0.016209634020924568,
          "volume_max": -0.0071795061230659485,
          "frac_active": 0.0625
        },
        {
          "step": 255,
          "loss": 0.4084070026874542,
          "floss_loss": 0.00021967118664178997,
          "kl_loss": 0.04081873223185539,
          "lyap1_mean": -0.010393470525741577,
          "lyap1_max": -0.004822279326617718,
          "lyap_mean": -0.012452512048184872,
          "volume_mean": -0.012452512048184872,
          "volume_max": -0.006185478996485472,
          "frac_active": 0.0625
        },
        {
          "step": 256,
          "loss": 0.6689307689666748,
          "floss_loss": 0.00014049300807528198,
          "kl_loss": 0.0668790265917778,
          "lyap1_mean": -4.6596163883805275e-05,
          "lyap1_max": 0.014863737858831882,
          "lyap_mean": -0.005771557800471783,
          "volume_mean": -0.005771557800471783,
          "volume_max": 0.0015480425208806992,
          "frac_active": 0.25
        },
        {
          "step": 257,
          "loss": 0.41673028469085693,
          "floss_loss": 0.00032494758488610387,
          "kl_loss": 0.04164053499698639,
          "lyap1_mean": -0.01567024365067482,
          "lyap1_max": -0.008697500452399254,
          "lyap_mean": -0.016884785145521164,
          "volume_mean": -0.016884785145521164,
          "volume_max": -0.008650460280478,
          "frac_active": 0.0
        },
        {
          "step": 258,
          "loss": 0.36963748931884766,
          "floss_loss": 0.00022027084196452051,
          "kl_loss": 0.03694172203540802,
          "lyap1_mean": -0.0032949955202639103,
          "lyap1_max": 0.014947056770324707,
          "lyap_mean": -0.009429455734789371,
          "volume_mean": -0.009429455734789371,
          "volume_max": 0.004661667160689831,
          "frac_active": 0.1875
        },
        {
          "step": 259,
          "loss": 0.4915042221546173,
          "floss_loss": 0.0005520302802324295,
          "kl_loss": 0.04909522086381912,
          "lyap1_mean": -0.018520114943385124,
          "lyap1_max": 0.0037260300014168024,
          "lyap_mean": -0.0212794691324234,
          "volume_mean": -0.0212794691324234,
          "volume_max": -0.00823269598186016,
          "frac_active": 0.0625
        },
        {
          "step": 260,
          "loss": 0.39007478952407837,
          "floss_loss": 0.00034359056735411286,
          "kl_loss": 0.038973119109869,
          "lyap1_mean": -0.0067610666155815125,
          "lyap1_max": 0.011658670380711555,
          "lyap_mean": -0.012043356895446777,
          "volume_mean": -0.012043356895446777,
          "volume_max": 0.008742539212107658,
          "frac_active": 0.25
        },
        {
          "step": 261,
          "loss": 0.5641282200813293,
          "floss_loss": 0.00033228856045752764,
          "kl_loss": 0.056379590183496475,
          "lyap1_mean": -0.00023639015853405,
          "lyap1_max": 0.019831830635666847,
          "lyap_mean": -0.011504337191581726,
          "volume_mean": -0.011504337191581726,
          "volume_max": 0.0033136846031993628,
          "frac_active": 0.1875
        },
        {
          "step": 262,
          "loss": 0.36187979578971863,
          "floss_loss": 0.0004976007621735334,
          "kl_loss": 0.03613821789622307,
          "lyap1_mean": -0.02134612947702408,
          "lyap1_max": -0.007815631106495857,
          "lyap_mean": -0.020689189434051514,
          "volume_mean": -0.020689189434051514,
          "volume_max": -0.009645439684391022,
          "frac_active": 0.0
        },
        {
          "step": 263,
          "loss": 0.5055248141288757,
          "floss_loss": 0.00041812064591795206,
          "kl_loss": 0.050510670989751816,
          "lyap1_mean": -0.01702246256172657,
          "lyap1_max": -0.0017593979137018323,
          "lyap_mean": -0.017622793093323708,
          "volume_mean": -0.017622793093323708,
          "volume_max": -0.0012652433943003416,
          "frac_active": 0.0
        },
        {
          "step": 264,
          "loss": 0.7827129364013672,
          "floss_loss": 0.000824332470074296,
          "kl_loss": 0.07818885892629623,
          "lyap1_mean": 0.0024625565856695175,
          "lyap1_max": 0.07417381554841995,
          "lyap_mean": -0.013565896078944206,
          "volume_mean": -0.013565896078944206,
          "volume_max": 0.011605769395828247,
          "frac_active": 0.125
        },
        {
          "step": 265,
          "loss": 0.49627137184143066,
          "floss_loss": 0.00041520490776747465,
          "kl_loss": 0.04958561807870865,
          "lyap1_mean": -0.016376111656427383,
          "lyap1_max": -0.009148616343736649,
          "lyap_mean": -0.019019270315766335,
          "volume_mean": -0.019019270315766335,
          "volume_max": -0.011362096294760704,
          "frac_active": 0.0
        },
        {
          "step": 266,
          "loss": 0.36561739444732666,
          "floss_loss": 0.0003882189921569079,
          "kl_loss": 0.036522917449474335,
          "lyap1_mean": -0.016122736036777496,
          "lyap1_max": -0.011049746535718441,
          "lyap_mean": -0.018599139526486397,
          "volume_mean": -0.018599139526486397,
          "volume_max": -0.01358184963464737,
          "frac_active": 0.0
        },
        {
          "step": 267,
          "loss": 0.3139656186103821,
          "floss_loss": 0.0003427152114454657,
          "kl_loss": 0.03136228770017624,
          "lyap1_mean": -0.01156870648264885,
          "lyap1_max": 0.00438619963824749,
          "lyap_mean": -0.013055926188826561,
          "volume_mean": -0.013055926188826561,
          "volume_max": 0.008129803463816643,
          "frac_active": 0.25
        },
        {
          "step": 268,
          "loss": 0.7473220229148865,
          "floss_loss": 0.0002165986516047269,
          "kl_loss": 0.07471054047346115,
          "lyap1_mean": -0.007156261242926121,
          "lyap1_max": 0.011524362489581108,
          "lyap_mean": -0.011507933028042316,
          "volume_mean": -0.011507933028042316,
          "volume_max": -0.0025643613189458847,
          "frac_active": 0.125
        },
        {
          "step": 269,
          "loss": 0.32864639163017273,
          "floss_loss": 0.0004938422353006899,
          "kl_loss": 0.03281525522470474,
          "lyap1_mean": -0.01917404867708683,
          "lyap1_max": -0.010575031861662865,
          "lyap_mean": -0.020399369299411774,
          "volume_mean": -0.020399369299411774,
          "volume_max": -0.010907965712249279,
          "frac_active": 0.0
        },
        {
          "step": 270,
          "loss": 0.5713704824447632,
          "floss_loss": 0.00040895678102970123,
          "kl_loss": 0.05709615349769592,
          "lyap1_mean": -0.015003938227891922,
          "lyap1_max": -0.0005199112929403782,
          "lyap_mean": -0.015309205278754234,
          "volume_mean": -0.015309205278754234,
          "volume_max": 0.003544424194842577,
          "frac_active": 0.1875
        },
        {
          "step": 271,
          "loss": 0.36189794540405273,
          "floss_loss": 0.0005847273860126734,
          "kl_loss": 0.036131322383880615,
          "lyap1_mean": -0.021238990128040314,
          "lyap1_max": -0.011333762668073177,
          "lyap_mean": -0.023209623992443085,
          "volume_mean": -0.023209623992443085,
          "volume_max": -0.014556447975337505,
          "frac_active": 0.0
        },
        {
          "step": 272,
          "loss": 0.45135030150413513,
          "floss_loss": 0.0008585879113525152,
          "kl_loss": 0.04504917189478874,
          "lyap1_mean": -0.026725701987743378,
          "lyap1_max": -0.013242281973361969,
          "lyap_mean": -0.02708587422966957,
          "volume_mean": -0.02708587422966957,
          "volume_max": -0.015788201242685318,
          "frac_active": 0.0
        },
        {
          "step": 273,
          "loss": 0.5105461478233337,
          "floss_loss": 0.000307274516671896,
          "kl_loss": 0.05102388560771942,
          "lyap1_mean": -0.014235548675060272,
          "lyap1_max": -0.005262248683720827,
          "lyap_mean": -0.014324760995805264,
          "volume_mean": -0.014324760995805264,
          "volume_max": -0.0012926575727760792,
          "frac_active": 0.125
        },
        {
          "step": 274,
          "loss": 0.4346091151237488,
          "floss_loss": 0.0004840312176384032,
          "kl_loss": 0.04341251030564308,
          "lyap1_mean": -0.018242698162794113,
          "lyap1_max": -0.00830690935254097,
          "lyap_mean": -0.020826805382966995,
          "volume_mean": -0.020826805382966995,
          "volume_max": -0.014510279521346092,
          "frac_active": 0.0
        },
        {
          "step": 275,
          "loss": 0.4086521565914154,
          "floss_loss": 0.0002393243194092065,
          "kl_loss": 0.04084128513932228,
          "lyap1_mean": -0.007751042954623699,
          "lyap1_max": 0.007404526229947805,
          "lyap_mean": -0.009529320523142815,
          "volume_mean": -0.009529320523142815,
          "volume_max": 0.0020729857496917248,
          "frac_active": 0.25
        },
        {
          "step": 276,
          "loss": 0.34771400690078735,
          "floss_loss": 0.0003497766447253525,
          "kl_loss": 0.03473642095923424,
          "lyap1_mean": -0.012627052143216133,
          "lyap1_max": -0.00020959539688192308,
          "lyap_mean": -0.0171462781727314,
          "volume_mean": -0.0171462781727314,
          "volume_max": -0.009515372104942799,
          "frac_active": 0.0
        },
        {
          "step": 277,
          "loss": 0.9761059284210205,
          "floss_loss": 0.0004145014099776745,
          "kl_loss": 0.09756914526224136,
          "lyap1_mean": -0.009872772730886936,
          "lyap1_max": -0.004342667758464813,
          "lyap_mean": -0.01846635527908802,
          "volume_mean": -0.01846635527908802,
          "volume_max": -0.010583322495222092,
          "frac_active": 0.0
        },
        {
          "step": 278,
          "loss": 0.502484917640686,
          "floss_loss": 0.00041886564576998353,
          "kl_loss": 0.050206609070301056,
          "lyap1_mean": -0.015264193527400494,
          "lyap1_max": -0.007479520980268717,
          "lyap_mean": -0.018390359356999397,
          "volume_mean": -0.018390359356999397,
          "volume_max": -0.008161039091646671,
          "frac_active": 0.0
        },
        {
          "step": 279,
          "loss": 0.35585176944732666,
          "floss_loss": 0.0003765857545658946,
          "kl_loss": 0.035547517240047455,
          "lyap1_mean": -0.014993610791862011,
          "lyap1_max": -0.0007406965596601367,
          "lyap_mean": -0.014670150354504585,
          "volume_mean": -0.014670150354504585,
          "volume_max": 0.0008680474129505455,
          "frac_active": 0.125
        },
        {
          "step": 280,
          "loss": 0.5711795687675476,
          "floss_loss": 0.00010405847569927573,
          "kl_loss": 0.05710754916071892,
          "lyap1_mean": 0.0012802209239453077,
          "lyap1_max": 0.005013709422200918,
          "lyap_mean": -0.004793879110366106,
          "volume_mean": -0.004793879110366106,
          "volume_max": 0.0068747177720069885,
          "frac_active": 0.3125
        },
        {
          "step": 281,
          "loss": 0.44822749495506287,
          "floss_loss": 0.00011613487731665373,
          "kl_loss": 0.04481113702058792,
          "lyap1_mean": -0.0013920185156166553,
          "lyap1_max": 0.004608591552823782,
          "lyap_mean": -0.007619459647685289,
          "volume_mean": -0.007619459647685289,
          "volume_max": -0.0012321751564741135,
          "frac_active": 0.25
        },
        {
          "step": 282,
          "loss": 0.1505407691001892,
          "floss_loss": 9.30473324842751e-05,
          "kl_loss": 0.015044772066175938,
          "lyap1_mean": 0.0053182137198746204,
          "lyap1_max": 0.016983266919851303,
          "lyap_mean": -0.003945795353502035,
          "volume_mean": -0.003945795353502035,
          "volume_max": 0.0042882743291556835,
          "frac_active": 0.3125
        },
        {
          "step": 283,
          "loss": 0.272164911031723,
          "floss_loss": 7.980885857250541e-05,
          "kl_loss": 0.027208510786294937,
          "lyap1_mean": -0.0044724117033183575,
          "lyap1_max": 0.013096895068883896,
          "lyap_mean": -0.0033190189860761166,
          "volume_mean": -0.0033190189860761166,
          "volume_max": 0.006106628570705652,
          "frac_active": 0.375
        },
        {
          "step": 284,
          "loss": 0.5126081705093384,
          "floss_loss": 0.0004476271278690547,
          "kl_loss": 0.05121605098247528,
          "lyap1_mean": -0.008523129858076572,
          "lyap1_max": 0.010360808111727238,
          "lyap_mean": -0.01308014988899231,
          "volume_mean": -0.01308014988899231,
          "volume_max": 0.006444574799388647,
          "frac_active": 0.1875
        },
        {
          "step": 285,
          "loss": 1.7446647882461548,
          "floss_loss": 0.00018535336130298674,
          "kl_loss": 0.1744479387998581,
          "lyap1_mean": -0.0016201343387365341,
          "lyap1_max": 0.01689258962869644,
          "lyap_mean": -0.01002252846956253,
          "volume_mean": -0.01002252846956253,
          "volume_max": -0.00427880696952343,
          "frac_active": 0.0625
        },
        {
          "step": 286,
          "loss": 0.5524482131004333,
          "floss_loss": 7.657971582375467e-05,
          "kl_loss": 0.05523716285824776,
          "lyap1_mean": -0.0027656294405460358,
          "lyap1_max": 0.00601432379335165,
          "lyap_mean": -0.00401991605758667,
          "volume_mean": -0.00401991605758667,
          "volume_max": 0.005107557401061058,
          "frac_active": 0.3125
        },
        {
          "step": 287,
          "loss": 0.4465659558773041,
          "floss_loss": 0.00011319819168420509,
          "kl_loss": 0.0446452759206295,
          "lyap1_mean": -0.004532204940915108,
          "lyap1_max": 0.0017557378159835935,
          "lyap_mean": -0.008103916421532631,
          "volume_mean": -0.008103916421532631,
          "volume_max": -0.0020942180417478085,
          "frac_active": 0.0625
        },
        {
          "step": 288,
          "loss": 0.6633797287940979,
          "floss_loss": 6.087376095820218e-05,
          "kl_loss": 0.0663318857550621,
          "lyap1_mean": -0.0017979149706661701,
          "lyap1_max": 0.007744830567389727,
          "lyap_mean": -0.0021338523365557194,
          "volume_mean": -0.0021338523365557194,
          "volume_max": 0.008084343746304512,
          "frac_active": 0.375
        },
        {
          "step": 289,
          "loss": 0.7316008806228638,
          "floss_loss": 0.00016679015243425965,
          "kl_loss": 0.07314340770244598,
          "lyap1_mean": 0.0012471526861190796,
          "lyap1_max": 0.014254032634198666,
          "lyap_mean": -0.005484194494783878,
          "volume_mean": -0.005484194494783878,
          "volume_max": 0.003156745107844472,
          "frac_active": 0.3125
        },
        {
          "step": 290,
          "loss": 0.42803168296813965,
          "floss_loss": 0.00015017000259831548,
          "kl_loss": 0.042788151651620865,
          "lyap1_mean": -0.0048972624354064465,
          "lyap1_max": 0.010133213363587856,
          "lyap_mean": -0.007794644683599472,
          "volume_mean": -0.007794644683599472,
          "volume_max": 0.0011439104564487934,
          "frac_active": 0.25
        },
        {
          "step": 291,
          "loss": 0.5228142142295837,
          "floss_loss": 0.0001407324889441952,
          "kl_loss": 0.052267346531152725,
          "lyap1_mean": -0.0025080495979636908,
          "lyap1_max": 0.017120156437158585,
          "lyap_mean": -0.006682178936898708,
          "volume_mean": -0.006682178936898708,
          "volume_max": 0.006363367196172476,
          "frac_active": 0.1875
        },
        {
          "step": 292,
          "loss": 1.7381398677825928,
          "floss_loss": 0.0003677014319691807,
          "kl_loss": 0.17377720773220062,
          "lyap1_mean": -0.007509087212383747,
          "lyap1_max": 0.020836343988776207,
          "lyap_mean": -0.012073458172380924,
          "volume_mean": -0.012073458172380924,
          "volume_max": 0.007264645770192146,
          "frac_active": 0.1875
        },
        {
          "step": 293,
          "loss": 0.3936561346054077,
          "floss_loss": 0.0005752578726969659,
          "kl_loss": 0.0393080897629261,
          "lyap1_mean": -0.022321362048387527,
          "lyap1_max": -0.015501904301345348,
          "lyap_mean": -0.02271544560790062,
          "volume_mean": -0.02271544560790062,
          "volume_max": -0.015957124531269073,
          "frac_active": 0.0625
        },
        {
          "step": 294,
          "loss": 0.3131170868873596,
          "floss_loss": 0.0005905969301238656,
          "kl_loss": 0.031252648681402206,
          "lyap1_mean": -0.020867642015218735,
          "lyap1_max": -0.015912942588329315,
          "lyap_mean": -0.02390526980161667,
          "volume_mean": -0.02390526980161667,
          "volume_max": -0.020562736317515373,
          "frac_active": 0.0
        },
        {
          "step": 295,
          "loss": 0.38598519563674927,
          "floss_loss": 0.0003589040134102106,
          "kl_loss": 0.038562629371881485,
          "lyap1_mean": -0.008946014568209648,
          "lyap1_max": 0.0023537748493254185,
          "lyap_mean": -0.017283152788877487,
          "volume_mean": -0.017283152788877487,
          "volume_max": -0.013875291682779789,
          "frac_active": 0.0625
        },
        {
          "step": 296,
          "loss": 0.5605280995368958,
          "floss_loss": 0.0004630527109839022,
          "kl_loss": 0.056006502360105515,
          "lyap1_mean": -0.014420544728636742,
          "lyap1_max": 0.006299963220953941,
          "lyap_mean": -0.01765810139477253,
          "volume_mean": -0.01765810139477253,
          "volume_max": -0.0011171936057507992,
          "frac_active": 0.0625
        },
        {
          "step": 297,
          "loss": 0.5916703343391418,
          "floss_loss": 0.0005801584338769317,
          "kl_loss": 0.05910901725292206,
          "lyap1_mean": -0.02311447635293007,
          "lyap1_max": -0.013843505643308163,
          "lyap_mean": -0.022764377295970917,
          "volume_mean": -0.022764377295970917,
          "volume_max": -0.011303947307169437,
          "frac_active": 0.0625
        },
        {
          "step": 298,
          "loss": 0.3993956446647644,
          "floss_loss": 0.0003775967052206397,
          "kl_loss": 0.03990180417895317,
          "lyap1_mean": -0.013938555493950844,
          "lyap1_max": -0.008456823416054249,
          "lyap_mean": -0.016451001167297363,
          "volume_mean": -0.016451001167297363,
          "volume_max": -0.007100462447851896,
          "frac_active": 0.0625
        },
        {
          "step": 299,
          "loss": 0.48312315344810486,
          "floss_loss": 0.00029349687974900007,
          "kl_loss": 0.048282966017723083,
          "lyap1_mean": -0.0125818382948637,
          "lyap1_max": -0.006136734504252672,
          "lyap_mean": -0.01639089360833168,
          "volume_mean": -0.01639089360833168,
          "volume_max": -0.012956425547599792,
          "frac_active": 0.0
        },
        {
          "step": 300,
          "loss": 0.5229504108428955,
          "floss_loss": 0.00039740646025165915,
          "kl_loss": 0.0522553026676178,
          "lyap1_mean": -0.014153690077364445,
          "lyap1_max": -0.008587395772337914,
          "lyap_mean": -0.01904495432972908,
          "volume_mean": -0.01904495432972908,
          "volume_max": -0.015051068738102913,
          "frac_active": 0.0
        },
        {
          "step": 301,
          "loss": 0.2959088981151581,
          "floss_loss": 0.0002720084157772362,
          "kl_loss": 0.02956368960440159,
          "lyap1_mean": -0.010174287483096123,
          "lyap1_max": -0.004856438841670752,
          "lyap_mean": -0.014691703952848911,
          "volume_mean": -0.014691703952848911,
          "volume_max": -0.0058443802408874035,
          "frac_active": 0.0625
        },
        {
          "step": 302,
          "loss": 0.6035503149032593,
          "floss_loss": 0.0002689779212232679,
          "kl_loss": 0.060328129678964615,
          "lyap1_mean": -0.012865986675024033,
          "lyap1_max": 0.0023820986971259117,
          "lyap_mean": -0.012863213196396828,
          "volume_mean": -0.012863213196396828,
          "volume_max": -0.00026855990290641785,
          "frac_active": 0.1875
        },
        {
          "step": 303,
          "loss": 0.4312373399734497,
          "floss_loss": 0.00016582188254687935,
          "kl_loss": 0.04310715198516846,
          "lyap1_mean": -0.0027575402054935694,
          "lyap1_max": 0.0012577692978084087,
          "lyap_mean": -0.010372649878263474,
          "volume_mean": -0.010372649878263474,
          "volume_max": -0.0071425652131438255,
          "frac_active": 0.125
        },
        {
          "step": 304,
          "loss": 0.49033626914024353,
          "floss_loss": 0.00011570868809940293,
          "kl_loss": 0.04902205616235733,
          "lyap1_mean": -0.0003655324690043926,
          "lyap1_max": 0.01393661368638277,
          "lyap_mean": -0.006790123879909515,
          "volume_mean": -0.006790123879909515,
          "volume_max": -0.002720059361308813,
          "frac_active": 0.125
        },
        {
          "step": 305,
          "loss": 0.45371443033218384,
          "floss_loss": 0.00017180069698952138,
          "kl_loss": 0.04535426199436188,
          "lyap1_mean": -0.01045939326286316,
          "lyap1_max": -0.0002519334084354341,
          "lyap_mean": -0.010550113394856453,
          "volume_mean": -0.010550113394856453,
          "volume_max": -0.004900405183434486,
          "frac_active": 0.125
        },
        {
          "step": 306,
          "loss": 0.3055846393108368,
          "floss_loss": 5.722791684092954e-05,
          "kl_loss": 0.030552741140127182,
          "lyap1_mean": -0.0012788381427526474,
          "lyap1_max": 0.004031535238027573,
          "lyap_mean": -0.004242702387273312,
          "volume_mean": -0.004242702387273312,
          "volume_max": -2.8032343834638596e-05,
          "frac_active": 0.25
        },
        {
          "step": 307,
          "loss": 0.2990646958351135,
          "floss_loss": 9.230861905962229e-05,
          "kl_loss": 0.029897239059209824,
          "lyap1_mean": -0.002349275629967451,
          "lyap1_max": 0.006679808255285025,
          "lyap_mean": -0.007854584604501724,
          "volume_mean": -0.007854584604501724,
          "volume_max": -0.005742591805756092,
          "frac_active": 0.0625
        },
        {
          "step": 308,
          "loss": 0.31748414039611816,
          "floss_loss": 0.0004451357526704669,
          "kl_loss": 0.031703900545835495,
          "lyap1_mean": -0.015878966078162193,
          "lyap1_max": -0.0007180860266089439,
          "lyap_mean": -0.017223630100488663,
          "volume_mean": -0.017223630100488663,
          "volume_max": -0.004267914220690727,
          "frac_active": 0.0
        },
        {
          "step": 309,
          "loss": 0.43578585982322693,
          "floss_loss": 0.00010585266863927245,
          "kl_loss": 0.043568000197410583,
          "lyap1_mean": 0.004888865165412426,
          "lyap1_max": 0.02315388061106205,
          "lyap_mean": -0.001348278601653874,
          "volume_mean": -0.001348278601653874,
          "volume_max": 0.010677482932806015,
          "frac_active": 0.4375
        },
        {
          "step": 310,
          "loss": 0.27714425325393677,
          "floss_loss": 6.353190110530704e-05,
          "kl_loss": 0.02770807221531868,
          "lyap1_mean": -0.0029503651894629,
          "lyap1_max": 0.0017075072973966599,
          "lyap_mean": -0.006423529703170061,
          "volume_mean": -0.006423529703170061,
          "volume_max": -0.0038572053890675306,
          "frac_active": 0.0625
        },
        {
          "step": 311,
          "loss": 0.34745603799819946,
          "floss_loss": 6.224923708941787e-05,
          "kl_loss": 0.034739378839731216,
          "lyap1_mean": 0.0001928380224853754,
          "lyap1_max": 0.007817295379936695,
          "lyap_mean": -0.004804263357073069,
          "volume_mean": -0.004804263357073069,
          "volume_max": 0.0033245335798710585,
          "frac_active": 0.25
        },
        {
          "step": 312,
          "loss": 0.32596778869628906,
          "floss_loss": 0.00020740080799441785,
          "kl_loss": 0.03257603943347931,
          "lyap1_mean": -0.002015656791627407,
          "lyap1_max": 0.0299298744648695,
          "lyap_mean": -0.007362642325460911,
          "volume_mean": -0.007362642325460911,
          "volume_max": 0.008216685615479946,
          "frac_active": 0.1875
        },
        {
          "step": 313,
          "loss": 0.5596954822540283,
          "floss_loss": 9.498442523181438e-05,
          "kl_loss": 0.055960047990083694,
          "lyap1_mean": 0.01389373280107975,
          "lyap1_max": 0.022321075201034546,
          "lyap_mean": 0.0047467295080423355,
          "volume_mean": 0.0047467295080423355,
          "volume_max": 0.010228309780359268,
          "frac_active": 0.6875
        },
        {
          "step": 314,
          "loss": 0.25734570622444153,
          "floss_loss": 0.0003114977735094726,
          "kl_loss": 0.025703420862555504,
          "lyap1_mean": -0.001795826479792595,
          "lyap1_max": 0.015775587409734726,
          "lyap_mean": -0.005781858693808317,
          "volume_mean": -0.005781858693808317,
          "volume_max": 0.01208958774805069,
          "frac_active": 0.5
        },
        {
          "step": 315,
          "loss": 0.2968815863132477,
          "floss_loss": 0.00011132345389341936,
          "kl_loss": 0.029677025973796844,
          "lyap1_mean": 0.00946042500436306,
          "lyap1_max": 0.020068423822522163,
          "lyap_mean": 0.0020587684120982885,
          "volume_mean": 0.0020587684120982885,
          "volume_max": 0.012008186429738998,
          "frac_active": 0.5625
        },
        {
          "step": 316,
          "loss": 0.45330074429512024,
          "floss_loss": 0.00016302896256092936,
          "kl_loss": 0.045313771814107895,
          "lyap1_mean": 0.013766963966190815,
          "lyap1_max": 0.047688547521829605,
          "lyap_mean": 0.004980414640158415,
          "volume_mean": 0.004980414640158415,
          "volume_max": 0.013851371593773365,
          "frac_active": 0.75
        },
        {
          "step": 317,
          "loss": 0.126138836145401,
          "floss_loss": 0.00014744601503480226,
          "kl_loss": 0.012599138543009758,
          "lyap1_mean": -0.003993631806224585,
          "lyap1_max": 0.02207046002149582,
          "lyap_mean": -0.005772911012172699,
          "volume_mean": -0.005772911012172699,
          "volume_max": 0.0077833617106080055,
          "frac_active": 0.1875
        },
        {
          "step": 318,
          "loss": 0.38263413310050964,
          "floss_loss": 6.637290789512917e-05,
          "kl_loss": 0.038256775587797165,
          "lyap1_mean": -0.004855269566178322,
          "lyap1_max": 0.005933132488280535,
          "lyap_mean": -0.00564933568239212,
          "volume_mean": -0.00564933568239212,
          "volume_max": 0.0024969372898340225,
          "frac_active": 0.1875
        },
        {
          "step": 319,
          "loss": 0.255239874124527,
          "floss_loss": 5.428229633253068e-05,
          "kl_loss": 0.02551855891942978,
          "lyap1_mean": 0.0035391170531511307,
          "lyap1_max": 0.010751914232969284,
          "lyap_mean": -0.0010554904583841562,
          "volume_mean": -0.0010554904583841562,
          "volume_max": 0.008847453631460667,
          "frac_active": 0.375
        },
        {
          "step": 320,
          "loss": 0.4262007474899292,
          "floss_loss": 6.18199264863506e-05,
          "kl_loss": 0.04261389374732971,
          "lyap1_mean": -0.0016307334881275892,
          "lyap1_max": 0.00266513810493052,
          "lyap_mean": -0.003994125872850418,
          "volume_mean": -0.003994125872850418,
          "volume_max": 0.00012038229033350945,
          "frac_active": 0.3125
        },
        {
          "step": 321,
          "loss": 0.5208457708358765,
          "floss_loss": 0.0005227660294622183,
          "kl_loss": 0.05203229561448097,
          "lyap1_mean": -0.017023101449012756,
          "lyap1_max": -0.009886316023766994,
          "lyap_mean": -0.021123846992850304,
          "volume_mean": -0.021123846992850304,
          "volume_max": -0.014601198956370354,
          "frac_active": 0.0
        },
        {
          "step": 322,
          "loss": 0.41418978571891785,
          "floss_loss": 0.00032757758162915707,
          "kl_loss": 0.04138622060418129,
          "lyap1_mean": -0.010282093659043312,
          "lyap1_max": 0.001331425504758954,
          "lyap_mean": -0.01666850410401821,
          "volume_mean": -0.01666850410401821,
          "volume_max": -0.013767591677606106,
          "frac_active": 0.0625
        },
        {
          "step": 323,
          "loss": 0.3522178530693054,
          "floss_loss": 0.0006851622019894421,
          "kl_loss": 0.03515326976776123,
          "lyap1_mean": -0.023011427372694016,
          "lyap1_max": -0.01425462681800127,
          "lyap_mean": -0.02472037263214588,
          "volume_mean": -0.02472037263214588,
          "volume_max": -0.013508565723896027,
          "frac_active": 0.0
        },
        {
          "step": 324,
          "loss": 0.44449660181999207,
          "floss_loss": 0.00023951588082127273,
          "kl_loss": 0.044425707310438156,
          "lyap1_mean": -0.008525039069354534,
          "lyap1_max": -0.003212370676919818,
          "lyap_mean": -0.014324034564197063,
          "volume_mean": -0.014324034564197063,
          "volume_max": -0.012998723424971104,
          "frac_active": 0.0
        },
        {
          "step": 325,
          "loss": 0.3986687660217285,
          "floss_loss": 0.00024961354210972786,
          "kl_loss": 0.0398419126868248,
          "lyap1_mean": -0.008308812975883484,
          "lyap1_max": 0.0011588287306949496,
          "lyap_mean": -0.013534288853406906,
          "volume_mean": -0.013534288853406906,
          "volume_max": -0.003444477217271924,
          "frac_active": 0.0625
        },
        {
          "step": 326,
          "loss": 0.3034186065196991,
          "floss_loss": 0.0002717385650612414,
          "kl_loss": 0.03031468763947487,
          "lyap1_mean": -0.012350913137197495,
          "lyap1_max": -0.0002174063993152231,
          "lyap_mean": -0.013991966843605042,
          "volume_mean": -0.013991966843605042,
          "volume_max": -0.004965954925864935,
          "frac_active": 0.0625
        },
        {
          "step": 327,
          "loss": 0.2627478241920471,
          "floss_loss": 0.00021042124717496336,
          "kl_loss": 0.02625373937189579,
          "lyap1_mean": -0.005331863649189472,
          "lyap1_max": 0.012969698756933212,
          "lyap_mean": -0.01014961488544941,
          "volume_mean": -0.01014961488544941,
          "volume_max": 0.005411739926785231,
          "frac_active": 0.1875
        },
        {
          "step": 328,
          "loss": 0.395489364862442,
          "floss_loss": 0.0004768449580296874,
          "kl_loss": 0.03950125351548195,
          "lyap1_mean": -0.010728594847023487,
          "lyap1_max": 0.0017836615443229675,
          "lyap_mean": -0.01756938360631466,
          "volume_mean": -0.01756938360631466,
          "volume_max": -0.001342272269539535,
          "frac_active": 0.125
        },
        {
          "step": 329,
          "loss": 0.4663647413253784,
          "floss_loss": 0.00044990464812144637,
          "kl_loss": 0.04659148305654526,
          "lyap1_mean": -0.014196297153830528,
          "lyap1_max": -0.002285132184624672,
          "lyap_mean": -0.015080882236361504,
          "volume_mean": -0.015080882236361504,
          "volume_max": -0.0030904426239430904,
          "frac_active": 0.125
        },
        {
          "step": 330,
          "loss": 0.25810644030570984,
          "floss_loss": 0.00022033200366422534,
          "kl_loss": 0.02578861266374588,
          "lyap1_mean": -0.011086471378803253,
          "lyap1_max": -0.004677999299019575,
          "lyap_mean": -0.013003014959394932,
          "volume_mean": -0.013003014959394932,
          "volume_max": -0.007264935877174139,
          "frac_active": 0.0625
        },
        {
          "step": 331,
          "loss": 0.3108072578907013,
          "floss_loss": 9.524922643322498e-05,
          "kl_loss": 0.031071200966835022,
          "lyap1_mean": -0.004166482016444206,
          "lyap1_max": 0.010477500036358833,
          "lyap_mean": -0.0058904243633151054,
          "volume_mean": -0.0058904243633151054,
          "volume_max": -0.000547698000445962,
          "frac_active": 0.1875
        },
        {
          "step": 332,
          "loss": 0.3906654715538025,
          "floss_loss": 9.243814565706998e-05,
          "kl_loss": 0.039057303220033646,
          "lyap1_mean": -0.0029214199166744947,
          "lyap1_max": 0.0034247732255607843,
          "lyap_mean": -0.008023800328373909,
          "volume_mean": -0.008023800328373909,
          "volume_max": -0.006139547564089298,
          "frac_active": 0.125
        },
        {
          "step": 333,
          "loss": 0.2617241442203522,
          "floss_loss": 0.0002111676731146872,
          "kl_loss": 0.026151297613978386,
          "lyap1_mean": -0.008681226521730423,
          "lyap1_max": -0.000749658967833966,
          "lyap_mean": -0.013268468901515007,
          "volume_mean": -0.013268468901515007,
          "volume_max": -0.006146138999611139,
          "frac_active": 0.0
        },
        {
          "step": 334,
          "loss": 0.6067808866500854,
          "floss_loss": 8.567694021621719e-05,
          "kl_loss": 0.06066952273249626,
          "lyap1_mean": 0.000334419310092926,
          "lyap1_max": 0.005489064380526543,
          "lyap_mean": -0.002557056024670601,
          "volume_mean": -0.002557056024670601,
          "volume_max": 0.006961523089557886,
          "frac_active": 0.5
        },
        {
          "step": 335,
          "loss": 0.443209171295166,
          "floss_loss": 7.974606705829501e-05,
          "kl_loss": 0.044312942773103714,
          "lyap1_mean": 0.00812077522277832,
          "lyap1_max": 0.01866457238793373,
          "lyap_mean": 0.005598903633654118,
          "volume_mean": 0.005598903633654118,
          "volume_max": 0.014472326263785362,
          "frac_active": 0.6875
        },
        {
          "step": 336,
          "loss": 0.44574007391929626,
          "floss_loss": 0.00011758936307160184,
          "kl_loss": 0.04456224665045738,
          "lyap1_mean": 0.005020708777010441,
          "lyap1_max": 0.015491227619349957,
          "lyap_mean": 0.0014138603582978249,
          "volume_mean": 0.0014138603582978249,
          "volume_max": 0.015357987955212593,
          "frac_active": 0.5625
        },
        {
          "step": 337,
          "loss": 0.630853533744812,
          "floss_loss": 0.00043786439346149564,
          "kl_loss": 0.0630415678024292,
          "lyap1_mean": 0.008225426077842712,
          "lyap1_max": 0.033511172980070114,
          "lyap_mean": 0.006304027512669563,
          "volume_mean": 0.006304027512669563,
          "volume_max": 0.025043517351150513,
          "frac_active": 0.75
        },
        {
          "step": 338,
          "loss": 0.5575039982795715,
          "floss_loss": 0.0005038612871430814,
          "kl_loss": 0.055700015276670456,
          "lyap1_mean": 0.03542459383606911,
          "lyap1_max": 0.0428546741604805,
          "lyap_mean": 0.018625780940055847,
          "volume_mean": 0.018625780940055847,
          "volume_max": 0.026400525122880936,
          "frac_active": 1.0
        },
        {
          "step": 339,
          "loss": 0.3020988702774048,
          "floss_loss": 0.0004984157276339829,
          "kl_loss": 0.030160045251250267,
          "lyap1_mean": 0.0286547988653183,
          "lyap1_max": 0.06164611130952835,
          "lyap_mean": 0.014990748837590218,
          "volume_mean": 0.014990748837590218,
          "volume_max": 0.03550456091761589,
          "frac_active": 0.9375
        },
        {
          "step": 340,
          "loss": 0.38232526183128357,
          "floss_loss": 0.0004372237017378211,
          "kl_loss": 0.03818880394101143,
          "lyap1_mean": 0.024269219487905502,
          "lyap1_max": 0.048991523683071136,
          "lyap_mean": 0.016302920877933502,
          "volume_mean": 0.016302920877933502,
          "volume_max": 0.0246209055185318,
          "frac_active": 0.9375
        },
        {
          "step": 341,
          "loss": 0.5382483601570129,
          "floss_loss": 0.00015234193415381014,
          "kl_loss": 0.053809601813554764,
          "lyap1_mean": 0.015568080358207226,
          "lyap1_max": 0.02393432706594467,
          "lyap_mean": 0.01018429547548294,
          "volume_mean": 0.01018429547548294,
          "volume_max": 0.016894672065973282,
          "frac_active": 0.875
        },
        {
          "step": 342,
          "loss": 0.3293866515159607,
          "floss_loss": 0.0001815204304875806,
          "kl_loss": 0.032920513302087784,
          "lyap1_mean": 0.016697749495506287,
          "lyap1_max": 0.0405404306948185,
          "lyap_mean": 0.00681519228965044,
          "volume_mean": 0.00681519228965044,
          "volume_max": 0.015663504600524902,
          "frac_active": 0.6875
        },
        {
          "step": 343,
          "loss": 0.3816758394241333,
          "floss_loss": 0.00023484291159547865,
          "kl_loss": 0.038144100457429886,
          "lyap1_mean": 0.012152303010225296,
          "lyap1_max": 0.031805604696273804,
          "lyap_mean": 0.007002498023211956,
          "volume_mean": 0.007002498023211956,
          "volume_max": 0.01865403540432453,
          "frac_active": 0.625
        },
        {
          "step": 344,
          "loss": 0.5794788599014282,
          "floss_loss": 0.0001981069945031777,
          "kl_loss": 0.05792807415127754,
          "lyap1_mean": 0.017371537163853645,
          "lyap1_max": 0.04181034117937088,
          "lyap_mean": 0.006247733719646931,
          "volume_mean": 0.006247733719646931,
          "volume_max": 0.01778554357588291,
          "frac_active": 0.75
        },
        {
          "step": 345,
          "loss": 1.5661686658859253,
          "floss_loss": 0.00010613849735818803,
          "kl_loss": 0.1566062569618225,
          "lyap1_mean": 0.0029219461139291525,
          "lyap1_max": 0.020172322168946266,
          "lyap_mean": -0.003914747387170792,
          "volume_mean": -0.003914747387170792,
          "volume_max": 0.007200019434094429,
          "frac_active": 0.3125
        },
        {
          "step": 346,
          "loss": 0.29391688108444214,
          "floss_loss": 8.085911395028234e-05,
          "kl_loss": 0.029383601620793343,
          "lyap1_mean": -0.0007816110737621784,
          "lyap1_max": 0.008001357316970825,
          "lyap_mean": -0.006155918352305889,
          "volume_mean": -0.006155918352305889,
          "volume_max": -0.001533009810373187,
          "frac_active": 0.25
        },
        {
          "step": 347,
          "loss": 0.3813117444515228,
          "floss_loss": 0.00023753984714858234,
          "kl_loss": 0.03810741752386093,
          "lyap1_mean": -0.0029413390439003706,
          "lyap1_max": 0.01691339910030365,
          "lyap_mean": -0.009121985174715519,
          "volume_mean": -0.009121985174715519,
          "volume_max": 0.00341851357370615,
          "frac_active": 0.25
        },
        {
          "step": 348,
          "loss": 0.5053761601448059,
          "floss_loss": 0.0002517011307645589,
          "kl_loss": 0.05051244795322418,
          "lyap1_mean": -0.0063909003511071205,
          "lyap1_max": 0.00912468135356903,
          "lyap_mean": -0.012651778757572174,
          "volume_mean": -0.012651778757572174,
          "volume_max": -0.006315608508884907,
          "frac_active": 0.125
        },
        {
          "step": 349,
          "loss": 0.4082051217556,
          "floss_loss": 0.0005922701093368232,
          "kl_loss": 0.04076128453016281,
          "lyap1_mean": -0.016814567148685455,
          "lyap1_max": -0.0017880756640806794,
          "lyap_mean": -0.022420966997742653,
          "volume_mean": -0.022420966997742653,
          "volume_max": -0.016257058829069138,
          "frac_active": 0.0
        },
        {
          "step": 350,
          "loss": 0.41503679752349854,
          "floss_loss": 0.0002772646548692137,
          "kl_loss": 0.04147595539689064,
          "lyap1_mean": -0.004755067173391581,
          "lyap1_max": 0.02269517257809639,
          "lyap_mean": -0.006266927812248468,
          "volume_mean": -0.006266927812248468,
          "volume_max": 0.01580912433564663,
          "frac_active": 0.3125
        },
        {
          "step": 351,
          "loss": 0.2978830635547638,
          "floss_loss": 0.0003309031017124653,
          "kl_loss": 0.02975521795451641,
          "lyap1_mean": -0.014392782002687454,
          "lyap1_max": -0.00476779555901885,
          "lyap_mean": -0.015196582302451134,
          "volume_mean": -0.015196582302451134,
          "volume_max": -0.006183868274092674,
          "frac_active": 0.0625
        },
        {
          "step": 352,
          "loss": 0.6799109578132629,
          "floss_loss": 0.00020319211762398481,
          "kl_loss": 0.06797077506780624,
          "lyap1_mean": -0.00533348647877574,
          "lyap1_max": 0.009920557960867882,
          "lyap_mean": -0.011608390137553215,
          "volume_mean": -0.011608390137553215,
          "volume_max": -0.004025349859148264,
          "frac_active": 0.0625
        },
        {
          "step": 353,
          "loss": 0.5515011548995972,
          "floss_loss": 0.00030928224441595376,
          "kl_loss": 0.05511918663978577,
          "lyap1_mean": -0.015359463170170784,
          "lyap1_max": -0.011528149247169495,
          "lyap_mean": -0.016413763165473938,
          "volume_mean": -0.016413763165473938,
          "volume_max": -0.012686631642282009,
          "frac_active": 0.0
        },
        {
          "step": 354,
          "loss": 0.5708072185516357,
          "floss_loss": 0.00042377045610919595,
          "kl_loss": 0.05703834444284439,
          "lyap1_mean": -0.015431576408445835,
          "lyap1_max": -0.005833367817103863,
          "lyap_mean": -0.019474701955914497,
          "volume_mean": -0.019474701955914497,
          "volume_max": -0.015256292186677456,
          "frac_active": 0.0
        },
        {
          "step": 355,
          "loss": 0.4077112376689911,
          "floss_loss": 0.00015585498476866633,
          "kl_loss": 0.040755536407232285,
          "lyap1_mean": -0.00556524470448494,
          "lyap1_max": -0.0001856568269431591,
          "lyap_mean": -0.010500423610210419,
          "volume_mean": -0.010500423610210419,
          "volume_max": -0.005371317267417908,
          "frac_active": 0.0
        },
        {
          "step": 356,
          "loss": 3.0344784259796143,
          "floss_loss": 0.0001704151218291372,
          "kl_loss": 0.30343079566955566,
          "lyap1_mean": -0.006373400799930096,
          "lyap1_max": 0.002459740499034524,
          "lyap_mean": -0.010686412453651428,
          "volume_mean": -0.010686412453651428,
          "volume_max": -0.006526274606585503,
          "frac_active": 0.0625
        },
        {
          "step": 357,
          "loss": 3.247894763946533,
          "floss_loss": 0.00010437321907375008,
          "kl_loss": 0.32477903366088867,
          "lyap1_mean": -0.0017657262505963445,
          "lyap1_max": 0.006367083173245192,
          "lyap_mean": -0.006859786342829466,
          "volume_mean": -0.006859786342829466,
          "volume_max": -0.000503022107295692,
          "frac_active": 0.1875
        },
        {
          "step": 358,
          "loss": 0.6909576654434204,
          "floss_loss": 6.006435432936996e-05,
          "kl_loss": 0.06908975541591644,
          "lyap1_mean": 0.0022753572557121515,
          "lyap1_max": 0.011385518126189709,
          "lyap_mean": -0.0008123294683173299,
          "volume_mean": -0.0008123294683173299,
          "volume_max": 0.00895810965448618,
          "frac_active": 0.4375
        },
        {
          "step": 359,
          "loss": 0.43423932790756226,
          "floss_loss": 0.00012433830124791712,
          "kl_loss": 0.0434115007519722,
          "lyap1_mean": -0.0007626744918525219,
          "lyap1_max": 0.011640244163572788,
          "lyap_mean": -0.004604068119078875,
          "volume_mean": -0.004604068119078875,
          "volume_max": 0.0036046472378075123,
          "frac_active": 0.3125
        },
        {
          "step": 360,
          "loss": 0.2867684066295624,
          "floss_loss": 0.000197243265574798,
          "kl_loss": 0.02865711785852909,
          "lyap1_mean": 0.018988091498613358,
          "lyap1_max": 0.0419318787753582,
          "lyap_mean": 0.006101256236433983,
          "volume_mean": 0.006101256236433983,
          "volume_max": 0.013188415206968784,
          "frac_active": 0.625
        },
        {
          "step": 361,
          "loss": 0.33772531151771545,
          "floss_loss": 0.0001566691935295239,
          "kl_loss": 0.033756863325834274,
          "lyap1_mean": -0.0032349657267332077,
          "lyap1_max": 0.016559859737753868,
          "lyap_mean": -0.007578173186630011,
          "volume_mean": -0.007578173186630011,
          "volume_max": 0.000847972056362778,
          "frac_active": 0.25
        },
        {
          "step": 362,
          "loss": 0.19039107859134674,
          "floss_loss": 0.00023336854064837098,
          "kl_loss": 0.01901577040553093,
          "lyap1_mean": 0.020916204899549484,
          "lyap1_max": 0.0323764868080616,
          "lyap_mean": 0.008463192731142044,
          "volume_mean": 0.008463192731142044,
          "volume_max": 0.02049754001200199,
          "frac_active": 0.8125
        },
        {
          "step": 363,
          "loss": 0.2139992117881775,
          "floss_loss": 5.875674105482176e-05,
          "kl_loss": 0.021394046023488045,
          "lyap1_mean": 0.007730378303676844,
          "lyap1_max": 0.016825556755065918,
          "lyap_mean": 0.0017756460001692176,
          "volume_mean": 0.0017756460001692176,
          "volume_max": 0.005570287816226482,
          "frac_active": 0.5
        },
        {
          "step": 364,
          "loss": 0.29768022894859314,
          "floss_loss": 9.754011261975393e-05,
          "kl_loss": 0.02975826896727085,
          "lyap1_mean": 0.010936358943581581,
          "lyap1_max": 0.01786898449063301,
          "lyap_mean": 0.0025645860005170107,
          "volume_mean": 0.0025645860005170107,
          "volume_max": 0.012908834964036942,
          "frac_active": 0.5625
        },
        {
          "step": 365,
          "loss": 0.1594754010438919,
          "floss_loss": 4.868436371907592e-05,
          "kl_loss": 0.01594267226755619,
          "lyap1_mean": -0.0002252780832350254,
          "lyap1_max": 0.005505550652742386,
          "lyap_mean": 0.0009335937211290002,
          "volume_mean": 0.0009335937211290002,
          "volume_max": 0.0067993225529789925,
          "frac_active": 0.5625
        },
        {
          "step": 366,
          "loss": 0.4439393877983093,
          "floss_loss": 8.132487710099667e-05,
          "kl_loss": 0.04438580572605133,
          "lyap1_mean": 0.008748684078454971,
          "lyap1_max": 0.01816917583346367,
          "lyap_mean": 0.0032324600033462048,
          "volume_mean": 0.0032324600033462048,
          "volume_max": 0.007749492302536964,
          "frac_active": 0.5625
        },
        {
          "step": 367,
          "loss": 0.3793938159942627,
          "floss_loss": 8.99311708053574e-05,
          "kl_loss": 0.03793038800358772,
          "lyap1_mean": 0.00595624465495348,
          "lyap1_max": 0.0134105971083045,
          "lyap_mean": 0.002952725160866976,
          "volume_mean": 0.002952725160866976,
          "volume_max": 0.008928176015615463,
          "frac_active": 0.6875
        },
        {
          "step": 368,
          "loss": 0.538318395614624,
          "floss_loss": 0.0001916122273541987,
          "kl_loss": 0.053812675178050995,
          "lyap1_mean": 0.005654924549162388,
          "lyap1_max": 0.040197618305683136,
          "lyap_mean": 0.001984215807169676,
          "volume_mean": 0.001984215807169676,
          "volume_max": 0.020124461501836777,
          "frac_active": 0.375
        },
        {
          "step": 369,
          "loss": 2.1922996044158936,
          "floss_loss": 4.560676097753458e-05,
          "kl_loss": 0.21922540664672852,
          "lyap1_mean": 0.004803680349141359,
          "lyap1_max": 0.009044475853443146,
          "lyap_mean": 0.002595644211396575,
          "volume_mean": 0.002595644211396575,
          "volume_max": 0.0054864417761564255,
          "frac_active": 0.625
        },
        {
          "step": 370,
          "loss": 0.33230555057525635,
          "floss_loss": 0.0001399600732838735,
          "kl_loss": 0.033216558396816254,
          "lyap1_mean": 0.0017655857373028994,
          "lyap1_max": 0.008039887063205242,
          "lyap_mean": 0.00328106083907187,
          "volume_mean": 0.00328106083907187,
          "volume_max": 0.018017709255218506,
          "frac_active": 0.5625
        },
        {
          "step": 371,
          "loss": 0.31214138865470886,
          "floss_loss": 5.652236723108217e-05,
          "kl_loss": 0.03120848536491394,
          "lyap1_mean": -0.002490397309884429,
          "lyap1_max": 0.0004121808160562068,
          "lyap_mean": -0.0035855851601809263,
          "volume_mean": -0.0035855851601809263,
          "volume_max": -0.00034558866173028946,
          "frac_active": 0.125
        },
        {
          "step": 372,
          "loss": 0.2744385600090027,
          "floss_loss": 0.00022812372480984777,
          "kl_loss": 0.027421042323112488,
          "lyap1_mean": 0.01598113216459751,
          "lyap1_max": 0.028009718284010887,
          "lyap_mean": 0.010661641135811806,
          "volume_mean": 0.010661641135811806,
          "volume_max": 0.024439651519060135,
          "frac_active": 0.8125
        },
        {
          "step": 373,
          "loss": 0.37898746132850647,
          "floss_loss": 7.800116145517677e-05,
          "kl_loss": 0.037890948355197906,
          "lyap1_mean": -0.0021219318732619286,
          "lyap1_max": 0.01669454574584961,
          "lyap_mean": -0.004140808712691069,
          "volume_mean": -0.004140808712691069,
          "volume_max": 0.005720615852624178,
          "frac_active": 0.25
        },
        {
          "step": 374,
          "loss": 0.3092711865901947,
          "floss_loss": 0.00013872540148440748,
          "kl_loss": 0.03091324493288994,
          "lyap1_mean": -0.005222531966865063,
          "lyap1_max": 0.006474748253822327,
          "lyap_mean": -0.008687533438205719,
          "volume_mean": -0.008687533438205719,
          "volume_max": -0.0015081254532560706,
          "frac_active": 0.125
        },
        {
          "step": 375,
          "loss": 0.2297932207584381,
          "floss_loss": 0.00024999678134918213,
          "kl_loss": 0.022954322397708893,
          "lyap1_mean": 0.025675300508737564,
          "lyap1_max": 0.0398593470454216,
          "lyap_mean": 0.00995168648660183,
          "volume_mean": 0.00995168648660183,
          "volume_max": 0.014399024657905102,
          "frac_active": 0.875
        },
        {
          "step": 376,
          "loss": 0.31132766604423523,
          "floss_loss": 0.00010328396456316113,
          "kl_loss": 0.031122436746954918,
          "lyap1_mean": 0.004090501926839352,
          "lyap1_max": 0.013903338462114334,
          "lyap_mean": -0.0026996529195457697,
          "volume_mean": -0.0026996529195457697,
          "volume_max": 0.005673442967236042,
          "frac_active": 0.3125
        },
        {
          "step": 377,
          "loss": 0.3813321888446808,
          "floss_loss": 0.0003529927344061434,
          "kl_loss": 0.03809792175889015,
          "lyap1_mean": -0.01260503102093935,
          "lyap1_max": 0.005208273883908987,
          "lyap_mean": -0.01434607058763504,
          "volume_mean": -0.01434607058763504,
          "volume_max": 0.0006241456721909344,
          "frac_active": 0.1875
        },
        {
          "step": 378,
          "loss": 0.2351704090833664,
          "floss_loss": 6.824003503425047e-05,
          "kl_loss": 0.023510215803980827,
          "lyap1_mean": -0.0037830471992492676,
          "lyap1_max": 0.004705434665083885,
          "lyap_mean": -0.005655704531818628,
          "volume_mean": -0.005655704531818628,
          "volume_max": -0.0018135677091777325,
          "frac_active": 0.1875
        },
        {
          "step": 379,
          "loss": 0.3443554639816284,
          "floss_loss": 6.300384848145768e-05,
          "kl_loss": 0.034429244697093964,
          "lyap1_mean": 0.0014704077038913965,
          "lyap1_max": 0.00709791574627161,
          "lyap_mean": -0.0025133672170341015,
          "volume_mean": -0.0025133672170341015,
          "volume_max": 0.0009483468602411449,
          "frac_active": 0.25
        },
        {
          "step": 380,
          "loss": 0.330779492855072,
          "floss_loss": 0.00017663766629993916,
          "kl_loss": 0.033060286194086075,
          "lyap1_mean": -0.004204597324132919,
          "lyap1_max": 0.0058806296437978745,
          "lyap_mean": -0.009233739227056503,
          "volume_mean": -0.009233739227056503,
          "volume_max": 0.0021064872853457928,
          "frac_active": 0.3125
        },
        {
          "step": 381,
          "loss": 0.1438315063714981,
          "floss_loss": 0.0004738590505439788,
          "kl_loss": 0.014335764572024345,
          "lyap1_mean": -0.00044152699410915375,
          "lyap1_max": 0.030018167570233345,
          "lyap_mean": -0.0077135879546403885,
          "volume_mean": -0.0077135879546403885,
          "volume_max": 0.022259285673499107,
          "frac_active": 0.3125
        },
        {
          "step": 382,
          "loss": 0.497580349445343,
          "floss_loss": 0.00018002153956331313,
          "kl_loss": 0.049740031361579895,
          "lyap1_mean": 0.008429519832134247,
          "lyap1_max": 0.01923186704516411,
          "lyap_mean": -0.0048692976124584675,
          "volume_mean": -0.0048692976124584675,
          "volume_max": 0.005059999879449606,
          "frac_active": 0.3125
        },
        {
          "step": 383,
          "loss": 0.3880373239517212,
          "floss_loss": 0.0002572311204858124,
          "kl_loss": 0.03877801075577736,
          "lyap1_mean": 0.001232808455824852,
          "lyap1_max": 0.037939444184303284,
          "lyap_mean": -0.009247399866580963,
          "volume_mean": -0.009247399866580963,
          "volume_max": 0.003833536524325609,
          "frac_active": 0.0625
        },
        {
          "step": 384,
          "loss": 0.5982335209846497,
          "floss_loss": 0.00014436677156481892,
          "kl_loss": 0.05980891361832619,
          "lyap1_mean": -0.004269084893167019,
          "lyap1_max": 0.005649724509567022,
          "lyap_mean": -0.009682226926088333,
          "volume_mean": -0.009682226926088333,
          "volume_max": -0.005220414604991674,
          "frac_active": 0.125
        },
        {
          "step": 385,
          "loss": 0.4399251937866211,
          "floss_loss": 0.00010006978118326515,
          "kl_loss": 0.04398251324892044,
          "lyap1_mean": -0.004222005605697632,
          "lyap1_max": 0.0033604048658162355,
          "lyap_mean": -0.009174499660730362,
          "volume_mean": -0.009174499660730362,
          "volume_max": -0.006297681946307421,
          "frac_active": 0.0625
        },
        {
          "step": 386,
          "loss": 0.38356950879096985,
          "floss_loss": 0.00023548101307824254,
          "kl_loss": 0.038333404809236526,
          "lyap1_mean": -0.00653980765491724,
          "lyap1_max": 0.0021158079616725445,
          "lyap_mean": -0.012925744988024235,
          "volume_mean": -0.012925744988024235,
          "volume_max": -0.0065666744485497475,
          "frac_active": 0.125
        },
        {
          "step": 387,
          "loss": 1.5929542779922485,
          "floss_loss": 0.00022593833273276687,
          "kl_loss": 0.15927283465862274,
          "lyap1_mean": -0.009618163108825684,
          "lyap1_max": 0.012012689374387264,
          "lyap_mean": -0.011579195968806744,
          "volume_mean": -0.011579195968806744,
          "volume_max": 0.00197397917509079,
          "frac_active": 0.125
        },
        {
          "step": 388,
          "loss": 0.3799367845058441,
          "floss_loss": 0.00014418009959626943,
          "kl_loss": 0.037979260087013245,
          "lyap1_mean": 0.0012468118220567703,
          "lyap1_max": 0.015228129923343658,
          "lyap_mean": -0.007601594086736441,
          "volume_mean": -0.007601594086736441,
          "volume_max": -0.0040441397577524185,
          "frac_active": 0.125
        },
        {
          "step": 389,
          "loss": 0.47508344054222107,
          "floss_loss": 0.0002460337127558887,
          "kl_loss": 0.047483738511800766,
          "lyap1_mean": -0.016602756455540657,
          "lyap1_max": -0.01028288621455431,
          "lyap_mean": -0.013567250221967697,
          "volume_mean": -0.013567250221967697,
          "volume_max": -0.006542901508510113,
          "frac_active": 0.0625
        },
        {
          "step": 390,
          "loss": 0.33457931876182556,
          "floss_loss": 0.00010827432561200112,
          "kl_loss": 0.033447105437517166,
          "lyap1_mean": -0.004071136005222797,
          "lyap1_max": 0.01035915594547987,
          "lyap_mean": -0.006083394400775433,
          "volume_mean": -0.006083394400775433,
          "volume_max": 0.002646117005497217,
          "frac_active": 0.25
        },
        {
          "step": 391,
          "loss": 0.24198570847511292,
          "floss_loss": 0.0001496233744546771,
          "kl_loss": 0.02418360859155655,
          "lyap1_mean": -0.007537027820944786,
          "lyap1_max": -0.0006516936118714511,
          "lyap_mean": -0.011255950666964054,
          "volume_mean": -0.011255950666964054,
          "volume_max": -0.009952126070857048,
          "frac_active": 0.0
        },
        {
          "step": 392,
          "loss": 0.47346678376197815,
          "floss_loss": 0.000168094047694467,
          "kl_loss": 0.047329869121313095,
          "lyap1_mean": -0.006193270906805992,
          "lyap1_max": 0.0019530991557985544,
          "lyap_mean": -0.01078033447265625,
          "volume_mean": -0.01078033447265625,
          "volume_max": -0.004069676157087088,
          "frac_active": 0.125
        },
        {
          "step": 393,
          "loss": 0.5589001774787903,
          "floss_loss": 8.588065975345671e-05,
          "kl_loss": 0.055881425738334656,
          "lyap1_mean": -0.004451178014278412,
          "lyap1_max": -0.0008808213751763105,
          "lyap_mean": -0.00689175259321928,
          "volume_mean": -0.00689175259321928,
          "volume_max": -0.0004207885358482599,
          "frac_active": 0.125
        },
        {
          "step": 394,
          "loss": 0.2183043658733368,
          "floss_loss": 0.0001284915633732453,
          "kl_loss": 0.021817587316036224,
          "lyap1_mean": -0.003962984308600426,
          "lyap1_max": 0.006412330083549023,
          "lyap_mean": -0.007717093452811241,
          "volume_mean": -0.007717093452811241,
          "volume_max": 0.002113310620188713,
          "frac_active": 0.3125
        },
        {
          "step": 395,
          "loss": 0.2790072560310364,
          "floss_loss": 0.00035928288707509637,
          "kl_loss": 0.027864797040820122,
          "lyap1_mean": -0.00764793436974287,
          "lyap1_max": 0.0016417670994997025,
          "lyap_mean": -0.01515517570078373,
          "volume_mean": -0.01515517570078373,
          "volume_max": -0.007998088374733925,
          "frac_active": 0.0625
        },
        {
          "step": 396,
          "loss": 0.3207534849643707,
          "floss_loss": 4.537688073469326e-05,
          "kl_loss": 0.03207080811262131,
          "lyap1_mean": 0.000805462128482759,
          "lyap1_max": 0.015269448980689049,
          "lyap_mean": 6.0189166106283665e-05,
          "volume_mean": 6.0189166106283665e-05,
          "volume_max": 0.006332932505756617,
          "frac_active": 0.5
        },
        {
          "step": 397,
          "loss": 0.17989343404769897,
          "floss_loss": 0.00026036924100480974,
          "kl_loss": 0.017963306978344917,
          "lyap1_mean": -0.0009305095300078392,
          "lyap1_max": 0.012118265964090824,
          "lyap_mean": -0.00512396264821291,
          "volume_mean": -0.00512396264821291,
          "volume_max": 0.006924641318619251,
          "frac_active": 0.375
        },
        {
          "step": 398,
          "loss": 0.42130017280578613,
          "floss_loss": 0.00021308059513103217,
          "kl_loss": 0.04210870712995529,
          "lyap1_mean": -0.0007186480797827244,
          "lyap1_max": 0.011596023105084896,
          "lyap_mean": -0.0024564387276768684,
          "volume_mean": -0.0024564387276768684,
          "volume_max": 0.009203268215060234,
          "frac_active": 0.625
        },
        {
          "step": 399,
          "loss": 0.27536773681640625,
          "floss_loss": 0.00017760111950337887,
          "kl_loss": 0.027519013732671738,
          "lyap1_mean": 0.013914374634623528,
          "lyap1_max": 0.023663930594921112,
          "lyap_mean": 0.011229117400944233,
          "volume_mean": 0.011229117400944233,
          "volume_max": 0.01720704510807991,
          "frac_active": 0.9375
        },
        {
          "step": 400,
          "loss": 0.5112131834030151,
          "floss_loss": 5.245393913355656e-05,
          "kl_loss": 0.05111607164144516,
          "lyap1_mean": 0.007701126392930746,
          "lyap1_max": 0.016077974811196327,
          "lyap_mean": 0.004170358180999756,
          "volume_mean": 0.004170358180999756,
          "volume_max": 0.009413398802280426,
          "frac_active": 0.625
        },
        {
          "step": 401,
          "loss": 0.5604389309883118,
          "floss_loss": 7.31532636564225e-05,
          "kl_loss": 0.056036580353975296,
          "lyap1_mean": 0.007933477871119976,
          "lyap1_max": 0.013132905587553978,
          "lyap_mean": 0.004668676294386387,
          "volume_mean": 0.004668676294386387,
          "volume_max": 0.013221452943980694,
          "frac_active": 0.75
        },
        {
          "step": 402,
          "loss": 0.32058948278427124,
          "floss_loss": 0.00014983079745434225,
          "kl_loss": 0.03204396739602089,
          "lyap1_mean": 0.01245945692062378,
          "lyap1_max": 0.01479965727776289,
          "lyap_mean": 0.010914690792560577,
          "volume_mean": 0.010914690792560577,
          "volume_max": 0.013893110677599907,
          "frac_active": 0.9375
        },
        {
          "step": 403,
          "loss": 0.3329673409461975,
          "floss_loss": 6.690216832794249e-05,
          "kl_loss": 0.03329004347324371,
          "lyap1_mean": 0.004423509351909161,
          "lyap1_max": 0.014772281050682068,
          "lyap_mean": 0.0006410907953977585,
          "volume_mean": 0.0006410907953977585,
          "volume_max": 0.006060359068214893,
          "frac_active": 0.4375
        },
        {
          "step": 404,
          "loss": 0.348193883895874,
          "floss_loss": 0.0002151060471078381,
          "kl_loss": 0.034797877073287964,
          "lyap1_mean": 0.012721373699605465,
          "lyap1_max": 0.01864858716726303,
          "lyap_mean": 0.009922504425048828,
          "volume_mean": 0.009922504425048828,
          "volume_max": 0.015371245332062244,
          "frac_active": 0.8125
        },
        {
          "step": 405,
          "loss": 0.4651016592979431,
          "floss_loss": 0.0001298909046454355,
          "kl_loss": 0.046497177332639694,
          "lyap1_mean": 0.014536803588271141,
          "lyap1_max": 0.030046792700886726,
          "lyap_mean": 0.007982086390256882,
          "volume_mean": 0.007982086390256882,
          "volume_max": 0.012700697407126427,
          "frac_active": 0.8125
        },
        {
          "step": 406,
          "loss": 0.2382693588733673,
          "floss_loss": 0.00030398648232221603,
          "kl_loss": 0.02379653789103031,
          "lyap1_mean": 0.024688860401511192,
          "lyap1_max": 0.04215405136346817,
          "lyap_mean": 0.010963805951178074,
          "volume_mean": 0.010963805951178074,
          "volume_max": 0.022488459944725037,
          "frac_active": 0.75
        },
        {
          "step": 407,
          "loss": 0.3589184880256653,
          "floss_loss": 5.925137156737037e-05,
          "kl_loss": 0.035885922610759735,
          "lyap1_mean": 0.007631555665284395,
          "lyap1_max": 0.01656511053442955,
          "lyap_mean": 0.004521097056567669,
          "volume_mean": 0.004521097056567669,
          "volume_max": 0.008665865287184715,
          "frac_active": 0.75
        },
        {
          "step": 408,
          "loss": 0.2830875813961029,
          "floss_loss": 0.0001851552806328982,
          "kl_loss": 0.028290241956710815,
          "lyap1_mean": 0.009504582732915878,
          "lyap1_max": 0.03393721207976341,
          "lyap_mean": -0.0012058473657816648,
          "volume_mean": -0.0012058473657816648,
          "volume_max": 0.015259869396686554,
          "frac_active": 0.3125
        },
        {
          "step": 409,
          "loss": 0.31690308451652527,
          "floss_loss": 6.602438952540979e-05,
          "kl_loss": 0.03168370574712753,
          "lyap1_mean": 0.0003472157404758036,
          "lyap1_max": 0.008187057450413704,
          "lyap_mean": -0.005141633562743664,
          "volume_mean": -0.005141633562743664,
          "volume_max": -0.0014613457024097443,
          "frac_active": 0.1875
        },
        {
          "step": 410,
          "loss": 0.2716144919395447,
          "floss_loss": 7.573122275061905e-05,
          "kl_loss": 0.027153877541422844,
          "lyap1_mean": 0.0023793932050466537,
          "lyap1_max": 0.015934664756059647,
          "lyap_mean": -0.0001568684820085764,
          "volume_mean": -0.0001568684820085764,
          "volume_max": 0.008932637050747871,
          "frac_active": 0.5625
        },
        {
          "step": 411,
          "loss": 0.3373989462852478,
          "floss_loss": 0.00010011714039137587,
          "kl_loss": 0.03372988477349281,
          "lyap1_mean": -0.0073386323638260365,
          "lyap1_max": 0.004404397681355476,
          "lyap_mean": -0.008606624789536,
          "volume_mean": -0.008606624789536,
          "volume_max": -0.0010432970011606812,
          "frac_active": 0.0625
        },
        {
          "step": 412,
          "loss": 0.2911587059497833,
          "floss_loss": 4.965269545209594e-05,
          "kl_loss": 0.029110906645655632,
          "lyap1_mean": -0.0012338000815361738,
          "lyap1_max": 0.010365216992795467,
          "lyap_mean": -0.0019939509220421314,
          "volume_mean": -0.0019939509220421314,
          "volume_max": 0.002334210556000471,
          "frac_active": 0.4375
        },
        {
          "step": 413,
          "loss": 0.1483972668647766,
          "floss_loss": 0.00013739746646024287,
          "kl_loss": 0.01482598576694727,
          "lyap1_mean": 0.0016299413982778788,
          "lyap1_max": 0.01815366931259632,
          "lyap_mean": -0.003983360715210438,
          "volume_mean": -0.003983360715210438,
          "volume_max": 0.005658856593072414,
          "frac_active": 0.25
        },
        {
          "step": 414,
          "loss": 0.3810977637767792,
          "floss_loss": 0.00013255987141747028,
          "kl_loss": 0.03809652104973793,
          "lyap1_mean": -0.0071502020582556725,
          "lyap1_max": -0.002065361011773348,
          "lyap_mean": -0.009605512022972107,
          "volume_mean": -0.009605512022972107,
          "volume_max": -0.006137833930552006,
          "frac_active": 0.0625
        },
        {
          "step": 415,
          "loss": 0.3066560626029968,
          "floss_loss": 0.0002008264564210549,
          "kl_loss": 0.030645523220300674,
          "lyap1_mean": -0.00492149917408824,
          "lyap1_max": 0.003704690607264638,
          "lyap_mean": -0.011479749344289303,
          "volume_mean": -0.011479749344289303,
          "volume_max": -0.004121133126318455,
          "frac_active": 0.125
        },
        {
          "step": 416,
          "loss": 0.4887431263923645,
          "floss_loss": 0.00015672594599891454,
          "kl_loss": 0.0488586388528347,
          "lyap1_mean": -0.005411665420979261,
          "lyap1_max": 0.0008477773517370224,
          "lyap_mean": -0.010001289658248425,
          "volume_mean": -0.010001289658248425,
          "volume_max": -0.003095255233347416,
          "frac_active": 0.0625
        },
        {
          "step": 417,
          "loss": 0.4137744605541229,
          "floss_loss": 0.00013088634295854717,
          "kl_loss": 0.04136435687541962,
          "lyap1_mean": 0.00037375022657215595,
          "lyap1_max": 0.008576090447604656,
          "lyap_mean": -0.008466990664601326,
          "volume_mean": -0.008466990664601326,
          "volume_max": -0.004920251201838255,
          "frac_active": 0.125
        },
        {
          "step": 418,
          "loss": 0.5971567630767822,
          "floss_loss": 0.00018665712559595704,
          "kl_loss": 0.05969700962305069,
          "lyap1_mean": -0.012443345040082932,
          "lyap1_max": -0.004838101100176573,
          "lyap_mean": -0.012128051370382309,
          "volume_mean": -0.012128051370382309,
          "volume_max": -0.004684014245867729,
          "frac_active": 0.0625
        },
        {
          "step": 419,
          "loss": 0.35590025782585144,
          "floss_loss": 0.00017444699187763035,
          "kl_loss": 0.03557258099317551,
          "lyap1_mean": -0.005870341323316097,
          "lyap1_max": 0.008853296749293804,
          "lyap_mean": -0.011504710651934147,
          "volume_mean": -0.011504710651934147,
          "volume_max": -0.009015943855047226,
          "frac_active": 0.0625
        },
        {
          "step": 420,
          "loss": 0.37752607464790344,
          "floss_loss": 4.694944072980434e-05,
          "kl_loss": 0.03774791210889816,
          "lyap1_mean": -0.0003242903621867299,
          "lyap1_max": 0.012611445039510727,
          "lyap_mean": -0.0034826728515326977,
          "volume_mean": -0.0034826728515326977,
          "volume_max": 0.002804914955049753,
          "frac_active": 0.3125
        },
        {
          "step": 421,
          "loss": 0.39923593401908875,
          "floss_loss": 0.0003166917013004422,
          "kl_loss": 0.03989192470908165,
          "lyap1_mean": -0.014527508988976479,
          "lyap1_max": -0.005275526084005833,
          "lyap_mean": -0.015765562653541565,
          "volume_mean": -0.015765562653541565,
          "volume_max": -0.005293608643114567,
          "frac_active": 0.0
        },
        {
          "step": 422,
          "loss": 0.3460356891155243,
          "floss_loss": 7.396807632176206e-05,
          "kl_loss": 0.034596171230077744,
          "lyap1_mean": 0.004293343517929316,
          "lyap1_max": 0.01819082535803318,
          "lyap_mean": -0.0033099986612796783,
          "volume_mean": -0.0033099986612796783,
          "volume_max": 0.004616002552211285,
          "frac_active": 0.1875
        },
        {
          "step": 423,
          "loss": 0.37318024039268494,
          "floss_loss": 0.00013859396858606488,
          "kl_loss": 0.037304166704416275,
          "lyap1_mean": 0.006011203862726688,
          "lyap1_max": 0.01854492351412773,
          "lyap_mean": -0.003671262413263321,
          "volume_mean": -0.003671262413263321,
          "volume_max": 0.004557696171104908,
          "frac_active": 0.3125
        },
        {
          "step": 424,
          "loss": 0.36190906167030334,
          "floss_loss": 0.00012225925456732512,
          "kl_loss": 0.03617868199944496,
          "lyap1_mean": 0.0036435911897569895,
          "lyap1_max": 0.022159937769174576,
          "lyap_mean": -0.0003130579134449363,
          "volume_mean": -0.0003130579134449363,
          "volume_max": 0.010982474312186241,
          "frac_active": 0.5625
        },
        {
          "step": 425,
          "loss": 0.3867282271385193,
          "floss_loss": 9.041297744261101e-05,
          "kl_loss": 0.03866378217935562,
          "lyap1_mean": 0.006146607920527458,
          "lyap1_max": 0.021324824541807175,
          "lyap_mean": 0.0020163236185908318,
          "volume_mean": 0.0020163236185908318,
          "volume_max": 0.013450251892209053,
          "frac_active": 0.5625
        },
        {
          "step": 426,
          "loss": 0.19052252173423767,
          "floss_loss": 2.7580754249356687e-05,
          "kl_loss": 0.019049493595957756,
          "lyap1_mean": 0.002329573966562748,
          "lyap1_max": 0.012999065220355988,
          "lyap_mean": -0.0011303728679195046,
          "volume_mean": -0.0011303728679195046,
          "volume_max": 0.0013218059903010726,
          "frac_active": 0.375
        },
        {
          "step": 427,
          "loss": 0.309180349111557,
          "floss_loss": 0.00017575130914337933,
          "kl_loss": 0.030900461599230766,
          "lyap1_mean": 0.01789304055273533,
          "lyap1_max": 0.03877738118171692,
          "lyap_mean": 0.009650465101003647,
          "volume_mean": 0.009650465101003647,
          "volume_max": 0.015842348337173462,
          "frac_active": 0.875
        },
        {
          "step": 428,
          "loss": 0.3889220952987671,
          "floss_loss": 0.00017002128879539669,
          "kl_loss": 0.03887520730495453,
          "lyap1_mean": 0.00020719144959002733,
          "lyap1_max": 0.020735029131174088,
          "lyap_mean": -0.001338032423518598,
          "volume_mean": -0.001338032423518598,
          "volume_max": 0.015319790691137314,
          "frac_active": 0.5
        },
        {
          "step": 429,
          "loss": 0.1267891675233841,
          "floss_loss": 0.0005335186724551022,
          "kl_loss": 0.012625564821064472,
          "lyap1_mean": 0.027525026351213455,
          "lyap1_max": 0.04867733269929886,
          "lyap_mean": 0.01499910093843937,
          "volume_mean": 0.01499910093843937,
          "volume_max": 0.03599635511636734,
          "frac_active": 0.75
        },
        {
          "step": 430,
          "loss": 0.5423554182052612,
          "floss_loss": 0.00015176167653407902,
          "kl_loss": 0.05422036349773407,
          "lyap1_mean": 0.011769728735089302,
          "lyap1_max": 0.015310775488615036,
          "lyap_mean": 0.0084733497351408,
          "volume_mean": 0.0084733497351408,
          "volume_max": 0.010291846469044685,
          "frac_active": 0.8125
        },
        {
          "step": 431,
          "loss": 0.36919575929641724,
          "floss_loss": 0.00029474686016328633,
          "kl_loss": 0.03689010068774223,
          "lyap1_mean": 0.017120249569416046,
          "lyap1_max": 0.035128720104694366,
          "lyap_mean": 0.013504715636372566,
          "volume_mean": 0.013504715636372566,
          "volume_max": 0.02777520939707756,
          "frac_active": 0.9375
        },
        {
          "step": 432,
          "loss": 0.41132667660713196,
          "floss_loss": 0.0003491957322694361,
          "kl_loss": 0.04109774902462959,
          "lyap1_mean": 0.030275816097855568,
          "lyap1_max": 0.03444733843207359,
          "lyap_mean": 0.014524002559483051,
          "volume_mean": 0.014524002559483051,
          "volume_max": 0.017911478877067566,
          "frac_active": 0.9375
        },
        {
          "step": 433,
          "loss": 0.5163977742195129,
          "floss_loss": 0.00012582972703967243,
          "kl_loss": 0.05162719637155533,
          "lyap1_mean": 0.017142143100500107,
          "lyap1_max": 0.022063111886382103,
          "lyap_mean": 0.007737856358289719,
          "volume_mean": 0.007737856358289719,
          "volume_max": 0.012970205396413803,
          "frac_active": 0.75
        },
        {
          "step": 434,
          "loss": 0.34211650490760803,
          "floss_loss": 0.00040439999429509044,
          "kl_loss": 0.034171212464571,
          "lyap1_mean": 0.025181423872709274,
          "lyap1_max": 0.045670442283153534,
          "lyap_mean": 0.013684690929949284,
          "volume_mean": 0.013684690929949284,
          "volume_max": 0.027590496465563774,
          "frac_active": 0.875
        },
        {
          "step": 435,
          "loss": 0.35557642579078674,
          "floss_loss": 0.00013611034955829382,
          "kl_loss": 0.035544030368328094,
          "lyap1_mean": 0.012732641771435738,
          "lyap1_max": 0.023611167445778847,
          "lyap_mean": 0.0065150088630616665,
          "volume_mean": 0.0065150088630616665,
          "volume_max": 0.011242854408919811,
          "frac_active": 0.5625
        },
        {
          "step": 436,
          "loss": 0.4357641935348511,
          "floss_loss": 0.00041486526606604457,
          "kl_loss": 0.04353493079543114,
          "lyap1_mean": 0.02342452108860016,
          "lyap1_max": 0.03821375221014023,
          "lyap_mean": 0.016884446144104004,
          "volume_mean": 0.016884446144104004,
          "volume_max": 0.027508316561579704,
          "frac_active": 0.9375
        },
        {
          "step": 437,
          "loss": 0.4561319351196289,
          "floss_loss": 0.0002127528132405132,
          "kl_loss": 0.045591916888952255,
          "lyap1_mean": 0.012503371573984623,
          "lyap1_max": 0.038015756756067276,
          "lyap_mean": 0.008162549696862698,
          "volume_mean": 0.008162549696862698,
          "volume_max": 0.0269707553088665,
          "frac_active": 0.75
        },
        {
          "step": 438,
          "loss": 0.2080029994249344,
          "floss_loss": 7.218770042527467e-05,
          "kl_loss": 0.02079308219254017,
          "lyap1_mean": 0.011362725868821144,
          "lyap1_max": 0.018834376707673073,
          "lyap_mean": 0.0038284026086330414,
          "volume_mean": 0.0038284026086330414,
          "volume_max": 0.0057737501338124275,
          "frac_active": 0.75
        },
        {
          "step": 439,
          "loss": 0.2354108691215515,
          "floss_loss": 0.0002301646745763719,
          "kl_loss": 0.023518070578575134,
          "lyap1_mean": 0.02154485136270523,
          "lyap1_max": 0.029302449896931648,
          "lyap_mean": 0.011256065219640732,
          "volume_mean": 0.011256065219640732,
          "volume_max": 0.016783449798822403,
          "frac_active": 0.875
        },
        {
          "step": 440,
          "loss": 0.40086182951927185,
          "floss_loss": 0.0001214718067785725,
          "kl_loss": 0.04007403552532196,
          "lyap1_mean": 0.01620728336274624,
          "lyap1_max": 0.025226419791579247,
          "lyap_mean": 0.008068948984146118,
          "volume_mean": 0.008068948984146118,
          "volume_max": 0.00982073787599802,
          "frac_active": 0.875
        },
        {
          "step": 441,
          "loss": 0.27511733770370483,
          "floss_loss": 0.00010423596540931612,
          "kl_loss": 0.027501309290528297,
          "lyap1_mean": 0.01718216761946678,
          "lyap1_max": 0.026052847504615784,
          "lyap_mean": 0.004245900548994541,
          "volume_mean": 0.004245900548994541,
          "volume_max": 0.005851049907505512,
          "frac_active": 0.6875
        },
        {
          "step": 442,
          "loss": 0.2634545862674713,
          "floss_loss": 9.530482930131257e-05,
          "kl_loss": 0.026335928589105606,
          "lyap1_mean": 0.009375983849167824,
          "lyap1_max": 0.027098070830106735,
          "lyap_mean": 0.00013372593093663454,
          "volume_mean": 0.00013372593093663454,
          "volume_max": 0.002842829329892993,
          "frac_active": 0.5625
        },
        {
          "step": 443,
          "loss": 0.3637002110481262,
          "floss_loss": 0.00013852343545295298,
          "kl_loss": 0.03635616973042488,
          "lyap1_mean": 0.009375190362334251,
          "lyap1_max": 0.021964669227600098,
          "lyap_mean": 0.0035075927153229713,
          "volume_mean": 0.0035075927153229713,
          "volume_max": 0.018586086109280586,
          "frac_active": 0.5625
        },
        {
          "step": 444,
          "loss": 0.29260003566741943,
          "floss_loss": 0.00023800350027158856,
          "kl_loss": 0.02923620492219925,
          "lyap1_mean": -0.00415729358792305,
          "lyap1_max": 0.011572018265724182,
          "lyap_mean": -0.010932140983641148,
          "volume_mean": -0.010932140983641148,
          "volume_max": -0.0012414241209626198,
          "frac_active": 0.1875
        },
        {
          "step": 445,
          "loss": 0.12528358399868011,
          "floss_loss": 8.404890104429796e-05,
          "kl_loss": 0.012519954703748226,
          "lyap1_mean": -0.008087825030088425,
          "lyap1_max": -0.0022320838179439306,
          "lyap_mean": -0.007824026979506016,
          "volume_mean": -0.007824026979506016,
          "volume_max": -0.004783375188708305,
          "frac_active": 0.0
        },
        {
          "step": 446,
          "loss": 0.2888416349887848,
          "floss_loss": 0.0001375318825012073,
          "kl_loss": 0.028870411217212677,
          "lyap1_mean": -0.005704062059521675,
          "lyap1_max": 0.002115456387400627,
          "lyap_mean": -0.01026076264679432,
          "volume_mean": -0.01026076264679432,
          "volume_max": -0.006408251356333494,
          "frac_active": 0.0625
        },
        {
          "step": 447,
          "loss": 0.22059647738933563,
          "floss_loss": 0.00023151264758780599,
          "kl_loss": 0.022036496549844742,
          "lyap1_mean": 0.005435247905552387,
          "lyap1_max": 0.03623161092400551,
          "lyap_mean": -0.007767352741211653,
          "volume_mean": -0.007767352741211653,
          "volume_max": 0.002445532474666834,
          "frac_active": 0.125
        },
        {
          "step": 448,
          "loss": 0.38500887155532837,
          "floss_loss": 0.0001074043830158189,
          "kl_loss": 0.03849014639854431,
          "lyap1_mean": 0.005183156579732895,
          "lyap1_max": 0.019325755536556244,
          "lyap_mean": -0.005013877525925636,
          "volume_mean": -0.005013877525925636,
          "volume_max": 0.0016470735426992178,
          "frac_active": 0.25
        },
        {
          "step": 449,
          "loss": 0.38527822494506836,
          "floss_loss": 0.00012145953951403499,
          "kl_loss": 0.03851567581295967,
          "lyap1_mean": -0.005472735036164522,
          "lyap1_max": 0.010534333065152168,
          "lyap_mean": -0.00852225162088871,
          "volume_mean": -0.00852225162088871,
          "volume_max": -0.00017241435125470161,
          "frac_active": 0.125
        },
        {
          "step": 450,
          "loss": 0.3369617164134979,
          "floss_loss": 0.00010135400953004137,
          "kl_loss": 0.03368603438138962,
          "lyap1_mean": -0.003307035658508539,
          "lyap1_max": -0.0011066744336858392,
          "lyap_mean": -0.00581949669867754,
          "volume_mean": -0.00581949669867754,
          "volume_max": 0.0035747280344367027,
          "frac_active": 0.1875
        },
        {
          "step": 451,
          "loss": 0.26035985350608826,
          "floss_loss": 6.924521585460752e-05,
          "kl_loss": 0.026029063388705254,
          "lyap1_mean": 0.004672178532928228,
          "lyap1_max": 0.021637575700879097,
          "lyap_mean": -0.0027607777155935764,
          "volume_mean": -0.0027607777155935764,
          "volume_max": 0.0034460495226085186,
          "frac_active": 0.1875
        },
        {
          "step": 452,
          "loss": 0.24168550968170166,
          "floss_loss": 4.3904761696467176e-05,
          "kl_loss": 0.02416416071355343,
          "lyap1_mean": -0.0021043873857706785,
          "lyap1_max": 0.0018478352576494217,
          "lyap_mean": -0.004751889500766993,
          "volume_mean": -0.004751889500766993,
          "volume_max": 7.596985960844904e-05,
          "frac_active": 0.3125
        },
        {
          "step": 453,
          "loss": 0.3045896887779236,
          "floss_loss": 4.396117583382875e-05,
          "kl_loss": 0.030454574152827263,
          "lyap1_mean": -0.0009114141575992107,
          "lyap1_max": 0.010768073610961437,
          "lyap_mean": -0.0016894483705982566,
          "volume_mean": -0.0016894483705982566,
          "volume_max": 0.0023831124417483807,
          "frac_active": 0.375
        },
        {
          "step": 454,
          "loss": 0.164525106549263,
          "floss_loss": 0.0001975760969799012,
          "kl_loss": 0.016432752832770348,
          "lyap1_mean": -0.010616166517138481,
          "lyap1_max": -0.0047963340766727924,
          "lyap_mean": -0.011996272020041943,
          "volume_mean": -0.011996272020041943,
          "volume_max": -0.005603622645139694,
          "frac_active": 0.0625
        },
        {
          "step": 455,
          "loss": 0.20802779495716095,
          "floss_loss": 0.00014109333278611302,
          "kl_loss": 0.02078866958618164,
          "lyap1_mean": 0.01252937875688076,
          "lyap1_max": 0.027749884873628616,
          "lyap_mean": 0.0048828125,
          "volume_mean": 0.0048828125,
          "volume_max": 0.013674283400177956,
          "frac_active": 0.5625
        },
        {
          "step": 456,
          "loss": 0.2733783423900604,
          "floss_loss": 8.297676686197519e-05,
          "kl_loss": 0.02732953615486622,
          "lyap1_mean": 0.004938188940286636,
          "lyap1_max": 0.01610439456999302,
          "lyap_mean": 0.00030673673609271646,
          "volume_mean": 0.00030673673609271646,
          "volume_max": 0.006530665792524815,
          "frac_active": 0.3125
        },
        {
          "step": 457,
          "loss": 0.2840368449687958,
          "floss_loss": 0.00030572028481401503,
          "kl_loss": 0.02837311290204525,
          "lyap1_mean": -0.004121807403862476,
          "lyap1_max": 0.018619485199451447,
          "lyap_mean": -0.009564748965203762,
          "volume_mean": -0.009564748965203762,
          "volume_max": -0.0008356529287993908,
          "frac_active": 0.125
        },
        {
          "step": 458,
          "loss": 0.178431898355484,
          "floss_loss": 0.0002455147332511842,
          "kl_loss": 0.017818639054894447,
          "lyap1_mean": 0.011670931242406368,
          "lyap1_max": 0.02852139063179493,
          "lyap_mean": 0.010169395245611668,
          "volume_mean": 0.010169395245611668,
          "volume_max": 0.020654182881116867,
          "frac_active": 0.6875
        },
        {
          "step": 459,
          "loss": 0.3053043782711029,
          "floss_loss": 9.923095058184117e-05,
          "kl_loss": 0.030520513653755188,
          "lyap1_mean": 0.010753622278571129,
          "lyap1_max": 0.01305202767252922,
          "lyap_mean": 0.0016649486497044563,
          "volume_mean": 0.0016649486497044563,
          "volume_max": 0.0066561647690832615,
          "frac_active": 0.625
        },
        {
          "step": 460,
          "loss": 0.3194831907749176,
          "floss_loss": 0.0001871965650934726,
          "kl_loss": 0.03192960098385811,
          "lyap1_mean": 0.011047838255763054,
          "lyap1_max": 0.033221013844013214,
          "lyap_mean": 0.003934135660529137,
          "volume_mean": 0.003934135660529137,
          "volume_max": 0.018618635833263397,
          "frac_active": 0.625
        },
        {
          "step": 461,
          "loss": 0.18538692593574524,
          "floss_loss": 0.00010170903988182545,
          "kl_loss": 0.018528521060943604,
          "lyap1_mean": 0.009413800202310085,
          "lyap1_max": 0.020481562241911888,
          "lyap_mean": 0.0047165388241410255,
          "volume_mean": 0.0047165388241410255,
          "volume_max": 0.01197223924100399,
          "frac_active": 0.75
        },
        {
          "step": 462,
          "loss": 0.5809723734855652,
          "floss_loss": 9.731852333061397e-05,
          "kl_loss": 0.05808750167489052,
          "lyap1_mean": -0.004915723577141762,
          "lyap1_max": 0.007486353628337383,
          "lyap_mean": -0.006471653934568167,
          "volume_mean": -0.006471653934568167,
          "volume_max": 0.0010768786305561662,
          "frac_active": 0.25
        },
        {
          "step": 463,
          "loss": 0.31065645813941956,
          "floss_loss": 7.823539635865018e-05,
          "kl_loss": 0.031057823449373245,
          "lyap1_mean": -0.0006913093966431916,
          "lyap1_max": 0.010461130179464817,
          "lyap_mean": -0.002123354934155941,
          "volume_mean": -0.002123354934155941,
          "volume_max": 0.0062407683581113815,
          "frac_active": 0.375
        },
        {
          "step": 464,
          "loss": 0.42570966482162476,
          "floss_loss": 9.95306036202237e-05,
          "kl_loss": 0.042561013251543045,
          "lyap1_mean": 0.005723780021071434,
          "lyap1_max": 0.019127747043967247,
          "lyap_mean": 0.002521372167393565,
          "volume_mean": 0.002521372167393565,
          "volume_max": 0.011836141347885132,
          "frac_active": 0.5625
        },
        {
          "step": 465,
          "loss": 0.4790860414505005,
          "floss_loss": 0.0003572853747755289,
          "kl_loss": 0.0478728748857975,
          "lyap1_mean": -0.004681567195802927,
          "lyap1_max": 0.012850874103605747,
          "lyap_mean": -0.010373562574386597,
          "volume_mean": -0.010373562574386597,
          "volume_max": 0.003052609506994486,
          "frac_active": 0.25
        },
        {
          "step": 466,
          "loss": 0.2982342839241028,
          "floss_loss": 0.0002686342631932348,
          "kl_loss": 0.02979656495153904,
          "lyap1_mean": -0.011163365095853806,
          "lyap1_max": -0.0017584835877642035,
          "lyap_mean": -0.013167962431907654,
          "volume_mean": -0.013167962431907654,
          "volume_max": -0.003855006769299507,
          "frac_active": 0.0625
        },
        {
          "step": 467,
          "loss": 0.3016364574432373,
          "floss_loss": 0.000695171533152461,
          "kl_loss": 0.030094129964709282,
          "lyap1_mean": -0.024825982749462128,
          "lyap1_max": -0.012902568094432354,
          "lyap_mean": -0.02389867976307869,
          "volume_mean": -0.02389867976307869,
          "volume_max": -0.007692337967455387,
          "frac_active": 0.0
        },
        {
          "step": 468,
          "loss": 0.26984265446662903,
          "floss_loss": 0.00022905315563548356,
          "kl_loss": 0.02696135826408863,
          "lyap1_mean": -0.006644735112786293,
          "lyap1_max": 0.006524188909679651,
          "lyap_mean": -0.012151183560490608,
          "volume_mean": -0.012151183560490608,
          "volume_max": -5.989568307995796e-05,
          "frac_active": 0.125
        },
        {
          "step": 469,
          "loss": 0.28775495290756226,
          "floss_loss": 4.658516627387144e-05,
          "kl_loss": 0.02877083607017994,
          "lyap1_mean": -0.006283200345933437,
          "lyap1_max": -0.0017531777266412973,
          "lyap_mean": -0.0044457265175879,
          "volume_mean": -0.0044457265175879,
          "volume_max": -0.0014859598595649004,
          "frac_active": 0.1875
        },
        {
          "step": 470,
          "loss": 0.25806066393852234,
          "floss_loss": 6.135118746897206e-05,
          "kl_loss": 0.025799930095672607,
          "lyap1_mean": -0.00417732447385788,
          "lyap1_max": -0.0005734027945436537,
          "lyap_mean": -0.006221619434654713,
          "volume_mean": -0.006221619434654713,
          "volume_max": -0.003345395438373089,
          "frac_active": 0.0
        },
        {
          "step": 471,
          "loss": 0.29679518938064575,
          "floss_loss": 3.420759821892716e-05,
          "kl_loss": 0.029676096513867378,
          "lyap1_mean": -0.0015815815422683954,
          "lyap1_max": 0.005771625321358442,
          "lyap_mean": -0.0021727129351347685,
          "volume_mean": -0.0021727129351347685,
          "volume_max": 0.0006768802413716912,
          "frac_active": 0.3125
        },
        {
          "step": 472,
          "loss": 0.3241017460823059,
          "floss_loss": 0.0001655559171922505,
          "kl_loss": 0.032393619418144226,
          "lyap1_mean": 0.012934088706970215,
          "lyap1_max": 0.019986677914857864,
          "lyap_mean": 0.00985284149646759,
          "volume_mean": 0.00985284149646759,
          "volume_max": 0.01874496415257454,
          "frac_active": 0.9375
        },
        {
          "step": 473,
          "loss": 0.3293744623661041,
          "floss_loss": 0.00018087541684508324,
          "kl_loss": 0.03291935846209526,
          "lyap1_mean": 0.0033543845638632774,
          "lyap1_max": 0.02304934523999691,
          "lyap_mean": -0.0018813629867509007,
          "volume_mean": -0.0018813629867509007,
          "volume_max": 0.014854284934699535,
          "frac_active": 0.25
        },
        {
          "step": 474,
          "loss": 0.3470434546470642,
          "floss_loss": 5.1577277190517634e-05,
          "kl_loss": 0.03469918668270111,
          "lyap1_mean": 0.005482858046889305,
          "lyap1_max": 0.014728852547705173,
          "lyap_mean": 0.003077078377828002,
          "volume_mean": 0.003077078377828002,
          "volume_max": 0.011073904111981392,
          "frac_active": 0.6875
        },
        {
          "step": 475,
          "loss": 0.25435253977775574,
          "floss_loss": 0.00019095276365987957,
          "kl_loss": 0.025416160002350807,
          "lyap1_mean": 0.012680952437222004,
          "lyap1_max": 0.03262748941779137,
          "lyap_mean": 0.006940754596143961,
          "volume_mean": 0.006940754596143961,
          "volume_max": 0.021419718861579895,
          "frac_active": 0.625
        },
        {
          "step": 476,
          "loss": 0.2488202303647995,
          "floss_loss": 6.0266960645094514e-05,
          "kl_loss": 0.024875996634364128,
          "lyap1_mean": -0.0017387993866577744,
          "lyap1_max": 0.004218112211674452,
          "lyap_mean": -0.004700218327343464,
          "volume_mean": -0.004700218327343464,
          "volume_max": -0.0018212294671684504,
          "frac_active": 0.25
        },
        {
          "step": 477,
          "loss": 0.15868031978607178,
          "floss_loss": 0.0001970124721992761,
          "kl_loss": 0.015848331153392792,
          "lyap1_mean": 0.00920847151428461,
          "lyap1_max": 0.025218980386853218,
          "lyap_mean": 0.0029046284034848213,
          "volume_mean": 0.0029046284034848213,
          "volume_max": 0.020584309473633766,
          "frac_active": 0.5625
        },
        {
          "step": 478,
          "loss": 0.3567173480987549,
          "floss_loss": 0.00014067659503780305,
          "kl_loss": 0.03565766662359238,
          "lyap1_mean": -0.005363353993743658,
          "lyap1_max": 0.000692434492520988,
          "lyap_mean": -0.01020556315779686,
          "volume_mean": -0.01020556315779686,
          "volume_max": -0.00290459836833179,
          "frac_active": 0.0625
        },
        {
          "step": 479,
          "loss": 0.3041524291038513,
          "floss_loss": 0.00012328187585808337,
          "kl_loss": 0.030402913689613342,
          "lyap1_mean": 0.014535216614603996,
          "lyap1_max": 0.026651376858353615,
          "lyap_mean": 0.0007485895184800029,
          "volume_mean": 0.0007485895184800029,
          "volume_max": 0.003990259952843189,
          "frac_active": 0.3125
        },
        {
          "step": 480,
          "loss": 0.3199961483478546,
          "floss_loss": 0.00013329254579730332,
          "kl_loss": 0.031986285001039505,
          "lyap1_mean": -0.004222536459565163,
          "lyap1_max": 0.009709123522043228,
          "lyap_mean": -0.004815316293388605,
          "volume_mean": -0.004815316293388605,
          "volume_max": 0.007470431737601757,
          "frac_active": 0.3125
        },
        {
          "step": 481,
          "loss": 0.3832969069480896,
          "floss_loss": 6.686678534606472e-05,
          "kl_loss": 0.03832300379872322,
          "lyap1_mean": -0.0013866283698007464,
          "lyap1_max": 0.006003531627357006,
          "lyap_mean": -0.006179005838930607,
          "volume_mean": -0.006179005838930607,
          "volume_max": -0.001864773454144597,
          "frac_active": 0.0625
        },
        {
          "step": 482,
          "loss": 0.25592151284217834,
          "floss_loss": 8.781741780694574e-05,
          "kl_loss": 0.02558336779475212,
          "lyap1_mean": -0.0013448263052850962,
          "lyap1_max": 0.0024981603492051363,
          "lyap_mean": -0.005673505365848541,
          "volume_mean": -0.005673505365848541,
          "volume_max": 0.00042955169919878244,
          "frac_active": 0.3125
        },
        {
          "step": 483,
          "loss": 0.2858031094074249,
          "floss_loss": 0.0001757103018462658,
          "kl_loss": 0.028562739491462708,
          "lyap1_mean": -0.0031811147928237915,
          "lyap1_max": 0.00973997637629509,
          "lyap_mean": -0.006761206313967705,
          "volume_mean": -0.006761206313967705,
          "volume_max": 0.004273229278624058,
          "frac_active": 0.25
        },
        {
          "step": 484,
          "loss": 0.24309498071670532,
          "floss_loss": 0.00026398885529488325,
          "kl_loss": 0.02428309991955757,
          "lyap1_mean": -0.013779336586594582,
          "lyap1_max": -0.005162074696272612,
          "lyap_mean": -0.014612155966460705,
          "volume_mean": -0.014612155966460705,
          "volume_max": -0.006655748002231121,
          "frac_active": 0.0
        },
        {
          "step": 485,
          "loss": 0.5591369867324829,
          "floss_loss": 0.00020529743051156402,
          "kl_loss": 0.05589316785335541,
          "lyap1_mean": 0.001910895574837923,
          "lyap1_max": 0.015756888315081596,
          "lyap_mean": -0.00859428197145462,
          "volume_mean": -0.00859428197145462,
          "volume_max": 0.00041857105679810047,
          "frac_active": 0.3125
        },
        {
          "step": 486,
          "loss": 0.2696172297000885,
          "floss_loss": 0.0003427234769333154,
          "kl_loss": 0.02692745067179203,
          "lyap1_mean": -0.015422258526086807,
          "lyap1_max": -0.006487723905593157,
          "lyap_mean": -0.017334483563899994,
          "volume_mean": -0.017334483563899994,
          "volume_max": -0.012895765714347363,
          "frac_active": 0.0
        },
        {
          "step": 487,
          "loss": 0.2240576148033142,
          "floss_loss": 0.0001263611193280667,
          "kl_loss": 0.022393126040697098,
          "lyap1_mean": -0.0021457471884787083,
          "lyap1_max": 0.004710844252258539,
          "lyap_mean": -0.006529409904032946,
          "volume_mean": -0.006529409904032946,
          "volume_max": 0.004871497396379709,
          "frac_active": 0.1875
        },
        {
          "step": 488,
          "loss": 0.38659897446632385,
          "floss_loss": 0.0002545433526393026,
          "kl_loss": 0.038634441792964935,
          "lyap1_mean": -0.013206858187913895,
          "lyap1_max": 0.002197075868025422,
          "lyap_mean": -0.014303634874522686,
          "volume_mean": -0.014303634874522686,
          "volume_max": -0.010377535596489906,
          "frac_active": 0.0625
        },
        {
          "step": 489,
          "loss": 0.3662464916706085,
          "floss_loss": 0.00019338249694555998,
          "kl_loss": 0.03660530969500542,
          "lyap1_mean": -0.006694752722978592,
          "lyap1_max": 0.006432893220335245,
          "lyap_mean": -0.011304594576358795,
          "volume_mean": -0.011304594576358795,
          "volume_max": -0.0069494703784585,
          "frac_active": 0.0625
        },
        {
          "step": 490,
          "loss": 0.1910959929227829,
          "floss_loss": 0.00013203301932662725,
          "kl_loss": 0.01909639500081539,
          "lyap1_mean": -0.005610155873000622,
          "lyap1_max": 0.012676538899540901,
          "lyap_mean": -0.00781496986746788,
          "volume_mean": -0.00781496986746788,
          "volume_max": -0.0008753207512199879,
          "frac_active": 0.1875
        },
        {
          "step": 491,
          "loss": 1.2009166479110718,
          "floss_loss": 0.0002738822950050235,
          "kl_loss": 0.12006428837776184,
          "lyap1_mean": -0.008356180042028427,
          "lyap1_max": 0.007035958115011454,
          "lyap_mean": -0.014933168888092041,
          "volume_mean": -0.014933168888092041,
          "volume_max": -0.009899364784359932,
          "frac_active": 0.0625
        },
        {
          "step": 492,
          "loss": 0.2875405251979828,
          "floss_loss": 9.2828951892443e-05,
          "kl_loss": 0.028744768351316452,
          "lyap1_mean": -0.002387974876910448,
          "lyap1_max": 0.005905994679778814,
          "lyap_mean": -0.005980554968118668,
          "volume_mean": -0.005980554968118668,
          "volume_max": 0.0011631835950538516,
          "frac_active": 0.25
        },
        {
          "step": 493,
          "loss": 0.25397178530693054,
          "floss_loss": 0.00010544469114392996,
          "kl_loss": 0.025386633351445198,
          "lyap1_mean": -0.0005209211958572268,
          "lyap1_max": 0.014142598956823349,
          "lyap_mean": -0.006661063525825739,
          "volume_mean": -0.006661063525825739,
          "volume_max": 0.0007371143437922001,
          "frac_active": 0.125
        },
        {
          "step": 494,
          "loss": 0.35137227177619934,
          "floss_loss": 0.00031127248075790703,
          "kl_loss": 0.035106100142002106,
          "lyap1_mean": -0.0029403662774711847,
          "lyap1_max": 0.015249314717948437,
          "lyap_mean": -0.009317319840192795,
          "volume_mean": -0.009317319840192795,
          "volume_max": 0.008420249447226524,
          "frac_active": 0.3125
        },
        {
          "step": 495,
          "loss": 0.24840061366558075,
          "floss_loss": 3.400592686375603e-05,
          "kl_loss": 0.02483666129410267,
          "lyap1_mean": 0.00027881970163434744,
          "lyap1_max": 0.005699230823665857,
          "lyap_mean": -0.0013190482277423143,
          "volume_mean": -0.0013190482277423143,
          "volume_max": 0.0030218716710805893,
          "frac_active": 0.3125
        },
        {
          "step": 496,
          "loss": 0.3720506727695465,
          "floss_loss": 4.55567569588311e-05,
          "kl_loss": 0.03720051050186157,
          "lyap1_mean": 0.005975417792797089,
          "lyap1_max": 0.013024869374930859,
          "lyap_mean": 0.0008145913598127663,
          "volume_mean": 0.0008145913598127663,
          "volume_max": 0.006740645505487919,
          "frac_active": 0.5625
        },
        {
          "step": 497,
          "loss": 0.4111747443675995,
          "floss_loss": 0.0002978438569698483,
          "kl_loss": 0.041087690740823746,
          "lyap1_mean": -0.002442651428282261,
          "lyap1_max": 0.02007380686700344,
          "lyap_mean": -0.004608362913131714,
          "volume_mean": -0.004608362913131714,
          "volume_max": 0.012163406237959862,
          "frac_active": 0.4375
        },
        {
          "step": 498,
          "loss": 0.3084067106246948,
          "floss_loss": 5.5445550970034674e-05,
          "kl_loss": 0.03083512745797634,
          "lyap1_mean": 0.0008167533669620752,
          "lyap1_max": 0.010597714222967625,
          "lyap_mean": -0.0005695611471310258,
          "volume_mean": -0.0005695611471310258,
          "volume_max": 0.005058505106717348,
          "frac_active": 0.375
        },
        {
          "step": 499,
          "loss": 0.18024350702762604,
          "floss_loss": 5.8540652389638126e-05,
          "kl_loss": 0.01801849529147148,
          "lyap1_mean": 0.00878806784749031,
          "lyap1_max": 0.011588695459067822,
          "lyap_mean": 0.0018037448171526194,
          "volume_mean": 0.0018037448171526194,
          "volume_max": 0.008622938767075539,
          "frac_active": 0.5
        }
      ]
    }
  ],
  "evals": [
    {
      "kind": "initial",
      "train_step": 0,
      "acc": 0.59765625,
      "tok_acc": 0.8614245756172839
    },
    {
      "kind": "after_floss",
      "train_step": 0,
      "episode": 0,
      "acc": 0.025390625,
      "tok_acc": 0.6345727237654321
    },
    {
      "kind": "after_floss",
      "train_step": 500,
      "episode": 1,
      "acc": 0.013671875,
      "tok_acc": 0.6422405478395061
    },
    {
      "kind": "task",
      "train_step": 1000,
      "acc": 0.041015625,
      "tok_acc": 0.6760223765432098
    },
    {
      "kind": "task",
      "train_step": 2000,
      "acc": 0.322265625,
      "tok_acc": 0.7650945216049383
    },
    {
      "kind": "task",
      "train_step": 3000,
      "acc": 0.439453125,
      "tok_acc": 0.8069540895061729
    },
    {
      "kind": "task",
      "train_step": 4000,
      "acc": 0.515625,
      "tok_acc": 0.8287760416666666
    },
    {
      "kind": "task",
      "train_step": 5000,
      "acc": 0.498046875,
      "tok_acc": 0.8277633101851852
    },
    {
      "kind": "task",
      "train_step": 6000,
      "acc": 0.57421875,
      "tok_acc": 0.8464506172839507
    },
    {
      "kind": "task",
      "train_step": 7000,
      "acc": 0.556640625,
      "tok_acc": 0.8420621141975309
    },
    {
      "kind": "task",
      "train_step": 8000,
      "acc": 0.55859375,
      "tok_acc": 0.8435329861111112
    },
    {
      "kind": "task",
      "train_step": 9000,
      "acc": 0.5546875,
      "tok_acc": 0.841820987654321
    },
    {
      "kind": "task",
      "train_step": 10000,
      "acc": 0.595703125,
      "tok_acc": 0.8568672839506173
    },
    {
      "kind": "final",
      "train_step": 10000,
      "acc": 0.595703125,
      "tok_acc": 0.8568672839506173
    }
  ],
  "final_acc": 0.595703125,
  "final_tok_acc": 0.8568672839506173
}