diff --git a/generator/outputs/logs/p3_1_vae.json b/generator/outputs/logs/p3_1_vae.json
index 0a3ee94..23edc6b 100644
--- a/generator/outputs/logs/p3_1_vae.json
+++ b/generator/outputs/logs/p3_1_vae.json
@@ -17,216 +17,218 @@
     "model": "vae",
     "latent_dim": 256,
     "ngf": 64,
+    "free_bits": 0.1,
+    "grad_clip": 1.0,
     "run_name": "p3_1_vae",
-    "lr": 0.001,
-    "beta_kl": 1.0,
+    "lr": 0.0005,
+    "beta_kl": 0.005,
     "lambda_perceptual": 0.0,
     "lambda_adversarial": 0.0
   },
   "history": {
     "recon_loss": [
-      0.23614721197603095,
-      0.23315699178821,
-      0.22991716011594504,
-      NaN,
-      0.23217070787253544,
-      0.23155480842941847,
-      0.23157141198459855,
-      0.23181156750418183,
-      0.23201335527193853,
-      0.23178868266379732,
-      0.2315022333755962,
-      0.2311908418042028,
-      0.23185610672474927,
-      0.23176095832107413,
-      0.23165411693163407,
-      0.23174296459581098,
-      0.2317636658747991,
-      0.2317118427883356,
-      0.23172695364834917,
-      0.2316696329567677,
-      0.23168399261358458,
-      0.2316194716681782,
-      0.23164867447354856,
-      0.2315481170757204,
-      0.23165068109957582,
-      0.23167062098653907,
-      0.23162642907765177,
-      0.2315922882567104,
-      0.2315914996414103,
-      0.23156180984189367,
-      0.23156551628286004,
-      0.2315698005259037,
-      0.2315660522470617,
-      0.23156735001720935,
-      0.23161396435183337,
-      0.23158050178844705,
-      0.23159921089680785,
-      0.23149616745674712,
-      0.23159087484336308,
-      0.23156312872201967,
-      0.23153820200863048,
-      0.2315863819203825,
-      0.23150022140043414,
-      0.23154497337646973,
-      0.2315601774961011,
-      0.23153368950399578,
-      0.23152085642019907,
-      0.23151608884461924,
-      0.23154898990805334,
-      0.23155892872784892,
-      NaN,
-      NaN,
-      NaN,
-      0.24157701413600874,
-      NaN,
-      NaN,
-      0.24151325464630738,
-      NaN,
-      0.24154121766233036,
-      0.24155463749526912,
-      0.24158300176008135,
-      0.24158118757554609,
-      0.2415294518901242,
-      0.24156020069096842,
-      0.2415176352374574,
-      0.2415566616015047,
-      NaN,
-      0.24161437115608117,
-      0.24159398913765565,
-      0.24149432768806434,
-      0.24153172199287984,
-      0.24161516999204954,
-      0.24158193846034187,
-      0.2415451397562129,
-      0.24155487772873324,
-      0.24155297130346298,
-      NaN,
-      0.24157197961313093,
-      NaN,
-      NaN,
-      0.24158605401459923,
-      0.24156368870893094,
-      0.24159100852333582,
-      0.24153350121699846,
-      0.24153158377505776,
-      NaN,
-      0.24161708673350832,
-      0.24158515879868442,
-      NaN,
-      0.24157126235146809,
-      0.24162366709265953,
-      NaN,
-      0.2415581897665293,
-      NaN,
-      NaN,
-      0.2415400046823371,
-      NaN,
-      0.2415627600927638,
-      0.2415567432076503,
-      0.2415620140476614
+      0.07730471308573954,
+      0.06593808977522402,
+      0.06487635838297698,
+      0.06536683298520043,
+      0.06676436582596129,
+      0.06822305706799285,
+      0.07009286439826345,
+      0.0716893181968958,
+      0.07343342525359148,
+      0.075231988938191,
+      0.07698834319718373,
+      0.0781789975734348,
+      0.07966307190875722,
+      0.08127817829959413,
+      0.08283065850090267,
+      0.08466114240706477,
+      0.08580355486299238,
+      0.08649945996192276,
+      0.08722176170374593,
+      0.0878506902541615,
+      0.08772704952484013,
+      0.0878396618187937,
+      0.08760359356354953,
+      0.08755779992311429,
+      0.08766099331406955,
+      0.08788286568198958,
+      0.0880148638612949,
+      0.0880719024496965,
+      0.08799157871140374,
+      0.08789104575084315,
+      0.08781521651161532,
+      0.08772920075262713,
+      0.08761804038260737,
+      0.08748868394356507,
+      0.0876523340996514,
+      0.08761706726991722,
+      0.08738589645005189,
+      0.08738058178215964,
+      0.08751415275037289,
+      0.08737453581112573,
+      0.08756884259100144,
+      0.08753167293392695,
+      0.08771920282170813,
+      0.08771870020999868,
+      0.08811150293829095,
+      0.08813091252858822,
+      0.08838469075023109,
+      0.08847005421725604,
+      0.08863859883167295,
+      0.08869685632232417,
+      0.08851750366962872,
+      0.08844129702983758,
+      0.08849497742823556,
+      0.08820333027750508,
+      0.0882302947087675,
+      0.08809250190408312,
+      0.08821920114449966,
+      0.08808790903506625,
+      0.08793783823075967,
+      0.08809815778627865,
+      0.08792639472800443,
+      0.08773358266514081,
+      0.08771742116182278,
+      0.08763711888374937,
+      0.08763587549647205,
+      0.08757383744112957,
+      0.08740564774817382,
+      0.08760064023618515,
+      0.08756772452631058,
+      0.08748207171248575,
+      0.08743569531883949,
+      0.08756201617164999,
+      0.08752684823722921,
+      0.08740174391458178,
+      0.08721604804771069,
+      0.08738407018220323,
+      0.08733521637498823,
+      0.0873828605772593,
+      0.08724558114623412,
+      0.0872362587823827,
+      0.08704395339083977,
+      0.0871862175468451,
+      0.08712937674907029,
+      0.08717458137971723,
+      0.08691909113246152,
+      0.08707771377048941,
+      0.0870408087523065,
+      0.08690907467061128,
+      0.08700524977384469,
+      0.08686018860938712,
+      0.08692594942374107,
+      0.08688000118375844,
+      0.08683701932557628,
+      0.08682236053113244,
+      0.08666048966284491,
+      0.0866338494591988,
+      0.08675910368498065,
+      0.08661951002083783,
+      0.0864848897466038,
+      0.08664045786946757
     ],
     "kl_loss": [
-      12.394881742504927,
-      184.775765717539,
-      127.26797539963681,
-      33346392.786626913,
-      35.72433020722153,
-      31.41954361882984,
-      16.178619678203876,
-      10.234501274223001,
-      14.817130448471787,
-      9.230570034084158,
-      9.643558593896719,
-      8.47786058498244,
-      5.573643362929678,
-      2.4644629534365783,
-      1.5757666807462516,
-      0.426466258131286,
-      1.7924597560404203,
-      0.2769168242652956,
-      0.21636260826236162,
-      0.48804672485870176,
-      0.10833573165453142,
-      0.13318477837671328,
-      0.17373992544877478,
-      0.09584700099678121,
-      0.0977757986014088,
-      0.07794108981282538,
-      0.05691333960853199,
-      0.07221067506167242,
-      0.036222075203704275,
-      0.03126689469696492,
-      0.04264315036642882,
-      0.016960328184147805,
-      0.03314871971324309,
-      0.014776984407789368,
-      0.011375301962312406,
-      0.013948339588828703,
-      0.01186063720120324,
-      0.0099704478863372,
-      0.00536374123289417,
-      0.009618068660179583,
-      0.00418840028031164,
-      0.004865833775052785,
-      0.005830266629345715,
-      0.0023000687699064487,
-      0.0038261460966199762,
-      0.0022056369562673136,
-      0.002220870125003987,
-      0.0024217167485139184,
-      0.001954249278483037,
-      0.0021431104709895756,
-      0.0022583500309011494,
-      0.002132287005193404,
-      56.80083886633675,
-      82.57108385134966,
-      82.55195800259582,
-      82.57428529527452,
-      82.56009972401155,
-      82.55269345666608,
-      82.57006728343474,
-      82.55670593131302,
-      82.54445134676419,
-      82.57745079301361,
-      82.57933913336859,
-      82.5570435157189,
-      82.56808758597089,
-      82.56800172267816,
-      82.56525711320404,
-      82.56189481621115,
-      82.55193622295673,
-      82.55375865382007,
-      82.56600202250685,
-      82.57064581324912,
-      82.55481151026538,
-      82.55367833324986,
-      82.56042112040724,
-      82.5616829048874,
-      82.5771528553759,
-      82.55317820035495,
-      82.57550573756552,
-      82.57334061973116,
-      82.56044387817383,
-      82.5752662593483,
-      82.56673936762361,
-      82.56828115740393,
-      82.56990289280557,
-      82.55218840052939,
-      82.56695372426611,
-      82.575043066954,
-      82.55754522991995,
-      82.56361721723508,
-      82.5628145821074,
-      82.56431990403395,
-      82.55777725806603,
-      82.5742861918914,
-      82.56361025622768,
-      82.56887233766736,
-      82.56539458902473,
-      82.55887828729091,
-      82.56073884882478,
-      82.55578186165573
+      97.64885638310359,
+      62.855894651168434,
+      51.748125516451324,
+      46.85192030719203,
+      43.88861534738133,
+      41.6780331440461,
+      40.001986854096764,
+      38.64640927110982,
+      37.41358267955291,
+      36.41640040405795,
+      35.55877452426486,
+      34.7982840415759,
+      34.11121509421585,
+      33.47962024884346,
+      32.765580996488914,
+      32.12892857983581,
+      31.5787715096759,
+      31.23355184457241,
+      30.95730068337204,
+      30.6620740116152,
+      30.60565506698739,
+      30.529209638253235,
+      30.491252128894512,
+      30.418221693772537,
+      30.34073096870357,
+      30.201212308345696,
+      30.061584020272278,
+      29.970419724782307,
+      29.8875841523847,
+      29.83627580985045,
+      29.782486483582066,
+      29.725191291580852,
+      29.67253409491645,
+      29.62634592382317,
+      29.588209629058838,
+      29.532620438143738,
+      29.495509889390732,
+      29.449343318613167,
+      29.384501049661228,
+      29.349107827895725,
+      29.295700040637936,
+      29.241306573916706,
+      29.166094502832134,
+      29.11005148113283,
+      29.028348319550865,
+      28.94603967666626,
+      28.882984246963108,
+      28.78164738467616,
+      28.735162518982193,
+      28.679843005971012,
+      28.659660514603313,
+      28.619067306192512,
+      28.596742373246414,
+      28.562309256985657,
+      28.52451869768974,
+      28.490606548439743,
+      28.49845803497184,
+      28.45612907002115,
+      28.454183386941242,
+      28.419613467322456,
+      28.41504774745713,
+      28.364814460787,
+      28.368678023672512,
+      28.338627220219017,
+      28.325380044105724,
+      28.32440700694027,
+      28.297185176458115,
+      28.266021508436936,
+      28.25893593649579,
+      28.229272165869038,
+      28.20741213692559,
+      28.18922700229873,
+      28.178358025021023,
+      28.16814616195157,
+      28.143704161684738,
+      28.140413304679413,
+      28.134240329775036,
+      28.117950944819,
+      28.1009096161932,
+      28.077010008004997,
+      28.061608836182163,
+      28.044516302581528,
+      28.028144938314064,
+      28.01604707016904,
+      27.992277870830307,
+      27.991227651253723,
+      27.98667734097212,
+      27.938560457311127,
+      27.945909381931664,
+      27.928414560790756,
+      27.932809756352352,
+      27.928062125148937,
+      27.91421982773349,
+      27.90757109161116,
+      27.89331733671009,
+      27.897906148535572,
+      27.880407887646275,
+      27.88155363360022,
+      27.862588609385693,
+      27.864495530087723
     ],
     "perc_loss": [
       0.0,
@@ -535,12 +537,12 @@
       0.0
     ],
     "fid": {
-      "25": 315.9393615722656,
-      "50": 419.273193359375,
-      "75": 360.4432678222656,
-      "100": 363.9911193847656
+      "25": 244.70846557617188,
+      "50": 227.29347229003906,
+      "75": 228.68997192382812,
+      "100": 234.52761840820312
     },
-    "train_time_s": 660.9630489349365
+    "train_time_s": 588.5088336467743
   },
   "n_params": 10608451
 }
\ No newline at end of file
diff --git a/generator/outputs/logs/p3_2_vae_perceptual.json b/generator/outputs/logs/p3_2_vae_perceptual.json
index 8c982bf..ce68137 100644
--- a/generator/outputs/logs/p3_2_vae_perceptual.json
+++ b/generator/outputs/logs/p3_2_vae_perceptual.json
@@ -17,318 +17,320 @@
     "model": "vae",
     "latent_dim": 256,
     "ngf": 64,
+    "free_bits": 0.1,
+    "grad_clip": 1.0,
     "run_name": "p3_2_vae_perceptual",
-    "lr": 0.001,
-    "beta_kl": 0.0001,
+    "lr": 0.0005,
+    "beta_kl": 0.005,
     "lambda_perceptual": 0.1,
     "lambda_adversarial": 0.0
   },
   "history": {
     "recon_loss": [
-      NaN,
-      0.041370747770127066,
-      0.035506031866002284,
-      0.06229733923672993,
-      0.06089382184048494,
-      0.053464704654856116,
-      0.04950355895213846,
-      0.04674786329269409,
-      0.05693557829811023,
-      0.04808994451076047,
-      NaN,
-      0.05869839608701121,
-      0.05293231666820426,
-      0.05057006603918779,
-      0.04713928615117175,
-      0.04481589820427008,
-      0.04331832689543565,
-      0.041820655449524395,
-      NaN,
-      0.24315394992884407,
-      0.24305414841470555,
-      0.2429187158998261,
-      0.24289727962424612,
-      0.2377373814328104,
-      0.23880945107875726,
-      0.24047312904626894,
-      0.23915709144411942,
-      NaN,
-      0.24000247061634675,
-      0.2362435321904655,
-      0.23705266075383905,
-      NaN,
-      0.23998981039238793,
-      0.23828768376738596,
-      0.23908851302077627,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN
+      0.07706982396447505,
+      0.06330727752393638,
+      0.060562346210209735,
+      0.06067247126792741,
+      0.06168853705669315,
+      0.06303556570703657,
+      0.06482723723842293,
+      0.06669721933097666,
+      0.0683129624200937,
+      0.06987012332926194,
+      0.07130966042606239,
+      0.07270127332681774,
+      0.07425614913654888,
+      0.07580115537867586,
+      0.07719200621876451,
+      0.078589656127569,
+      0.07977109738331065,
+      0.08097509191268021,
+      0.0821382127638556,
+      0.08324004981953365,
+      0.08320301078642026,
+      0.08301973822088833,
+      0.08318755321968825,
+      0.08317961664790781,
+      0.08314789212348624,
+      0.08329373801875319,
+      0.08338832993728992,
+      0.08336645318593225,
+      0.08318959130372247,
+      0.08320376128913501,
+      0.08317808063430154,
+      0.08312527894273274,
+      0.08306075425611602,
+      0.08289793944065897,
+      0.0828752614175662,
+      0.08287284457976492,
+      0.08268047039771181,
+      0.08271352856014019,
+      0.0829078034362477,
+      0.0827700739614984,
+      0.0827989391823355,
+      0.08292064927199967,
+      0.0828404097029796,
+      0.08288907250150657,
+      0.0829680990268532,
+      0.08282275031456071,
+      0.08290983905267511,
+      0.08281088118942884,
+      0.08267007887554474,
+      0.08269618300354889,
+      0.0827041374057786,
+      0.0825083223131732,
+      0.08238164631602092,
+      0.08237539884498996,
+      0.0822869433870173,
+      0.08221659919199271,
+      0.08213427105647886,
+      0.08218342011683008,
+      0.082064455613876,
+      0.08187834614426152,
+      0.08198784355424409,
+      0.08176947068263832,
+      0.08174051175642218,
+      0.08147429703519894,
+      0.08165825637550946,
+      0.08149343700363086,
+      0.08137227281227581,
+      0.08143208594594756,
+      0.08119541610408033,
+      0.08130555122326581,
+      0.08124906267238478,
+      0.08108980721260747,
+      0.08119610937423685,
+      0.08095862009586433,
+      0.08084884440351246,
+      0.08087979268250811,
+      0.08086979661423427,
+      0.08077665719275291,
+      0.08070505911914203,
+      0.08077280767835103,
+      0.0806472651564922,
+      0.08055042177757137,
+      0.08046174721203299,
+      0.08041473021173579,
+      0.08053297676846512,
+      0.08046999694699915,
+      0.0805438561603809,
+      0.0802764052636603,
+      0.08026964488065141,
+      0.08028597289170974,
+      0.0801090358191321,
+      0.08017482723181064,
+      0.08005877630387107,
+      0.08010221279075003,
+      0.07999415299258171,
+      0.0798476467975694,
+      0.07997323754124153,
+      0.0798028621217634,
+      0.07988279134544551,
+      0.07989695622052392
     ],
     "kl_loss": [
-      41526536.5961082,
-      1258.7768262553418,
-      1165.5987154968784,
-      3689.321748260759,
-      1320.3167513333833,
-      928.3473894168169,
-      789.754654843583,
-      693.4729607736963,
-      1017.8856519389357,
-      622.2623097998464,
-      304942.2120989938,
-      750.5718396830763,
-      618.6888906364767,
-      586.4194498958751,
-      579.2793892102363,
-      525.8830437945504,
-      497.5363791050055,
-      452.93342662061383,
-      172476214612.95245,
-      1112.027792645316,
-      1109.8156276605068,
-      1109.8872808472722,
-      1112.3999198196282,
-      1163.6093267457097,
-      1195.4633280436199,
-      1232.709252512353,
-      1253.0651078183428,
-      19923.65347159622,
-      6871.588723207132,
-      6872.017739842081,
-      6872.62934758113,
-      27062.17111336472,
-      4992.576698759683,
-      5000.895510942508,
-      5011.723715236044,
-      4.196954763212122e+20,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN
+      114.89598556257721,
+      77.33424778473683,
+      62.9100211705917,
+      55.875159540746964,
+      51.501892236562874,
+      48.34236133608044,
+      45.885433547517174,
+      43.94423294067383,
+      42.245688601436775,
+      40.89320367829413,
+      39.69322418147682,
+      38.6090751876179,
+      37.7343929486397,
+      36.89236228486411,
+      36.14446016458365,
+      35.46583691621438,
+      34.81329877967509,
+      34.25002269255809,
+      33.78917465862046,
+      33.34034808069213,
+      33.22343346196362,
+      33.14858524615948,
+      33.055273805928024,
+      32.99898836347792,
+      32.90522269713573,
+      32.836681887634796,
+      32.76146248874501,
+      32.68683499556322,
+      32.632087422232345,
+      32.59484779325306,
+      32.54658634642251,
+      32.49264952260205,
+      32.471018106509476,
+      32.43708103946132,
+      32.433846559280006,
+      32.40765173822387,
+      32.363136197766686,
+      32.320940767597946,
+      32.29645855813964,
+      32.26480372746786,
+      32.22680737829616,
+      32.176548860011955,
+      32.135557647444244,
+      32.11595243877835,
+      32.05515718867636,
+      32.04982715997941,
+      32.00999742084079,
+      31.98886978931916,
+      31.965849428095368,
+      31.92324785085825,
+      31.916118495484703,
+      31.915471476367397,
+      31.903850473909298,
+      31.890775949527058,
+      31.8645691993909,
+      31.869942917783035,
+      31.860487742301746,
+      31.850944148169624,
+      31.843142101907322,
+      31.839001268403145,
+      31.847271275316547,
+      31.829896576384193,
+      31.801259293515457,
+      31.81849612130059,
+      31.774390408116528,
+      31.791951089842705,
+      31.79695913526747,
+      31.78303485038953,
+      31.772433419512886,
+      31.75331918602316,
+      31.766783274137058,
+      31.768596893701798,
+      31.765920822436993,
+      31.724387494926777,
+      31.745759788741413,
+      31.730095512846596,
+      31.737777669205624,
+      31.726243728246445,
+      31.71437902532072,
+      31.71155218792777,
+      31.704629376403286,
+      31.707098475888245,
+      31.703360154078556,
+      31.695194660088955,
+      31.685226285559498,
+      31.6823458956857,
+      31.674420800983395,
+      31.686162447318054,
+      31.676387077722794,
+      31.680302566952175,
+      31.657646749773598,
+      31.67580248759343,
+      31.65489924259675,
+      31.662759666768913,
+      31.655795794266922,
+      31.663332922845825,
+      31.652285494356075,
+      31.666093015263222,
+      31.64289121546297,
+      31.66365980083107
     ],
     "perc_loss": [
-      NaN,
-      3.0017659954535656,
-      2.8931196978968434,
-      3.084328340159522,
-      3.1398269641093717,
-      3.0839009172896032,
-      3.0459561949102287,
-      3.015714469628456,
-      3.0917934143645134,
-      3.010985380054539,
-      NaN,
-      3.102917709411719,
-      3.055045795746339,
-      3.0351512865123587,
-      3.002274121993627,
-      2.9754957128793764,
-      2.9553630066733074,
-      2.936704080328982,
-      NaN,
-      3.66463458385223,
-      3.6419444985878773,
-      3.6321474706005845,
-      3.6272282503608966,
-      3.781295938878997,
-      3.740793755421272,
-      3.759704489993234,
-      3.7078149914741516,
-      NaN,
-      3.7647176323792872,
-      3.7441278461717133,
-      3.7264530261357627,
-      NaN,
-      3.7299226170931106,
-      3.7314435080585318,
-      3.754668933713538,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN
+      3.347367505232493,
+      3.1841943034758935,
+      3.1360840721008105,
+      3.1123677431008754,
+      3.102960804079333,
+      3.0996367656267605,
+      3.100349969843514,
+      3.1043578926314654,
+      3.1047080194848213,
+      3.10818816963424,
+      3.1107087919854712,
+      3.11433150116195,
+      3.1165808979262652,
+      3.1217524506087995,
+      3.124025124769944,
+      3.1286499158948913,
+      3.1313081755597367,
+      3.135032072535947,
+      3.1376487164415865,
+      3.140188893191835,
+      3.138313090699351,
+      3.136352961898869,
+      3.1350333721209793,
+      3.1340697214134736,
+      3.1320774601056027,
+      3.1314907603793674,
+      3.1301532255278692,
+      3.12954652411306,
+      3.12765454991251,
+      3.126848427658407,
+      3.1259387716268883,
+      3.1238634581239815,
+      3.1228250631919274,
+      3.1218308898118825,
+      3.120684378167503,
+      3.1207721839603195,
+      3.1186747016050878,
+      3.118101167373168,
+      3.11715858308678,
+      3.1159399615393744,
+      3.1157061961980967,
+      3.115308819163559,
+      3.114226241906484,
+      3.113662652989738,
+      3.112562283491477,
+      3.1123410055780005,
+      3.1108450038820252,
+      3.110834915413816,
+      3.11004300453724,
+      3.1096309413257828,
+      3.1089638593869333,
+      3.1076014179449816,
+      3.1071455493951454,
+      3.1054801660725193,
+      3.1046676870085235,
+      3.1034356477933054,
+      3.1028999105477943,
+      3.10203901773844,
+      3.100277382084447,
+      3.0997838067193317,
+      3.0997486939797034,
+      3.0976629460978713,
+      3.096854481432173,
+      3.0961881641648774,
+      3.0957974134347377,
+      3.094707619940114,
+      3.0938115603903418,
+      3.0930811013930883,
+      3.092930595589499,
+      3.091999230731247,
+      3.0908807531381264,
+      3.0901810919117723,
+      3.089734352551974,
+      3.0887419851417217,
+      3.087375036162189,
+      3.088294555998256,
+      3.087375529301472,
+      3.0865699587724147,
+      3.085814117366432,
+      3.086571911970774,
+      3.0848990953885593,
+      3.0839059826655264,
+      3.0831500668810983,
+      3.082290298408932,
+      3.083427106213366,
+      3.0819556769142804,
+      3.0830693351916776,
+      3.080666415202312,
+      3.0801111816341042,
+      3.080527792119572,
+      3.0795028067042685,
+      3.078920972143483,
+      3.078586568689754,
+      3.0789812258166127,
+      3.0772171998635316,
+      3.077096633422069,
+      3.0768553665560536,
+      3.0771814630581784,
+      3.076374116616371,
+      3.076963826122447
     ],
     "adv_g_loss": [
       0.0,
@@ -535,12 +537,12 @@
       0.0
     ],
     "fid": {
-      "25": 263.1458740234375,
-      "50": 598.3736572265625,
-      "75": 598.3736572265625,
-      "100": 598.3736572265625
+      "25": 244.7733612060547,
+      "50": 231.2705535888672,
+      "75": 229.399658203125,
+      "100": 230.25457763671875
     },
-    "train_time_s": 952.6596128940582
+    "train_time_s": 1399.3615255355835
   },
   "n_params": 10608451
 }
\ No newline at end of file
diff --git a/generator/outputs/logs/p3_3_vae_patchgan.json b/generator/outputs/logs/p3_3_vae_patchgan.json
index 68923b2..6c9e285 100644
--- a/generator/outputs/logs/p3_3_vae_patchgan.json
+++ b/generator/outputs/logs/p3_3_vae_patchgan.json
@@ -17,532 +17,534 @@
     "model": "vae",
     "latent_dim": 256,
     "ngf": 64,
+    "free_bits": 0.1,
+    "grad_clip": 1.0,
     "run_name": "p3_3_vae_patchgan",
-    "lr": 0.001,
+    "lr": 0.0005,
     "lr_d": 0.0001,
-    "beta_kl": 0.0001,
+    "beta_kl": 0.005,
     "lambda_perceptual": 0.1,
     "lambda_adversarial": 0.1,
     "ndf_patch": 64
   },
   "history": {
     "recon_loss": [
-      NaN,
-      NaN,
-      0.2550439156082451,
-      NaN,
-      0.3542575327377034,
-      0.32994148498162246,
-      0.327391546824549,
-      NaN,
-      0.32896743778489596,
-      0.32088901420943755,
-      0.35182940304024607,
-      0.3339548002068813,
-      0.3514746379139077,
-      0.355552134796595,
-      0.3535881389537428,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN
+      0.18861990762699363,
+      0.17771862838894892,
+      0.17625267265571487,
+      0.1655209145357466,
+      0.1592490807748758,
+      0.1487464247446539,
+      0.14603353411150286,
+      0.1451508063249863,
+      0.1428027419039072,
+      0.13992124921682045,
+      0.13796657876262808,
+      0.13881459523342612,
+      0.13775221871323565,
+      0.1388729761362585,
+      0.1385588059759038,
+      0.13607313768922263,
+      0.13542115280770847,
+      0.1356422903859972,
+      0.13568231646512818,
+      0.13521548069249362,
+      0.13449831399270612,
+      0.13600172715373018,
+      0.1340917594348773,
+      0.1334343557843031,
+      0.13268615652480695,
+      0.13145199839949098,
+      0.13097500251844907,
+      0.13025199375155136,
+      0.13000157777952331,
+      0.13074792513989994,
+      0.1292976242227432,
+      0.12862894693628335,
+      0.1277259295949569,
+      0.12738761995146927,
+      0.1277741045714953,
+      0.12675375163427785,
+      0.12609347720176745,
+      0.12532664889581183,
+      0.12525444449140474,
+      0.12526406627944392,
+      0.12543218458692232,
+      0.12475586244756849,
+      0.12460088995723133,
+      0.12324439659396298,
+      0.12345891071754134,
+      0.12453210038634446,
+      0.12279523380546488,
+      0.12330379579057041,
+      0.1228398924582025,
+      0.12300096057419084,
+      0.12221984697394392,
+      0.12254253446928456,
+      0.12105376292497684,
+      0.12092442976103888,
+      0.12121323749231987,
+      0.12089209086619891,
+      0.12085758982242173,
+      0.1203383455833054,
+      0.12019133290801293,
+      0.11909146809106709,
+      0.11888215768859427,
+      0.11911316719065365,
+      0.11878056021837088,
+      0.11874932859443192,
+      0.11939801261402093,
+      0.1182701296499397,
+      0.1178444561540571,
+      0.11677229886826797,
+      0.11659586265619494,
+      0.11801627119127502,
+      0.11732396741326039,
+      0.11760511196767673,
+      0.1185783229322515,
+      0.11797440910122843,
+      0.11624205051961108,
+      0.11628382372805196,
+      0.11632549541437218,
+      0.11546542154800178,
+      0.11584110010383475,
+      0.11488010527359115,
+      0.1169404620065903,
+      0.11736274040980726,
+      0.11596904937018696,
+      0.11574434848804759,
+      0.11690343212750223,
+      0.11699441231341444,
+      0.11600632573931645,
+      0.11448002388525723,
+      0.11636082480987932,
+      0.1159489756434137,
+      0.11456818167024697,
+      0.1150473960571819,
+      0.11517152621641628,
+      0.11601082697255999,
+      0.11600699826565564,
+      0.11550768732260434,
+      0.11478957516324316,
+      0.11459140368124358,
+      0.11571422099876098,
+      0.11537429563000672
     ],
     "kl_loss": [
-      13281713.181712223,
-      8879242356.567835,
-      4289.939060308994,
-      13291.21078882462,
-      2964.6168004223423,
-      2881.717480781751,
-      2817.9251735067774,
-      26206.0708080846,
-      3054.7347709460137,
-      3045.038165231036,
-      3176.499509632078,
-      3239.0840866219282,
-      3390.609429253472,
-      3374.538367116553,
-      3556.9082464234443,
-      5232728.427530403,
-      1200351222187.4932,
-      557673391674.235,
-      687785714385.8098,
-      451167040515.6966,
-      4161448879712.457,
-      3506947360774.84,
-      640311792154.062,
-      334276995872.6603,
-      1091409144901.9957,
-      410306176880.5192,
-      20118849921428.06,
-      468057687720.9594,
-      2081128665542.4016,
-      403224538747.06836,
-      98553230942188.75,
-      909009238374.0577,
-      594442830189.4722,
-      6358920601673.252,
-      765801126878.3354,
-      523422198654.6368,
-      749170626764.0299,
-      2510965243074.2544,
-      174694512065.203,
-      1841149037337.5,
-      3074909578060.154,
-      3299731319463.609,
-      1188893842751.9016,
-      1123029761662.4785,
-      439769666356.8568,
-      1481985020811.5044,
-      1227399205729.7693,
-      711698698367.3035,
-      279335895281.97437,
-      720625801842.0897,
-      202221867649.9562,
-      543446495162.29913,
-      175988484712.71368,
-      815170807713.688,
-      762637058414.9017,
-      863531881960.8505,
-      303024411620.3558,
-      909942303709.2073,
-      295614329249.5951,
-      170733932504.22437,
-      2103203516485.6304,
-      892083529658.5983,
-      895749634527.1005,
-      899311239324.7008,
-      1637751780589.5225,
-      1192357094564.0833,
-      484858905592.5705,
-      186463896968.73077,
-      888726863689.1987,
-      15502339561946.314,
-      170425499746.94873,
-      449657248854.87714,
-      330656068541.094,
-      272033123474.23932,
-      693443839763.4946,
-      1247571259921.861,
-      514054457621.8333,
-      1448574911180.951,
-      883422112293.2393,
-      1158489887757.5364,
-      8886462462547.04,
-      274201891523.62607,
-      289924448241.86536,
-      585746708708.7842,
-      1142763735703.111,
-      1130523430761.9956,
-      412582377508.9124,
-      1428507187223.3142,
-      165624474038.62393,
-      333074819594.1261,
-      1738273025208.5898,
-      202257609124.88675,
-      432264649605.9658,
-      1562684211706.267,
-      511279586047.094,
-      1419566063399.6196,
-      510649916352.63464,
-      587311395844.4445,
-      688530254154.5769,
-      63229654584.801285
+      276.7223569103795,
+      120.11881176223103,
+      97.074006153987,
+      84.7367372594328,
+      72.84500447297708,
+      58.20365924509163,
+      49.934763883933044,
+      44.43522434560661,
+      41.382670182448166,
+      39.62511647053254,
+      38.14286702310937,
+      37.18682645324968,
+      36.580295619801575,
+      35.8198753014589,
+      35.183456184517624,
+      34.65776822505853,
+      34.27475022046994,
+      33.969623728695076,
+      33.58452079642532,
+      33.40871410288362,
+      33.494045465420456,
+      33.57894731586815,
+      33.41263916146042,
+      33.5053150307419,
+      33.61977716592642,
+      33.53540447430733,
+      33.71047579936492,
+      33.75520780351427,
+      33.72239826887082,
+      33.84841570079836,
+      33.89454254737267,
+      33.796366430755356,
+      33.99421254997579,
+      33.82595450246436,
+      33.961408395033615,
+      33.92151303576608,
+      34.04281723804963,
+      34.03389532953246,
+      34.118032496199646,
+      34.17981781103672,
+      34.09320658496303,
+      34.11978490128477,
+      34.08164874508849,
+      34.24213214613434,
+      34.148709590618424,
+      34.237437354193794,
+      34.26119823129768,
+      34.16253843062963,
+      34.523601450471794,
+      34.10786265185756,
+      34.4834990134606,
+      34.436983410109825,
+      34.49521413623777,
+      34.37277274661594,
+      34.40642456315521,
+      34.58365509652684,
+      34.46540175543891,
+      34.62582559667082,
+      34.49816356561123,
+      34.425088417835724,
+      34.689441599397576,
+      34.49228691035866,
+      34.47755219386174,
+      34.66372329353268,
+      34.639683275141266,
+      34.53895970695039,
+      34.615915135440666,
+      34.72424408513257,
+      34.75890907874474,
+      34.440071921063286,
+      34.66837335244203,
+      34.842535931839905,
+      34.497212450728455,
+      34.58439721816625,
+      34.64464315593752,
+      34.63696094252106,
+      34.72094595330393,
+      34.87894229399852,
+      34.89129682687613,
+      34.79601346529447,
+      35.0156437311417,
+      34.86639989543165,
+      34.57888742593619,
+      34.73608997540596,
+      34.919344290708885,
+      35.13728781643077,
+      34.88758292564979,
+      35.08412531502226,
+      35.182489802694725,
+      35.11944770812988,
+      35.09966516902304,
+      34.9707060838357,
+      35.145755319513825,
+      35.09497239039494,
+      35.19841175405388,
+      35.11572735126202,
+      35.285146452422836,
+      35.23364441211407,
+      35.14119349585639,
+      35.07916343721569
     ],
     "perc_loss": [
-      NaN,
-      NaN,
-      4.991128759506421,
-      NaN,
-      4.873321874528869,
-      4.652275615268284,
-      4.5657713250217276,
-      NaN,
-      4.545961037660256,
-      4.552515344742017,
-      4.670070458681155,
-      4.55506570216937,
-      4.635318170245896,
-      4.757533762190077,
-      4.679769850184774,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN
+      4.165379504872184,
+      4.306908612577324,
+      4.406882733361334,
+      4.360000091740209,
+      4.347245921945979,
+      4.322648144685305,
+      4.296968904315916,
+      4.275595786734524,
+      4.264194417713035,
+      4.23915512745197,
+      4.253835638873597,
+      4.237967304184905,
+      4.23198981060941,
+      4.223654770953024,
+      4.2143101213324785,
+      4.195945755029336,
+      4.195242772245,
+      4.195372948279748,
+      4.185335707970155,
+      4.173882915423467,
+      4.16755078236262,
+      4.171633275146158,
+      4.159881540343293,
+      4.167673206227457,
+      4.159263728011368,
+      4.149037993871248,
+      4.139473092861665,
+      4.125925103313903,
+      4.114537838177803,
+      4.116090639024718,
+      4.109231275880439,
+      4.110566814740499,
+      4.096888745951857,
+      4.092171801461114,
+      4.0924049615859985,
+      4.071065393268553,
+      4.06966782023764,
+      4.061785459008991,
+      4.060950119271237,
+      4.061661668312856,
+      4.056711556565048,
+      4.0546094958598795,
+      4.048456468133845,
+      4.031783304662786,
+      4.03786053463944,
+      4.048189310946016,
+      4.024937762154473,
+      4.031513743930393,
+      4.041708946737469,
+      4.029254699874128,
+      4.020909377142915,
+      4.022204681848868,
+      4.0144958149673595,
+      4.014826006868965,
+      4.011397586419032,
+      4.009836474544982,
+      4.000827926346379,
+      4.00207864983469,
+      3.9983290128218822,
+      3.9805039707412067,
+      3.985023063981635,
+      3.9873007398385267,
+      3.9836343282308335,
+      3.9778986990961256,
+      3.9819785580675826,
+      3.963852386189322,
+      3.9681489824229836,
+      3.9571395709983306,
+      3.9520151141362314,
+      3.9639965010504437,
+      3.962235056436979,
+      3.9679050766504727,
+      3.970652671960684,
+      3.9660502493890943,
+      3.9473167010861583,
+      3.9518334758587375,
+      3.9450023963920073,
+      3.939890970531692,
+      3.9405837446196466,
+      3.9431381770688243,
+      3.9528527356620526,
+      3.9480060828037753,
+      3.933930030745319,
+      3.934511334977598,
+      3.944071548616784,
+      3.932478380508912,
+      3.9239180887866225,
+      3.922858091501089,
+      3.944245843296377,
+      3.929251870028993,
+      3.9123981141636515,
+      3.9128795622760415,
+      3.9116759814767756,
+      3.9187084161318264,
+      3.9145963553689485,
+      3.9085161375184345,
+      3.915584529057527,
+      3.912150815511361,
+      3.9175711910948796,
+      3.918475313573821
     ],
     "adv_g_loss": [
-      NaN,
-      NaN,
-      2.686214610042735,
-      NaN,
-      3.1861853966346154,
-      3.3312967414529915,
-      3.4314809945913463,
-      NaN,
-      3.9346955128205128,
-      4.042058961004273,
-      4.416372445913462,
-      4.483027009882479,
-      4.66728014823718,
-      4.537232488648504,
-      4.415139222756411,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN
+      1.8026715818728902,
+      2.23227376344367,
+      2.2183715565464435,
+      2.1278742490033817,
+      2.0971770584583282,
+      2.0484086269926696,
+      2.03275868207471,
+      2.116789224024257,
+      2.090583788151415,
+      1.9892138968675563,
+      1.9692096719677505,
+      1.9062933527713275,
+      1.894742165073259,
+      1.8143773762326145,
+      1.8193041879970293,
+      1.81429538425281,
+      1.8210619932486334,
+      1.7941538303191018,
+      1.8341339080749701,
+      1.8242450552459202,
+      1.8905006748600266,
+      1.890135900216957,
+      1.8945831565313742,
+      1.9415043775215108,
+      1.9855943465700898,
+      2.0438742065447997,
+      2.024762265774238,
+      2.0435926514144382,
+      2.0605728566869455,
+      2.137863679956167,
+      2.0963172315755205,
+      2.141162985625366,
+      2.177981933658449,
+      2.17414838136134,
+      2.2223903614486384,
+      2.2023396426413813,
+      2.275088946486258,
+      2.2592285347736287,
+      2.303790853637406,
+      2.3635131845839767,
+      2.34030539874071,
+      2.3638583834354696,
+      2.4543145986066923,
+      2.3815444054671078,
+      2.4706798928956957,
+      2.4403553184790487,
+      2.4697066351739516,
+      2.537486906043994,
+      2.5634906901380954,
+      2.5490670253833136,
+      2.5615056049174223,
+      2.6466195793488088,
+      2.604182060967144,
+      2.657348400889299,
+      2.6449959174944806,
+      2.656565827309576,
+      2.678831669764641,
+      2.688901748794776,
+      2.68519163182658,
+      2.709996796380251,
+      2.7448226344636364,
+      2.768862698832129,
+      2.7961852730084686,
+      2.780011566530945,
+      2.7335693817108107,
+      2.7967236773071127,
+      2.7932374021436415,
+      2.8125013566424704,
+      2.8752479591430764,
+      2.884957304622373,
+      2.9013289062895327,
+      2.8676841870332375,
+      2.9157643700257325,
+      2.9385765872449956,
+      2.971192495435731,
+      2.9460317586731706,
+      3.006316975905345,
+      3.0268147679475637,
+      3.0260748012453065,
+      3.098839063165534,
+      3.1647758919459124,
+      3.106013861986307,
+      3.110682492072766,
+      3.1142931325822816,
+      3.1455837691951003,
+      3.08899037552695,
+      3.1192317396147637,
+      3.1001298911551123,
+      3.083799856340783,
+      3.1157319148381553,
+      3.095861378388527,
+      3.1165227436611795,
+      3.086983766311254,
+      3.0861313699657082,
+      3.0921488047665,
+      3.0940884631923122,
+      3.0856645693126907,
+      3.0799090566798153,
+      3.086693912999243,
+      3.093890537054111
     ],
     "adv_d_loss": [
-      NaN,
-      NaN,
-      0.11442611563918936,
-      NaN,
-      0.05586824661646134,
-      0.04843576137836163,
-      0.05373259079761994,
-      NaN,
-      0.015966382291581895,
-      0.01352301099870959,
-      0.016935703591403797,
-      0.015551309682365157,
-      0.01746162848594861,
-      0.023566564433595054,
-      0.011101227285515549,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN,
-      NaN
+      0.3846818933136061,
+      0.19303795548840466,
+      0.22524197362204137,
+      0.23716845485962865,
+      0.25527555404756314,
+      0.2399256144069199,
+      0.25104449234075016,
+      0.2408019018988324,
+      0.2598570621548555,
+      0.2685285050612994,
+      0.2780756782899555,
+      0.2814109750633311,
+      0.30625250518449354,
+      0.3203703035146762,
+      0.30513450952294546,
+      0.30278418525161904,
+      0.3132180890593773,
+      0.31392792752410614,
+      0.2981438948653447,
+      0.3000434341434485,
+      0.2830737477693802,
+      0.2774529387521693,
+      0.27726965177899754,
+      0.2697514830969083,
+      0.2631684033056864,
+      0.2501653985430797,
+      0.2529780987777516,
+      0.24266270441440946,
+      0.25031912855358207,
+      0.22288694602047276,
+      0.2343698614077945,
+      0.2288323061174562,
+      0.22137120039735594,
+      0.22300533291239005,
+      0.20846376789367607,
+      0.21844437852119789,
+      0.19620536157909113,
+      0.20161379222622794,
+      0.19350602869422007,
+      0.1940044454561594,
+      0.19748605735217914,
+      0.18612161299420726,
+      0.18449116273759267,
+      0.1786681996164923,
+      0.17217153660220724,
+      0.1815700484917332,
+      0.17118102850981504,
+      0.1629717636646496,
+      0.1747021592166434,
+      0.1594025977433492,
+      0.1572152761519592,
+      0.1572253421649464,
+      0.14647067149575704,
+      0.14247114893494764,
+      0.14077772017027068,
+      0.1379101731392563,
+      0.13861483306042913,
+      0.13286431546872243,
+      0.1263120919545619,
+      0.12339711796619698,
+      0.12401506412997206,
+      0.11199609567530644,
+      0.11085018765929545,
+      0.11338936790632896,
+      0.12032712383084318,
+      0.10354128184640764,
+      0.1036931567817417,
+      0.10232575738436392,
+      0.08907810818308438,
+      0.08665908210409376,
+      0.09036243790521836,
+      0.09317310632835342,
+      0.08505074158868092,
+      0.08004704910593155,
+      0.07433714670264441,
+      0.0742792083332554,
+      0.07248415664618468,
+      0.06889059219676523,
+      0.06714722003915116,
+      0.05902106333205588,
+      0.05658946855940944,
+      0.059529210504494674,
+      0.05614422008785236,
+      0.05332565255678044,
+      0.0521102087576197,
+      0.052153651256305285,
+      0.04740947977098453,
+      0.048982632026060395,
+      0.05126904483494532,
+      0.046512605652260855,
+      0.04599587608757628,
+      0.04359768634327711,
+      0.0426614616647299,
+      0.04159874533120002,
+      0.03934315871845326,
+      0.03652780742233253,
+      0.036582552890579864,
+      0.03528311501384482,
+      0.034168972188017815,
+      0.03123981755378091
     ],
     "fid": {
-      "25": 300.18701171875,
-      "50": 328.4952392578125,
-      "75": 334.2494812011719,
-      "100": 334.40887451171875
+      "25": 217.87948608398438,
+      "50": 231.8101043701172,
+      "75": 233.5815887451172,
+      "100": 234.91502380371094
     },
-    "train_time_s": 1450.524620771408
+    "train_time_s": 2140.041033744812
   },
   "n_params": 10608451
 }
\ No newline at end of file
diff --git a/generator/outputs/pipeline/20260501T233340.557082+0000.json b/generator/outputs/pipeline/20260501T233340.557082+0000.json
new file mode 100644
index 0000000..5536a86
--- /dev/null
+++ b/generator/outputs/pipeline/20260501T233340.557082+0000.json
@@ -0,0 +1,14 @@
+{
+  "created_at": "2026-05-01T23:33:40.557082+00:00",
+  "config_paths": [
+    "generator/configs/phase3/p3_1_vae.json",
+    "generator/configs/phase3/p3_2_vae_perceptual.json",
+    "generator/configs/phase3/p3_3_vae_patchgan.json"
+  ],
+  "instance_id": 35985830,
+  "offer_id": 30940273,
+  "ssh_host": "ssh4.vast.ai",
+  "ssh_port": 25830,
+  "status": "completed",
+  "remote_workspace": "/workspace/DRL_PROJ"
+}
\ No newline at end of file
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0010_recon.png b/generator/outputs/samples/p3_1_vae/epoch_0010_recon.png
index fa822e4..f8cd8f8 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0010_recon.png and b/generator/outputs/samples/p3_1_vae/epoch_0010_recon.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0020_recon.png b/generator/outputs/samples/p3_1_vae/epoch_0020_recon.png
index 8628b1f..c3b3b1b 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0020_recon.png and b/generator/outputs/samples/p3_1_vae/epoch_0020_recon.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0040.png b/generator/outputs/samples/p3_1_vae/epoch_0040.png
index 82fe4e6..b7540a0 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0040.png and b/generator/outputs/samples/p3_1_vae/epoch_0040.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0050.png b/generator/outputs/samples/p3_1_vae/epoch_0050.png
index 1168734..3a178df 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0050.png and b/generator/outputs/samples/p3_1_vae/epoch_0050.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0060.png b/generator/outputs/samples/p3_1_vae/epoch_0060.png
index a3a6643..f504b91 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0060.png and b/generator/outputs/samples/p3_1_vae/epoch_0060.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0070.png b/generator/outputs/samples/p3_1_vae/epoch_0070.png
index 0632862..267ad19 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0070.png and b/generator/outputs/samples/p3_1_vae/epoch_0070.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0080.png b/generator/outputs/samples/p3_1_vae/epoch_0080.png
index df012e2..4eb65e9 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0080.png and b/generator/outputs/samples/p3_1_vae/epoch_0080.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0090.png b/generator/outputs/samples/p3_1_vae/epoch_0090.png
index 87a7a72..2c146f5 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0090.png and b/generator/outputs/samples/p3_1_vae/epoch_0090.png differ
diff --git a/generator/outputs/samples/p3_1_vae/epoch_0100.png b/generator/outputs/samples/p3_1_vae/epoch_0100.png
index d51e462..34d02aa 100644
Binary files a/generator/outputs/samples/p3_1_vae/epoch_0100.png and b/generator/outputs/samples/p3_1_vae/epoch_0100.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020.png
index fa9d579..6ab4ad8 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020_recon.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020_recon.png
index bac1a12..e07c663 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020_recon.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0020_recon.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0040.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0040.png
index e394c82..508b2cc 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0040.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0040.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0050.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0050.png
index e394c82..c562f90 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0050.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0050.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0060.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0060.png
index e394c82..0ed1c29 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0060.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0060.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0070.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0070.png
index e394c82..929864d 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0070.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0070.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0080.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0080.png
index e394c82..d4a0195 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0080.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0080.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0090.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0090.png
index e394c82..ec26901 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0090.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0090.png differ
diff --git a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0100.png b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0100.png
index e394c82..93b9649 100644
Binary files a/generator/outputs/samples/p3_2_vae_perceptual/epoch_0100.png and b/generator/outputs/samples/p3_2_vae_perceptual/epoch_0100.png differ
diff --git a/generator/outputs/samples/p3_3_vae_patchgan/epoch_0020.png b/generator/outputs/samples/p3_3_vae_patchgan/epoch_0020.png
index 5961881..7d6787f 100644
Binary files a/generator/outputs/samples/p3_3_vae_patchgan/epoch_0020.png and b/generator/outputs/samples/p3_3_vae_patchgan/epoch_0020.png differ