| {"step": 41000, "weights_source": "ema", "loss": 0.3725855791568756, "edge_loss": 0.19200483143329622, "diag_loss": 0.18058074414730071, "jvp_rms": 2.078083175420761, "examples": 5120, "loss_nfe_1": 0.3927511692047119, "edge_loss_nfe_1": 0.2175933986902237, "diag_loss_nfe_1": 0.17515776455402374, "jvp_rms_nfe_1": 2.4403680324554444, "loss_nfe_2": 0.369793975353241, "edge_loss_nfe_2": 0.18860338628292084, "diag_loss_nfe_2": 0.1811905950307846, "jvp_rms_nfe_2": 2.2941908717155455, "loss_nfe_3": 0.3564019322395325, "edge_loss_nfe_3": 0.17990776002407075, "diag_loss_nfe_3": 0.17649417072534562, "jvp_rms_nfe_3": 1.885271716117859, "loss_nfe_4": 0.3897887945175171, "edge_loss_nfe_4": 0.19601401388645173, "diag_loss_nfe_4": 0.1937747687101364, "jvp_rms_nfe_4": 2.270489048957825, "loss_nfe_5": 0.3541920244693756, "edge_loss_nfe_5": 0.17790559828281402, "diag_loss_nfe_5": 0.17628642171621323, "jvp_rms_nfe_5": 1.500096207857132, "raw_loss": 0.379677836894989, "raw_edge_loss": 0.19833322674036027, "raw_diag_loss": 0.1813446119427681, "raw_jvp_rms": 2.2722322368621826, "raw_loss_nfe_1": 0.4198586106300354, "raw_edge_loss_nfe_1": 0.24385497570037842, "raw_diag_loss_nfe_1": 0.17600363194942475, "raw_jvp_rms_nfe_1": 2.645507574081421, "raw_loss_nfe_2": 0.37339203953742983, "raw_edge_loss_nfe_2": 0.19148991107940674, "raw_diag_loss_nfe_2": 0.18190212845802306, "raw_jvp_rms_nfe_2": 2.5214332699775697, "raw_loss_nfe_3": 0.35822181701660155, "raw_edge_loss_nfe_3": 0.18091285228729248, "raw_diag_loss_nfe_3": 0.1773089662194252, "raw_jvp_rms_nfe_3": 2.062240254878998, "raw_loss_nfe_4": 0.3911686480045319, "raw_edge_loss_nfe_4": 0.19675154983997345, "raw_diag_loss_nfe_4": 0.1944171041250229, "raw_jvp_rms_nfe_4": 2.484368699789047, "raw_loss_nfe_5": 0.35574806928634645, "raw_edge_loss_nfe_5": 0.17865684479475022, "raw_diag_loss_nfe_5": 0.17709122896194457, "raw_jvp_rms_nfe_5": 1.6476113855838777} |
| {"step": 42000, "weights_source": "ema", "loss": 0.3723788583278656, "edge_loss": 0.19195123821496965, "diag_loss": 0.18042762011289598, "jvp_rms": 2.0993320310115813, "examples": 5120, "loss_nfe_1": 0.3929444789886475, "edge_loss_nfe_1": 0.21792266964912416, "diag_loss_nfe_1": 0.17502180635929107, "jvp_rms_nfe_1": 2.471006441116333, "loss_nfe_2": 0.36948872208595274, "edge_loss_nfe_2": 0.18845009803771973, "diag_loss_nfe_2": 0.18103862702846527, "jvp_rms_nfe_2": 2.3163293421268465, "loss_nfe_3": 0.35611130595207213, "edge_loss_nfe_3": 0.1797650545835495, "diag_loss_nfe_3": 0.17634625285863875, "jvp_rms_nfe_3": 1.9045085847377776, "loss_nfe_4": 0.3894401788711548, "edge_loss_nfe_4": 0.19585056602954865, "diag_loss_nfe_4": 0.19358961880207062, "jvp_rms_nfe_4": 2.2923266768455504, "loss_nfe_5": 0.3539096057415009, "edge_loss_nfe_5": 0.17776780277490617, "diag_loss_nfe_5": 0.1761417955160141, "jvp_rms_nfe_5": 1.5124891102313995, "raw_loss": 0.37814890921115873, "raw_edge_loss": 0.19729300439357758, "raw_diag_loss": 0.1808559051156044, "raw_jvp_rms": 2.621889500617981, "raw_loss_nfe_1": 0.4127004086971283, "raw_edge_loss_nfe_1": 0.23728247880935668, "raw_diag_loss_nfe_1": 0.17541792690753938, "raw_jvp_rms_nfe_1": 3.186359691619873, "raw_loss_nfe_2": 0.37278689742088317, "raw_edge_loss_nfe_2": 0.19128688871860505, "raw_diag_loss_nfe_2": 0.18150001168251037, "raw_jvp_rms_nfe_2": 2.9029066443443297, "raw_loss_nfe_3": 0.3582378625869751, "raw_edge_loss_nfe_3": 0.18145101070404052, "raw_diag_loss_nfe_3": 0.1767868474125862, "raw_jvp_rms_nfe_3": 2.349283879995346, "raw_loss_nfe_4": 0.3917541980743408, "raw_edge_loss_nfe_4": 0.19763608276844025, "raw_diag_loss_nfe_4": 0.1941181242465973, "raw_jvp_rms_nfe_4": 2.840165650844574, "raw_loss_nfe_5": 0.3552651792764664, "raw_edge_loss_nfe_5": 0.17880856096744538, "raw_diag_loss_nfe_5": 0.17645661532878876, "raw_jvp_rms_nfe_5": 1.830731636285782} |
| {"step": 43000, "weights_source": "ema", "loss": 0.37205516755580903, "edge_loss": 0.1917748808860779, "diag_loss": 0.18028028905391694, "jvp_rms": 2.125342252254486, "examples": 5120, "loss_nfe_1": 0.3925249338150024, "edge_loss_nfe_1": 0.21763605773448944, "diag_loss_nfe_1": 0.17488887906074524, "jvp_rms_nfe_1": 2.5001804351806642, "loss_nfe_2": 0.3691649556159973, "edge_loss_nfe_2": 0.1882774442434311, "diag_loss_nfe_2": 0.18088751137256623, "jvp_rms_nfe_2": 2.3510846734046935, "loss_nfe_3": 0.35585449934005736, "edge_loss_nfe_3": 0.17965261936187743, "diag_loss_nfe_3": 0.1762018769979477, "jvp_rms_nfe_3": 1.92358438372612, "loss_nfe_4": 0.3890950262546539, "edge_loss_nfe_4": 0.19568087458610534, "diag_loss_nfe_4": 0.1934141606092453, "jvp_rms_nfe_4": 2.3233995378017425, "loss_nfe_5": 0.35363642275333407, "edge_loss_nfe_5": 0.17762740850448608, "diag_loss_nfe_5": 0.1760090172290802, "jvp_rms_nfe_5": 1.5284622311592102, "raw_loss": 0.37374089658260345, "raw_edge_loss": 0.1935477900505066, "raw_diag_loss": 0.1801931056380272, "raw_jvp_rms": 2.299976586103439, "raw_loss_nfe_1": 0.39992823004722594, "raw_edge_loss_nfe_1": 0.2250307857990265, "raw_diag_loss_nfe_1": 0.17489744424819947, "raw_jvp_rms_nfe_1": 2.7114959239959715, "raw_loss_nfe_2": 0.36982461214065554, "raw_edge_loss_nfe_2": 0.18907322287559508, "raw_diag_loss_nfe_2": 0.18075139224529266, "raw_jvp_rms_nfe_2": 2.5491456270217894, "raw_loss_nfe_3": 0.3560143828392029, "raw_edge_loss_nfe_3": 0.17990127801895142, "raw_diag_loss_nfe_3": 0.17611309438943862, "raw_jvp_rms_nfe_3": 2.0785558223724365, "raw_loss_nfe_4": 0.3891355276107788, "raw_edge_loss_nfe_4": 0.19592745006084442, "raw_diag_loss_nfe_4": 0.19320808351039886, "raw_jvp_rms_nfe_4": 2.514833205938339, "raw_loss_nfe_5": 0.3538017302751541, "raw_edge_loss_nfe_5": 0.17780621349811554, "raw_diag_loss_nfe_5": 0.17599551379680634, "raw_jvp_rms_nfe_5": 1.6458523511886596} |
| {"step": 44000, "weights_source": "ema", "loss": 0.3718108052015304, "edge_loss": 0.19166429877281188, "diag_loss": 0.1801465055346489, "jvp_rms": 2.1521597826480865, "examples": 5120, "loss_nfe_1": 0.3922906398773193, "edge_loss_nfe_1": 0.2175253540277481, "diag_loss_nfe_1": 0.17476528584957124, "jvp_rms_nfe_1": 2.536211538314819, "loss_nfe_2": 0.3689300060272217, "edge_loss_nfe_2": 0.18817995488643646, "diag_loss_nfe_2": 0.18075005412101747, "jvp_rms_nfe_2": 2.3827747225761415, "loss_nfe_3": 0.35562258362770083, "edge_loss_nfe_3": 0.17955487966537476, "diag_loss_nfe_3": 0.17606770247220993, "jvp_rms_nfe_3": 1.947074192762375, "loss_nfe_4": 0.38880528807640075, "edge_loss_nfe_4": 0.19554311633110047, "diag_loss_nfe_4": 0.19326217472553253, "jvp_rms_nfe_4": 2.3517152607440948, "loss_nfe_5": 0.3534055083990097, "edge_loss_nfe_5": 0.17751818895339966, "diag_loss_nfe_5": 0.17588731050491332, "jvp_rms_nfe_5": 1.5430231988430023, "raw_loss": 0.37199350357055666, "raw_edge_loss": 0.19197442293167116, "raw_diag_loss": 0.18001908123493196, "raw_jvp_rms": 2.215284506082535, "raw_loss_nfe_1": 0.3944181323051453, "raw_edge_loss_nfe_1": 0.2197323888540268, "raw_diag_loss_nfe_1": 0.1746857464313507, "raw_jvp_rms_nfe_1": 2.620998001098633, "raw_loss_nfe_2": 0.3687027633190155, "raw_edge_loss_nfe_2": 0.18810227811336516, "raw_diag_loss_nfe_2": 0.18060048818588256, "raw_jvp_rms_nfe_2": 2.459484040737152, "raw_loss_nfe_3": 0.35545286536216736, "raw_edge_loss_nfe_3": 0.17949930727481841, "raw_diag_loss_nfe_3": 0.17595355361700057, "raw_jvp_rms_nfe_3": 2.002295124530792, "raw_loss_nfe_4": 0.38823755383491515, "raw_edge_loss_nfe_4": 0.19519887268543243, "raw_diag_loss_nfe_4": 0.1930386781692505, "raw_jvp_rms_nfe_4": 2.4217930257320406, "raw_loss_nfe_5": 0.35315620303153994, "raw_edge_loss_nfe_5": 0.1773392677307129, "raw_diag_loss_nfe_5": 0.17581693977117538, "raw_jvp_rms_nfe_5": 1.5718523383140564} |
| {"step": 45000, "weights_source": "ema", "loss": 0.37168434262275696, "edge_loss": 0.19165305346250533, "diag_loss": 0.18003128856420517, "jvp_rms": 2.1736205995082853, "examples": 5120, "loss_nfe_1": 0.3925355851650238, "edge_loss_nfe_1": 0.21787453591823577, "diag_loss_nfe_1": 0.17466104328632354, "jvp_rms_nfe_1": 2.5622494220733643, "loss_nfe_2": 0.3687441825866699, "edge_loss_nfe_2": 0.18811105489730834, "diag_loss_nfe_2": 0.18063313663005828, "jvp_rms_nfe_2": 2.406940096616745, "loss_nfe_3": 0.35541791915893556, "edge_loss_nfe_3": 0.1794615715742111, "diag_loss_nfe_3": 0.17595634311437608, "jvp_rms_nfe_3": 1.9702612817287446, "loss_nfe_4": 0.3885439932346344, "edge_loss_nfe_4": 0.19541764855384827, "diag_loss_nfe_4": 0.1931263417005539, "jvp_rms_nfe_4": 2.373984634876251, "loss_nfe_5": 0.35318003296852113, "edge_loss_nfe_5": 0.17740045636892318, "diag_loss_nfe_5": 0.17577957808971406, "jvp_rms_nfe_5": 1.5546675622463226, "raw_loss": 0.3740403318405151, "raw_edge_loss": 0.1942245441675186, "raw_diag_loss": 0.17981578558683395, "raw_jvp_rms": 2.3283275413513183, "raw_loss_nfe_1": 0.4041799962520599, "raw_edge_loss_nfe_1": 0.2296959787607193, "raw_diag_loss_nfe_1": 0.1744840145111084, "raw_jvp_rms_nfe_1": 2.760013055801392, "raw_loss_nfe_2": 0.36933073997497556, "raw_edge_loss_nfe_2": 0.18892420828342438, "raw_diag_loss_nfe_2": 0.18040653467178344, "raw_jvp_rms_nfe_2": 2.587478423118591, "raw_loss_nfe_3": 0.3554180383682251, "raw_edge_loss_nfe_3": 0.17966652810573577, "raw_diag_loss_nfe_3": 0.1757515013217926, "raw_jvp_rms_nfe_3": 2.1034037828445435, "raw_loss_nfe_4": 0.38829761147499087, "raw_edge_loss_nfe_4": 0.1954770267009735, "raw_diag_loss_nfe_4": 0.19282058477401734, "raw_jvp_rms_nfe_4": 2.5383615255355836, "raw_loss_nfe_5": 0.35297527313232424, "raw_edge_loss_nfe_5": 0.1773589789867401, "raw_diag_loss_nfe_5": 0.175616292655468, "raw_jvp_rms_nfe_5": 1.652380919456482} |
| {"step": 46000, "weights_source": "ema", "loss": 0.3715553200244904, "edge_loss": 0.19162928521633146, "diag_loss": 0.1799260401725769, "jvp_rms": 2.201610367298126, "examples": 5120, "loss_nfe_1": 0.3926394283771515, "edge_loss_nfe_1": 0.2180699735879898, "diag_loss_nfe_1": 0.17456946074962615, "jvp_rms_nfe_1": 2.6001721382141114, "loss_nfe_2": 0.3686067283153534, "edge_loss_nfe_2": 0.18808059692382811, "diag_loss_nfe_2": 0.18052614033222197, "jvp_rms_nfe_2": 2.439925366640091, "loss_nfe_3": 0.35523309707641604, "edge_loss_nfe_3": 0.1793862372636795, "diag_loss_nfe_3": 0.17584685534238814, "jvp_rms_nfe_3": 1.992531567811966, "loss_nfe_4": 0.38829939961433413, "edge_loss_nfe_4": 0.19529785215854645, "diag_loss_nfe_4": 0.19300155341625214, "jvp_rms_nfe_4": 2.4034175872802734, "loss_nfe_5": 0.35299794673919677, "edge_loss_nfe_5": 0.17731176614761351, "diag_loss_nfe_5": 0.17568619102239608, "jvp_rms_nfe_5": 1.5720051765441894, "raw_loss": 0.37294554710388184, "raw_edge_loss": 0.19305831611156463, "raw_diag_loss": 0.17988722890615463, "raw_jvp_rms": 2.5546124279499054, "raw_loss_nfe_1": 0.39817399382591245, "raw_edge_loss_nfe_1": 0.2236364871263504, "raw_diag_loss_nfe_1": 0.17453750371932983, "raw_jvp_rms_nfe_1": 3.0601311683654786, "raw_loss_nfe_2": 0.36934937834739684, "raw_edge_loss_nfe_2": 0.18887795507907867, "raw_diag_loss_nfe_2": 0.1804714173078537, "raw_jvp_rms_nfe_2": 2.8419912278652193, "raw_loss_nfe_3": 0.35562482476234436, "raw_edge_loss_nfe_3": 0.17977170050144195, "raw_diag_loss_nfe_3": 0.17585312575101852, "raw_jvp_rms_nfe_3": 2.303293800354004, "raw_loss_nfe_4": 0.3885082483291626, "raw_edge_loss_nfe_4": 0.19556552171707153, "raw_diag_loss_nfe_4": 0.1929427295923233, "raw_jvp_rms_nfe_4": 2.7833980321884155, "raw_loss_nfe_5": 0.3530712902545929, "raw_edge_loss_nfe_5": 0.1774399161338806, "raw_diag_loss_nfe_5": 0.1756313681602478, "raw_jvp_rms_nfe_5": 1.78424791097641} |
| {"step": 47000, "weights_source": "ema", "loss": 0.3713958901166916, "edge_loss": 0.1915695336461067, "diag_loss": 0.1798263558745384, "jvp_rms": 2.2319657707214353, "examples": 5120, "loss_nfe_1": 0.3926237761974335, "edge_loss_nfe_1": 0.2181445300579071, "diag_loss_nfe_1": 0.1744792491197586, "jvp_rms_nfe_1": 2.642152261734009, "loss_nfe_2": 0.36840869188308717, "edge_loss_nfe_2": 0.18798398077487946, "diag_loss_nfe_2": 0.18042470812797545, "jvp_rms_nfe_2": 2.475356990098953, "loss_nfe_3": 0.3550576686859131, "edge_loss_nfe_3": 0.17930547297000884, "diag_loss_nfe_3": 0.175752192735672, "jvp_rms_nfe_3": 2.0165530383586883, "loss_nfe_4": 0.3880728900432587, "edge_loss_nfe_4": 0.19519352316856384, "diag_loss_nfe_4": 0.19287936985492707, "jvp_rms_nfe_4": 2.435269385576248, "loss_nfe_5": 0.35281642377376554, "edge_loss_nfe_5": 0.17722016125917434, "diag_loss_nfe_5": 0.17559625953435898, "jvp_rms_nfe_5": 1.5904971778392791, "raw_loss": 0.3728845959901809, "raw_edge_loss": 0.19330160319805145, "raw_diag_loss": 0.17958299428224564, "raw_jvp_rms": 2.488507117033005, "raw_loss_nfe_1": 0.4004963099956512, "raw_edge_loss_nfe_1": 0.226207235455513, "raw_diag_loss_nfe_1": 0.1742890805006027, "raw_jvp_rms_nfe_1": 2.9745230674743652, "raw_loss_nfe_2": 0.36872340440750123, "raw_edge_loss_nfe_2": 0.18855189383029938, "raw_diag_loss_nfe_2": 0.18017151057720185, "raw_jvp_rms_nfe_2": 2.7699862122535706, "raw_loss_nfe_3": 0.3549445480108261, "raw_edge_loss_nfe_3": 0.17942589223384858, "raw_diag_loss_nfe_3": 0.17551865577697753, "raw_jvp_rms_nfe_3": 2.246046853065491, "raw_loss_nfe_4": 0.3877571880817413, "raw_edge_loss_nfe_4": 0.19519511461257935, "raw_diag_loss_nfe_4": 0.192562073469162, "raw_jvp_rms_nfe_4": 2.705729341506958, "raw_loss_nfe_5": 0.3525015294551849, "raw_edge_loss_nfe_5": 0.17712787985801698, "raw_diag_loss_nfe_5": 0.1753736510872841, "raw_jvp_rms_nfe_5": 1.7462501108646393} |
| {"step": 48000, "weights_source": "ema", "loss": 0.3712373948097229, "edge_loss": 0.19149310767650604, "diag_loss": 0.17974428683519364, "jvp_rms": 2.250151746273041, "examples": 5120, "loss_nfe_1": 0.39254979491233827, "edge_loss_nfe_1": 0.21814591586589813, "diag_loss_nfe_1": 0.17440388202667237, "jvp_rms_nfe_1": 2.665694570541382, "loss_nfe_2": 0.3681997716426849, "edge_loss_nfe_2": 0.18785875737667085, "diag_loss_nfe_2": 0.1803410142660141, "jvp_rms_nfe_2": 2.495949363708496, "loss_nfe_3": 0.3549005001783371, "edge_loss_nfe_3": 0.17922913730144502, "diag_loss_nfe_3": 0.17567135840654374, "jvp_rms_nfe_3": 2.0347792744636535, "loss_nfe_4": 0.3878713846206665, "edge_loss_nfe_4": 0.19508881270885467, "diag_loss_nfe_4": 0.19278257191181183, "jvp_rms_nfe_4": 2.4547479271888735, "loss_nfe_5": 0.35266552269458773, "edge_loss_nfe_5": 0.17714291512966157, "diag_loss_nfe_5": 0.17552260756492616, "jvp_rms_nfe_5": 1.5995875954627992, "raw_loss": 0.37239629626274107, "raw_edge_loss": 0.19247002869844437, "raw_diag_loss": 0.17992626249790195, "raw_jvp_rms": 2.1689469707012177, "raw_loss_nfe_1": 0.396082466840744, "raw_edge_loss_nfe_1": 0.2214716374874115, "raw_diag_loss_nfe_1": 0.17461082339286804, "raw_jvp_rms_nfe_1": 2.5685171604156496, "raw_loss_nfe_2": 0.36995372772216795, "raw_edge_loss_nfe_2": 0.18943111300468446, "raw_diag_loss_nfe_2": 0.1805226117372513, "raw_jvp_rms_nfe_2": 2.4097091972827913, "raw_loss_nfe_3": 0.35518380999565125, "raw_edge_loss_nfe_3": 0.17940330505371094, "raw_diag_loss_nfe_3": 0.17578050196170808, "raw_jvp_rms_nfe_3": 1.9646742224693299, "raw_loss_nfe_4": 0.3877831220626831, "raw_edge_loss_nfe_4": 0.19488065540790558, "raw_diag_loss_nfe_4": 0.19290245771408082, "raw_jvp_rms_nfe_4": 2.366553395986557, "raw_loss_nfe_5": 0.3529783546924591, "raw_edge_loss_nfe_5": 0.17716343253850936, "raw_diag_loss_nfe_5": 0.17581491768360138, "raw_jvp_rms_nfe_5": 1.5352808773517608} |
| {"step": 49000, "weights_source": "ema", "loss": 0.3710547721385956, "edge_loss": 0.19137788712978362, "diag_loss": 0.17967688530683518, "jvp_rms": 2.26550262093544, "examples": 5120, "loss_nfe_1": 0.39221864342689516, "edge_loss_nfe_1": 0.21786801517009735, "diag_loss_nfe_1": 0.17435062825679778, "jvp_rms_nfe_1": 2.686102199554443, "loss_nfe_2": 0.3680577754974365, "edge_loss_nfe_2": 0.1877869725227356, "diag_loss_nfe_2": 0.18027080297470094, "jvp_rms_nfe_2": 2.5132674276828766, "loss_nfe_3": 0.3547529995441437, "edge_loss_nfe_3": 0.17915709614753722, "diag_loss_nfe_3": 0.17559590488672255, "jvp_rms_nfe_3": 2.0510569334030153, "loss_nfe_4": 0.3877130031585693, "edge_loss_nfe_4": 0.19501143097877502, "diag_loss_nfe_4": 0.19270157516002656, "jvp_rms_nfe_4": 2.470135527849197, "loss_nfe_5": 0.3525314390659332, "edge_loss_nfe_5": 0.17706592082977296, "diag_loss_nfe_5": 0.17546551525592805, "jvp_rms_nfe_5": 1.606951016187668, "raw_loss": 0.3729947149753571, "raw_edge_loss": 0.19341899901628495, "raw_diag_loss": 0.17957571774721146, "raw_jvp_rms": 2.3479965436458587, "raw_loss_nfe_1": 0.4013635516166687, "raw_edge_loss_nfe_1": 0.22706393301486968, "raw_diag_loss_nfe_1": 0.17429961562156676, "raw_jvp_rms_nfe_1": 2.7829211711883546, "raw_loss_nfe_2": 0.3686074197292328, "raw_edge_loss_nfe_2": 0.18846496045589448, "raw_diag_loss_nfe_2": 0.18014245927333833, "raw_jvp_rms_nfe_2": 2.6206358075141907, "raw_loss_nfe_3": 0.35482444763183596, "raw_edge_loss_nfe_3": 0.17929541170597077, "raw_diag_loss_nfe_3": 0.17552904039621353, "raw_jvp_rms_nfe_3": 2.1178201019763945, "raw_loss_nfe_4": 0.3876810848712921, "raw_edge_loss_nfe_4": 0.19514916241168975, "raw_diag_loss_nfe_4": 0.19253192245960235, "raw_jvp_rms_nfe_4": 2.551815760135651, "raw_loss_nfe_5": 0.35249707102775574, "raw_edge_loss_nfe_5": 0.17712152749300003, "raw_diag_loss_nfe_5": 0.1753755509853363, "raw_jvp_rms_nfe_5": 1.6667898774147034} |
| {"step": 50000, "weights_source": "ema", "loss": 0.3709731566905975, "edge_loss": 0.19136443853378296, "diag_loss": 0.1796087145805359, "jvp_rms": 2.2817168676853177, "examples": 5120, "loss_nfe_1": 0.39239816665649413, "edge_loss_nfe_1": 0.21810964345932007, "diag_loss_nfe_1": 0.1742885172367096, "jvp_rms_nfe_1": 2.700194501876831, "loss_nfe_2": 0.3679115891456604, "edge_loss_nfe_2": 0.18770996630191802, "diag_loss_nfe_2": 0.18020162284374236, "jvp_rms_nfe_2": 2.536956226825714, "loss_nfe_3": 0.35463637113571167, "edge_loss_nfe_3": 0.1791027843952179, "diag_loss_nfe_3": 0.17553358674049377, "jvp_rms_nfe_3": 2.0643750607967375, "loss_nfe_4": 0.38753209710121156, "edge_loss_nfe_4": 0.19491184055805205, "diag_loss_nfe_4": 0.19262024462223054, "jvp_rms_nfe_4": 2.4871247947216033, "loss_nfe_5": 0.3523875594139099, "edge_loss_nfe_5": 0.17698795795440675, "diag_loss_nfe_5": 0.17539960145950317, "jvp_rms_nfe_5": 1.6199337542057037, "raw_loss": 0.37272677540779114, "raw_edge_loss": 0.19294691383838652, "raw_diag_loss": 0.17977986514568328, "raw_jvp_rms": 2.4338310968875883, "raw_loss_nfe_1": 0.39853897094726565, "raw_edge_loss_nfe_1": 0.2240687519311905, "raw_diag_loss_nfe_1": 0.1744702249765396, "raw_jvp_rms_nfe_1": 2.9251581192016602, "raw_loss_nfe_2": 0.3692192077636719, "raw_edge_loss_nfe_2": 0.18882804214954377, "raw_diag_loss_nfe_2": 0.18039117157459258, "raw_jvp_rms_nfe_2": 2.7281934142112734, "raw_loss_nfe_3": 0.35528815984725953, "raw_edge_loss_nfe_3": 0.1796249717473984, "raw_diag_loss_nfe_3": 0.17566319108009337, "raw_jvp_rms_nfe_3": 2.1835286557674407, "raw_loss_nfe_4": 0.3877676367759705, "raw_edge_loss_nfe_4": 0.19502896666526795, "raw_diag_loss_nfe_4": 0.1927386701107025, "raw_jvp_rms_nfe_4": 2.6403229236602783, "raw_loss_nfe_5": 0.3528199017047882, "raw_edge_loss_nfe_5": 0.1771838366985321, "raw_diag_loss_nfe_5": 0.17563606798648834, "raw_jvp_rms_nfe_5": 1.69195237159729} |
|
|
