diff --git a/generator/configs/phase0/_base_phase0.json b/generator/configs/phase0/_base_phase0.json
new file mode 100644
index 0000000..62649ec
--- /dev/null
+++ b/generator/configs/phase0/_base_phase0.json
@@ -0,0 +1,7 @@
+{
+  "data_dir": "data",
+  "sources": ["wiki"],
+  "augment": false,
+  "image_size": 64,
+  "batch_size": 64
+}
diff --git a/generator/configs/phase0/p0_ddpm.json b/generator/configs/phase0/p0_ddpm.json
new file mode 100644
index 0000000..1bea241
--- /dev/null
+++ b/generator/configs/phase0/p0_ddpm.json
@@ -0,0 +1,15 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_ddpm",
+  "model": "ddpm",
+  "base_ch": 128,
+  "T": 1000,
+  "beta_start": 0.0001,
+  "beta_end": 0.02,
+  "dropout": 0.1,
+  "epochs": 200,
+  "lr": 0.0001,
+  "subsample": 1.0,
+  "sample_interval": 10,
+  "sample_steps": 100
+}
diff --git a/generator/configs/phase0/p0_ddpm_small.json b/generator/configs/phase0/p0_ddpm_small.json
new file mode 100644
index 0000000..af6f33c
--- /dev/null
+++ b/generator/configs/phase0/p0_ddpm_small.json
@@ -0,0 +1,15 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_ddpm_small",
+  "model": "ddpm",
+  "base_ch": 64,
+  "T": 1000,
+  "beta_start": 0.0001,
+  "beta_end": 0.02,
+  "dropout": 0.1,
+  "epochs": 100,
+  "lr": 0.0002,
+  "subsample": 0.2,
+  "sample_interval": 10,
+  "sample_steps": 100
+}
diff --git a/generator/configs/phase0/p0_vae.json b/generator/configs/phase0/p0_vae.json
new file mode 100644
index 0000000..cf5c396
--- /dev/null
+++ b/generator/configs/phase0/p0_vae.json
@@ -0,0 +1,12 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_vae",
+  "model": "vae",
+  "latent_dim": 256,
+  "epochs": 100,
+  "lr": 0.0005,
+  "beta": 1.0,
+  "perceptual_weight": 0.1,
+  "subsample": 1.0,
+  "sample_interval": 5
+}
diff --git a/generator/configs/phase0/p0_wgan.json b/generator/configs/phase0/p0_wgan.json
new file mode 100644
index 0000000..6fec98b
--- /dev/null
+++ b/generator/configs/phase0/p0_wgan.json
@@ -0,0 +1,17 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_wgan",
+  "model": "wgan",
+  "latent_dim": 128,
+  "ngf": 64,
+  "ndf": 64,
+  "epochs": 200,
+  "lr_g": 0.0001,
+  "lr_d": 0.0001,
+  "beta1": 0.0,
+  "beta2": 0.9,
+  "n_critic": 5,
+  "gp_lambda": 10,
+  "subsample": 1.0,
+  "sample_interval": 10
+}
diff --git a/generator/outputs/logs/p0_ddpm.json b/generator/outputs/logs/p0_ddpm.json
new file mode 100644
index 0000000..6e114c1
--- /dev/null
+++ b/generator/outputs/logs/p0_ddpm.json
@@ -0,0 +1,227 @@
+{
+  "run_name": "p0_ddpm",
+  "config": {
+    "run_name": "p0_ddpm",
+    "model": "ddpm",
+    "image_size": 64,
+    "base_ch": 128,
+    "T": 1000,
+    "beta_start": 0.0001,
+    "beta_end": 0.02,
+    "dropout": 0.1,
+    "epochs": 200,
+    "batch_size": 64,
+    "lr": 0.0001,
+    "data_dir": "data",
+    "sources": [
+      "wiki"
+    ],
+    "subsample": 1.0,
+    "sample_interval": 10,
+    "sample_steps": 100
+  },
+  "history": {
+    "loss": [
+      0.1083670402598432,
+      0.04629855174332475,
+      0.03927792408543392,
+      0.03648682562315948,
+      0.034014404620020054,
+      0.03234251832159666,
+      0.030846774018902935,
+      0.030006713196078044,
+      0.03014388835280497,
+      0.02850124903787405,
+      0.028395157695437472,
+      0.027885630450203504,
+      0.027339542316447977,
+      0.027508525955928568,
+      0.02726412267408246,
+      0.026637520621188432,
+      0.027042127676053435,
+      0.02702308123589008,
+      0.027003993411572315,
+      0.025956713819366872,
+      0.026461017499160435,
+      0.026138577759504702,
+      0.02589148448772219,
+      0.025798799915828258,
+      0.025888724787893076,
+      0.026004614308476448,
+      0.02490535442932294,
+      0.025254002792171687,
+      0.025024926007335257,
+      0.02529874118243973,
+      0.02479060086158988,
+      0.025004531465805113,
+      0.02481336768469813,
+      0.024719654377470285,
+      0.02482008694822335,
+      0.024336259418095533,
+      0.025327550165721368,
+      0.024988760037395436,
+      0.024449641408366907,
+      0.024988751213711042,
+      0.024216544261186294,
+      0.024626739183838807,
+      0.02405230538187246,
+      0.02450195714457231,
+      0.024484938772745494,
+      0.024367860245765146,
+      0.024223141513494216,
+      0.024255985758291222,
+      0.024372014694680005,
+      0.02413457283614856,
+      0.023963905616193756,
+      0.02441367820207762,
+      0.02433791215945449,
+      0.02456985757105116,
+      0.02404512893050336,
+      0.024265815064104065,
+      0.023772775438319668,
+      0.024229186833796337,
+      0.023757771352051288,
+      0.024260703422543075,
+      0.023852092454719365,
+      0.02406276131852761,
+      0.023678840800689962,
+      0.023432825142756488,
+      0.02385735059252534,
+      0.02377281899755009,
+      0.02348707383299549,
+      0.02369726858396306,
+      0.023631208038960513,
+      0.02311639924151584,
+      0.02383487479777163,
+      0.023870027238009576,
+      0.023441708230803538,
+      0.024035497753618237,
+      0.02300195097246677,
+      0.023707804898930412,
+      0.023455506710049052,
+      0.023611091168867983,
+      0.024028617561333135,
+      0.0232626189215061,
+      0.023289684098787032,
+      0.02343361178198113,
+      0.023422238150309038,
+      0.0232180315348455,
+      0.023305013184793867,
+      0.02392511722894433,
+      0.02336770045555109,
+      0.02318744173942086,
+      0.02320095007586428,
+      0.02295187271685682,
+      0.0233883468581475,
+      0.02344029073794492,
+      0.02342299423697922,
+      0.023732792942299165,
+      0.023115517183517415,
+      0.023331191981784426,
+      0.02352578725076766,
+      0.0230310578396719,
+      0.02284456534573856,
+      0.02327608281913667,
+      0.02291412464876333,
+      0.02318612803728917,
+      0.023585573692495625,
+      0.022689972158409972,
+      0.023414890062955454,
+      0.02354874097320259,
+      0.023317397044748705,
+      0.022987709951817863,
+      0.02321497991314938,
+      0.023314894041699223,
+      0.02330707421956154,
+      0.023357170003139947,
+      0.023417106554962885,
+      0.023055344092675574,
+      0.022828684811297264,
+      0.022907353412264433,
+      0.022932124652493842,
+      0.02278750507821703,
+      0.022915415256881178,
+      0.023283357745928004,
+      0.022942685305228472,
+      0.02283522265796096,
+      0.02293384290085389,
+      0.023348970316812143,
+      0.023261155360020123,
+      0.022875425483410556,
+      0.023388160398612037,
+      0.02314243713616688,
+      0.023365848144500427,
+      0.022828353748012047,
+      0.022968567352598675,
+      0.023286700292498384,
+      0.022814179418417506,
+      0.02317798297263236,
+      0.022502902497211073,
+      0.022226932952100903,
+      0.022558437107114963,
+      0.023172160780702073,
+      0.0226754893321926,
+      0.02281457268529468,
+      0.023307094891738687,
+      0.022783862096536107,
+      0.022608044509513255,
+      0.02264707699481748,
+      0.0228006450427521,
+      0.022862335525723733,
+      0.023164480990276504,
+      0.022886841307179287,
+      0.02281724872927253,
+      0.0224764158244794,
+      0.02299271177691527,
+      0.022401189933029506,
+      0.022301576679381423,
+      0.023051011495483227,
+      0.022674021506124836,
+      0.023254253867711138,
+      0.022600062249196518,
+      0.022994125096732352,
+      0.023044346577018246,
+      0.022450523532560874,
+      0.022708959248649258,
+      0.022672621071593374,
+      0.0226655634951929,
+      0.02268093158530756,
+      0.0231474475023679,
+      0.023222818055277705,
+      0.02250314637238526,
+      0.022926766914912522,
+      0.02281918561157699,
+      0.022894049795646954,
+      0.022437431771929067,
+      0.022276719749515127,
+      0.023031598454516414,
+      0.022932796695261683,
+      0.02267399260129493,
+      0.022674424667309366,
+      0.022683103929042943,
+      0.0230639888268187,
+      0.022548409081342764,
+      0.022087385764536567,
+      0.021826119812874075,
+      0.022922894372166987,
+      0.022938080926019795,
+      0.02203184187523702,
+      0.022286941173971973,
+      0.0225165595069655,
+      0.022823427537153673,
+      0.022163119194949538,
+      0.02246519502522981,
+      0.022458927598423682,
+      0.02173858560804819,
+      0.022764349513504114,
+      0.022648208464185398,
+      0.021945582256994695,
+      0.022437607624735206,
+      0.022418903443628013,
+      0.02227352012314985,
+      0.022173316878640752,
+      0.022317954088346317,
+      0.02246333107463697
+    ]
+  }
+}
\ No newline at end of file
diff --git a/generator/outputs/logs/p0_ddpm_small.json b/generator/outputs/logs/p0_ddpm_small.json
new file mode 100644
index 0000000..2b1ea73
--- /dev/null
+++ b/generator/outputs/logs/p0_ddpm_small.json
@@ -0,0 +1,127 @@
+{
+  "run_name": "p0_ddpm_small",
+  "config": {
+    "run_name": "p0_ddpm_small",
+    "model": "ddpm",
+    "image_size": 64,
+    "base_ch": 64,
+    "T": 1000,
+    "beta_start": 0.0001,
+    "beta_end": 0.02,
+    "dropout": 0.1,
+    "epochs": 100,
+    "batch_size": 64,
+    "lr": 0.0002,
+    "data_dir": "data",
+    "sources": [
+      "wiki"
+    ],
+    "subsample": 0.2,
+    "sample_interval": 10,
+    "sample_steps": 100
+  },
+  "history": {
+    "loss": [
+      0.2678032471768318,
+      0.09589410829608158,
+      0.07335762967986445,
+      0.0649665344546559,
+      0.05737644754430299,
+      0.052905762227632665,
+      0.04984104056512156,
+      0.0454951891655563,
+      0.04398588173251639,
+      0.044020078775863496,
+      0.04207056302136632,
+      0.04043278923278214,
+      0.039955350660508676,
+      0.03724756500413341,
+      0.03665647719816495,
+      0.038215308159749994,
+      0.036488514173255174,
+      0.035779101073101005,
+      0.035255871212450406,
+      0.036489000984577726,
+      0.03383178988932281,
+      0.03419857377046218,
+      0.033235303496801726,
+      0.033483810442429714,
+      0.031014470643894648,
+      0.03326657182106408,
+      0.031699600939949356,
+      0.03173744650457495,
+      0.0322553291315994,
+      0.03234702623098768,
+      0.03161054465078538,
+      0.030852873198768144,
+      0.03200384822263513,
+      0.031742659808006335,
+      0.030217854826841303,
+      0.031679999583991626,
+      0.029489911351633327,
+      0.028711399854591455,
+      0.029677809045840334,
+      0.030612801151570453,
+      0.029807658155038153,
+      0.030668188747699544,
+      0.02895499796917041,
+      0.028708126405955003,
+      0.029014079511085505,
+      0.0281836397905824,
+      0.029067626433266748,
+      0.028298665388857804,
+      0.027657570247288033,
+      0.027329172037782207,
+      0.027731550667154534,
+      0.028330990283559728,
+      0.029598585159708094,
+      0.028600716683012183,
+      0.028036352287056625,
+      0.0285629206487248,
+      0.02861545857803155,
+      0.027790598511215178,
+      0.028236197407848093,
+      0.026964312299124656,
+      0.027373770281912818,
+      0.027388963888409317,
+      0.029223351749361203,
+      0.02739422419859517,
+      0.02748733340331944,
+      0.027431275315021954,
+      0.02745134727929228,
+      0.028024198807856088,
+      0.02867266363514367,
+      0.026144217199055096,
+      0.026972133546106277,
+      0.027528228488580515,
+      0.027562430388824916,
+      0.027393175549404597,
+      0.0265277804286089,
+      0.025383217280269953,
+      0.027007741653310356,
+      0.025879052337459338,
+      0.026285308823790602,
+      0.026477108579329264,
+      0.028581748666462077,
+      0.026456236048170958,
+      0.02589529524407079,
+      0.026571117862257906,
+      0.026034350496947124,
+      0.0268421945592729,
+      0.027832103742947503,
+      0.02694581709401582,
+      0.027404129525185915,
+      0.026766230872962423,
+      0.02709349239825882,
+      0.027885750516928653,
+      0.02651280081600592,
+      0.026352581538019643,
+      0.026670256619571998,
+      0.0255209504656734,
+      0.026436442599421548,
+      0.02504222024913116,
+      0.025400650226861558,
+      0.027037389025652923
+    ]
+  }
+}
\ No newline at end of file
diff --git a/generator/outputs/logs/p0_vae.json b/generator/outputs/logs/p0_vae.json
new file mode 100644
index 0000000..52c3d39
--- /dev/null
+++ b/generator/outputs/logs/p0_vae.json
@@ -0,0 +1,430 @@
+{
+  "run_name": "p0_vae",
+  "config": {
+    "run_name": "p0_vae",
+    "model": "vae",
+    "image_size": 64,
+    "latent_dim": 256,
+    "epochs": 100,
+    "batch_size": 64,
+    "lr": 0.0005,
+    "beta": 1.0,
+    "perceptual_weight": 0.1,
+    "data_dir": "data",
+    "sources": [
+      "wiki"
+    ],
+    "subsample": 1.0,
+    "sample_interval": 5
+  },
+  "history": {
+    "loss": [
+      4.250324185587402,
+      4.134648546194419,
+      4.094182982913449,
+      4.063824407565288,
+      4.0326321787304344,
+      4.003465635654254,
+      3.974764764818371,
+      3.9508974959707666,
+      3.9271021245891213,
+      3.906269828478495,
+      3.8868031960267286,
+      3.8682583764068084,
+      3.8515498037012215,
+      3.8359983634744954,
+      3.8221397761605744,
+      3.809864979523879,
+      3.798683610227373,
+      3.78772361512877,
+      3.7773428635719495,
+      3.768211925131643,
+      3.759337167454581,
+      3.7517892585860357,
+      3.743788587231921,
+      3.735912532887907,
+      3.7286843754287458,
+      3.7223885329360638,
+      3.715175391262413,
+      3.7087125569327264,
+      3.7038757984454813,
+      3.6965429665695906,
+      3.691004909001864,
+      3.6857250995106168,
+      3.6793667913502097,
+      3.6750445773458886,
+      3.6694462722183294,
+      3.664994449696989,
+      3.659889499346415,
+      3.6559000580738754,
+      3.651372315537216,
+      3.6471689791760893,
+      3.6424113588455396,
+      3.637727607519199,
+      3.6339953450056224,
+      3.6298532022370233,
+      3.6273992504829016,
+      3.623114150304061,
+      3.619541567614955,
+      3.6168203715585237,
+      3.6138987872335644,
+      3.608925356824174,
+      3.6048866056988382,
+      3.6027798637365684,
+      3.600335784447499,
+      3.595807208464696,
+      3.5932340413077264,
+      3.5909022021497417,
+      3.588857959987771,
+      3.586156122195415,
+      3.582110808445857,
+      3.5804258804036,
+      3.578975318843483,
+      3.5756211387805448,
+      3.5734891743741484,
+      3.5706143093924236,
+      3.5675539843037596,
+      3.565658889265142,
+      3.56276819950495,
+      3.5605046489299874,
+      3.559286260197305,
+      3.5578912598455052,
+      3.5546631288324666,
+      3.553219128368247,
+      3.5495453558416448,
+      3.5479163568244023,
+      3.546648928752312,
+      3.5447635762711878,
+      3.5432754383127913,
+      3.540637479378627,
+      3.5402203767727585,
+      3.537344281999474,
+      3.5353173695058904,
+      3.5337163637845945,
+      3.5310573751090937,
+      3.5295781304693628,
+      3.5287446262490034,
+      3.5265913447763166,
+      3.5268727235305004,
+      3.5250504770849505,
+      3.5212181495805073,
+      3.521022683534867,
+      3.519402818292634,
+      3.5169585255476146,
+      3.5171003142992654,
+      3.5146776891162252,
+      3.5131581868880835,
+      3.5116431331023192,
+      3.5113982497117457,
+      3.508979608360519,
+      3.50840662789141,
+      3.5074753414871345
+    ],
+    "recon_loss": [
+      0.08314752892360218,
+      0.07855224641215088,
+      0.07766621149121186,
+      0.07724331894045712,
+      0.07686529886455108,
+      0.07642129698816018,
+      0.07581346730391185,
+      0.07500511466756336,
+      0.07364231738078798,
+      0.07157875317284185,
+      0.06891512811088409,
+      0.0661920348707682,
+      0.06391830628530847,
+      0.06201417692419556,
+      0.06046784213051582,
+      0.05918571319526587,
+      0.0580066671817858,
+      0.057031127313772835,
+      0.05595446255408291,
+      0.05514845847446694,
+      0.05429254599616059,
+      0.05357683884600798,
+      0.05280969729726641,
+      0.05211603768870362,
+      0.051566887559353285,
+      0.05080651687696958,
+      0.05030167758719534,
+      0.04969718606553526,
+      0.04914569419323125,
+      0.048616886688157536,
+      0.048137954961604036,
+      0.047627850459554255,
+      0.047129555661072076,
+      0.046656576622054614,
+      0.046228237091921844,
+      0.04574223620514584,
+      0.04545198885612508,
+      0.045023377117119796,
+      0.044643198219565756,
+      0.04434523899426572,
+      0.04387786846735284,
+      0.04364254737161418,
+      0.04329113803931281,
+      0.0428919303111541,
+      0.04272017235525398,
+      0.04239159133922086,
+      0.04209632497344516,
+      0.04185059678573639,
+      0.04161195915478926,
+      0.04135766227403258,
+      0.04115677575588736,
+      0.04086931493992989,
+      0.04076992701261471,
+      0.04041119487398965,
+      0.040200370650451914,
+      0.0400576183381371,
+      0.039909450456691094,
+      0.03970725073391556,
+      0.03947478756467756,
+      0.03935474883287381,
+      0.03919493802624126,
+      0.03901928393409038,
+      0.03884959949228244,
+      0.038668815770910844,
+      0.03841047747554178,
+      0.038346435938380725,
+      0.038163057369235746,
+      0.038028685718329035,
+      0.03786847321714601,
+      0.037713001227468,
+      0.037593389065283485,
+      0.03751569933807238,
+      0.03734617850655674,
+      0.03719269978598906,
+      0.03706175414447346,
+      0.0369408268115332,
+      0.03681376709356012,
+      0.036733934202064306,
+      0.03657762347282762,
+      0.036514967544657044,
+      0.03640143988001296,
+      0.036243789049231596,
+      0.03615107964803903,
+      0.035999311524260245,
+      0.035903908702361785,
+      0.03579426486777444,
+      0.03576036798170744,
+      0.03566311308556897,
+      0.03552897459962684,
+      0.03539964183368999,
+      0.03534168970540293,
+      0.03520329746728142,
+      0.035151520210644625,
+      0.035101048847357944,
+      0.0350069641852035,
+      0.03486888159989801,
+      0.034861071043225944,
+      0.034744354585806526,
+      0.03465007174298422,
+      0.03454487510901104
+    ],
+    "kl_loss": [
+      0.0159953803752218,
+      0.044202877240430594,
+      0.06116550275657931,
+      0.07635654457327393,
+      0.09037123305293229,
+      0.10460591577311866,
+      0.11865333892787114,
+      0.13245739485336167,
+      0.14517442560476115,
+      0.1569571604904456,
+      0.16872903736483338,
+      0.18024130191049006,
+      0.19072035840179166,
+      0.20048105551136863,
+      0.20899017826996297,
+      0.21741188874738848,
+      0.2247571811143659,
+      0.23275449531327966,
+      0.23966090094584686,
+      0.24611625218620667,
+      0.25281740190126956,
+      0.25888903620533454,
+      0.2645308631989691,
+      0.27056845175659555,
+      0.27532415525016624,
+      0.28112532414941704,
+      0.2858431101736859,
+      0.29094230332690424,
+      0.2955150889535236,
+      0.3001613756402945,
+      0.3049032572242949,
+      0.3095898434010326,
+      0.31319689776143456,
+      0.3179963257832405,
+      0.322185754776001,
+      0.3264285368032945,
+      0.3299771794396588,
+      0.3337477809216222,
+      0.3377718361906516,
+      0.34097307831303686,
+      0.3449199155737192,
+      0.34790686887298894,
+      0.3513057490444591,
+      0.35493463105880296,
+      0.35747312830808836,
+      0.3610215593352277,
+      0.3638480869878052,
+      0.366655437864809,
+      0.36968781888230234,
+      0.37230672591771835,
+      0.3748398758152611,
+      0.37769231722395646,
+      0.38097035623768455,
+      0.383052895593847,
+      0.38532960580454934,
+      0.38812654013307685,
+      0.3900898962448805,
+      0.3934439178715404,
+      0.3946356102824211,
+      0.39688008622481274,
+      0.3985537563761075,
+      0.4018082413663212,
+      0.4037678490082423,
+      0.4062399629216928,
+      0.4081735656174839,
+      0.4093499289363877,
+      0.4120887863075631,
+      0.4140137403439253,
+      0.41596079267497754,
+      0.418453937348647,
+      0.41991064423679286,
+      0.42121546307944846,
+      0.4233135558728479,
+      0.4248411596203462,
+      0.42699897142811716,
+      0.4287081425770735,
+      0.4301795422648772,
+      0.4317423408994308,
+      0.43337412885366344,
+      0.43470359224284816,
+      0.43652185428346324,
+      0.43851066215170753,
+      0.4411619116480534,
+      0.4418926327529117,
+      0.4436655201845699,
+      0.4451010020242797,
+      0.4460634829906317,
+      0.44681531662105495,
+      0.44934566471821225,
+      0.4504435484608014,
+      0.4513984547338934,
+      0.45284434554413855,
+      0.4547643432887191,
+      0.45665359598958594,
+      0.45710637777024865,
+      0.4585863476635044,
+      0.45928521460702276,
+      0.46097695604603517,
+      0.4617650115973929,
+      0.46387570485090596
+    ],
+    "perc_loss": [
+      41.51181179437882,
+      40.11893345759465,
+      39.553511986365685,
+      39.10224482544467,
+      38.653955842694664,
+      38.22438364762526,
+      37.80297894763132,
+      37.43434936979897,
+      37.08285335801605,
+      36.77733863928379,
+      36.49158988039718,
+      36.2182500138242,
+      35.96911093312451,
+      35.73503083449144,
+      35.52681698758378,
+      35.3326733418,
+      35.15919718375573,
+      34.97937949498495,
+      34.817274513407654,
+      34.669471732571594,
+      34.52227175948966,
+      34.393233368539406,
+      34.26447987760234,
+      34.13227987289429,
+      34.01793288369464,
+      33.904566475468826,
+      33.79030554111187,
+      33.68073015131502,
+      33.592149718194946,
+      33.477646607619064,
+      33.37963648739024,
+      33.28507361452804,
+      33.19040275638939,
+      33.10391625379905,
+      33.01032242408166,
+      32.928236337808464,
+      32.844602796766495,
+      32.7712886720641,
+      32.689572305760834,
+      32.61850610553709,
+      32.536135265969826,
+      32.4617814080328,
+      32.3939840650966,
+      32.320265892224434,
+      32.27205904528626,
+      32.19700959605029,
+      32.13597111009125,
+      32.08314285115299,
+      32.02598962213239,
+      31.95260914370545,
+      31.88889909401918,
+      31.84218183338133,
+      31.785954487629425,
+      31.723430678375767,
+      31.677040259043377,
+      31.62718008726071,
+      31.58858571908413,
+      31.530049095805893,
+      31.480003752260128,
+      31.441910014193283,
+      31.412265757210236,
+      31.347935664348114,
+      31.308716773986816,
+      31.25705478945349,
+      31.209698994954426,
+      31.179624773498276,
+      31.12516305793045,
+      31.084621735108204,
+      31.05456941963261,
+      31.017242786211845,
+      30.97159047412057,
+      30.944879160987007,
+      30.88885566108247,
+      30.858824599502434,
+      30.825881578983406,
+      30.791145471426155,
+      30.762820814409828,
+      30.721611569070408,
+      30.702685759617733,
+      30.661256769783478,
+      30.62394029666216,
+      30.589618682861328,
+      30.53744347890218,
+      30.516861418373566,
+      30.49175152819381,
+      30.456960311302772,
+      30.450488273914043,
+      30.425719966236343,
+      30.363434677450066,
+      30.351794434408855,
+      30.326626260056454,
+      30.289108406784187,
+      30.271844048785347,
+      30.22922993928958,
+      30.210448004241684,
+      30.18187862379938,
+      30.172519317040077,
+      30.13258263596103,
+      30.119914890354515,
+      30.090547162243443
+    ]
+  }
+}
\ No newline at end of file
diff --git a/generator/outputs/logs/p0_wgan.json b/generator/outputs/logs/p0_wgan.json
new file mode 100644
index 0000000..de4d558
--- /dev/null
+++ b/generator/outputs/logs/p0_wgan.json
@@ -0,0 +1,835 @@
+{
+  "run_name": "p0_wgan",
+  "config": {
+    "run_name": "p0_wgan",
+    "model": "wgan",
+    "image_size": 64,
+    "latent_dim": 128,
+    "ngf": 64,
+    "ndf": 64,
+    "epochs": 200,
+    "batch_size": 64,
+    "lr_g": 0.0001,
+    "lr_d": 0.0001,
+    "beta1": 0.0,
+    "beta2": 0.9,
+    "n_critic": 5,
+    "gp_lambda": 10,
+    "data_dir": "data",
+    "sources": [
+      "wiki"
+    ],
+    "subsample": 1.0,
+    "sample_interval": 10
+  },
+  "history": {
+    "g_loss": [
+      20.861786262219788,
+      24.484501724569206,
+      27.638709512531246,
+      29.97861582193619,
+      30.747561157259167,
+      32.02338776221642,
+      32.58458664885953,
+      32.48439505976489,
+      32.60738040646936,
+      33.003309531089585,
+      33.24444713347997,
+      33.23093043025742,
+      33.24205948348738,
+      33.317376132704254,
+      34.61083252817138,
+      35.4782878435575,
+      36.53802290533343,
+      36.75384027937539,
+      37.40440757458027,
+      37.68868056729308,
+      39.003664285708695,
+      40.184907000289,
+      40.930287255181206,
+      41.812280516339165,
+      43.252323949438896,
+      44.134575664487656,
+      44.931291221553444,
+      45.39425446436955,
+      45.518413951254296,
+      46.378283052362946,
+      47.568172291812736,
+      48.945309712336616,
+      50.388243675231934,
+      50.95788703005538,
+      52.825113280206665,
+      54.260103519146256,
+      54.699992840106674,
+      55.44318327944503,
+      56.13598096472585,
+      57.308240254720054,
+      58.77519280686338,
+      59.568145572629746,
+      60.179820125938484,
+      61.30784735720382,
+      62.04800804659852,
+      63.70076908209385,
+      65.67355463239882,
+      67.33006281730457,
+      68.60418007516454,
+      69.44280386378622,
+      70.77793149051503,
+      71.9100425752819,
+      73.41760413667076,
+      74.51540679605598,
+      75.452604782887,
+      78.0290306286934,
+      79.30968159080571,
+      81.6189056624714,
+      81.96908844841852,
+      83.14590193267561,
+      85.76070547511435,
+      87.85846160823463,
+      89.14186991178073,
+      90.0450213831714,
+      90.76150916988014,
+      92.32169165978065,
+      93.37803798251682,
+      93.73559182321924,
+      95.50361321930193,
+      98.06297312027368,
+      100.93305644825992,
+      102.58422611921262,
+      103.14872221661429,
+      104.5206852285271,
+      105.36486633822449,
+      106.95519620129186,
+      106.96137643471742,
+      109.60918868301262,
+      110.37198870406192,
+      111.43057123819987,
+      112.18183998368744,
+      114.1453188219641,
+      116.76321564372788,
+      118.57912932501898,
+      120.00456071511293,
+      121.75296041700575,
+      123.70820495409843,
+      124.7461356953678,
+      126.87444047846346,
+      128.34590644510382,
+      130.42692292042267,
+      133.1212801321959,
+      133.53329549284064,
+      134.6670674870157,
+      135.20846710857163,
+      135.7633072290665,
+      136.747866866935,
+      138.21183105207916,
+      141.35360564533462,
+      143.2757459787222,
+      143.69141322730954,
+      144.18379667885284,
+      145.65889485677084,
+      148.21175534501035,
+      149.56626344338443,
+      150.20800259581998,
+      150.77553385547083,
+      151.51304365630844,
+      152.34877252171182,
+      152.99383108024924,
+      154.3369990617801,
+      155.0102022611178,
+      156.3105682633881,
+      157.5694733644143,
+      160.06239657931857,
+      162.7010110382341,
+      163.5938239464393,
+      164.63512857551248,
+      165.91321407220303,
+      166.7981478699252,
+      168.27971988254123,
+      168.4678201919947,
+      171.6658657758664,
+      173.33122484908145,
+      175.36023030729376,
+      175.6105014768421,
+      175.98719784337231,
+      176.204406184009,
+      177.0283332563873,
+      177.69539446708484,
+      179.01333510570038,
+      180.97915476611536,
+      180.58325472448627,
+      180.95932714348166,
+      181.29422433967264,
+      183.12958875476804,
+      184.72860085250986,
+      185.62071612757495,
+      187.34349144829645,
+      188.05630806164865,
+      188.91328374748556,
+      191.1254445915548,
+      193.46458715455145,
+      193.5155346536229,
+      193.36818796141534,
+      194.24686131925665,
+      195.29775626027686,
+      196.8086325164534,
+      197.8473025916988,
+      198.7120099842039,
+      201.01687902466864,
+      202.734738146138,
+      203.01483610756378,
+      203.7443778209197,
+      204.40923987494574,
+      205.957951244126,
+      206.44182172595944,
+      207.52170383420764,
+      208.78479584261902,
+      209.91720509325336,
+      210.64656940688434,
+      211.7385871104705,
+      212.85203676142245,
+      215.43744362725153,
+      217.4224612244174,
+      218.57217746310764,
+      219.697512373965,
+      220.86698065863715,
+      221.4034518054408,
+      221.7560550005008,
+      223.57493336995444,
+      223.67208334115836,
+      224.73386810172317,
+      226.168578645103,
+      226.6503629765959,
+      227.94742900489743,
+      229.02756053565915,
+      230.93083852784247,
+      233.9660879933936,
+      234.1545915522127,
+      235.77698177761502,
+      238.3829990941235,
+      240.31401208730844,
+      240.7560836922409,
+      240.9462320376665,
+      243.4258527837248,
+      245.41892604338818,
+      247.01629814734827,
+      248.11878276075055,
+      250.72459242078992,
+      250.5925274058285,
+      250.52350140433026,
+      251.9943748213287,
+      252.25795233962882,
+      253.35346818581607,
+      254.52986927521533,
+      255.25971032819177,
+      256.89842582767847,
+      257.64977570884247,
+      259.6900681715745
+    ],
+    "c_loss": [
+      -11.094487810440553,
+      -12.968918835811138,
+      -13.043902460530274,
+      -12.723203683714576,
+      -12.381025380354647,
+      -11.684047953287763,
+      -10.904886481089465,
+      -10.52504440800756,
+      -10.343131445208163,
+      -10.157757118013183,
+      -9.974721721910003,
+      -9.812809922552512,
+      -9.648554773208422,
+      -9.48779460140783,
+      -9.35256861356588,
+      -9.245894571654818,
+      -9.137809677816868,
+      -9.044281771447931,
+      -8.902716490244245,
+      -8.837238697924166,
+      -8.756545627015267,
+      -8.717208363459656,
+      -8.639838231832547,
+      -8.595160510499246,
+      -8.554713706379266,
+      -8.463848097507775,
+      -8.455797974676145,
+      -8.391317713056873,
+      -8.3552374481136,
+      -8.357543099741656,
+      -8.349546556391266,
+      -8.302606204839853,
+      -8.291382703006779,
+      -8.294043176500207,
+      -8.274945335204787,
+      -8.272878311485313,
+      -8.207685407410318,
+      -8.192721331628976,
+      -8.192964592856223,
+      -8.161896302455512,
+      -8.14710073654468,
+      -8.158445111553892,
+      -8.170576908853324,
+      -8.166485660911624,
+      -8.177774573493211,
+      -8.15733466459137,
+      -8.154382880732546,
+      -8.182609847824791,
+      -8.186987533936131,
+      -8.22017425833604,
+      -8.147444259610952,
+      -8.178992070805307,
+      -8.195840435099395,
+      -8.200222732623416,
+      -8.172246947400591,
+      -8.180237101234937,
+      -8.173259688901076,
+      -8.19695118044176,
+      -8.206097737438657,
+      -8.219509731653407,
+      -8.190714883600544,
+      -8.206095539097086,
+      -8.188711438015993,
+      -8.200624450455361,
+      -8.217389455208412,
+      -8.238748331762778,
+      -8.233382728211891,
+      -8.23679813172063,
+      -8.21735338003207,
+      -8.215547685949211,
+      -8.238630391593665,
+      -8.227639290664948,
+      -8.234513328116165,
+      -8.23839012440454,
+      -8.278589575311056,
+      -8.26879572919291,
+      -8.274233976541417,
+      -8.253637050003068,
+      -8.291616514503442,
+      -8.312002382013526,
+      -8.266745121000156,
+      -8.275645991472095,
+      -8.318238798369707,
+      -8.331304942644566,
+      -8.349434795848328,
+      -8.362061607022575,
+      -8.346733781160454,
+      -8.338351263755408,
+      -8.34393249301828,
+      -8.367601214514842,
+      -8.377104435364409,
+      -8.338137237027158,
+      -8.39384140041139,
+      -8.408796421992472,
+      -8.384771224983735,
+      -8.431214986295778,
+      -8.445121946905408,
+      -8.446992207490474,
+      -8.434450187846124,
+      -8.410135668159555,
+      -8.48167750631642,
+      -8.496310178082211,
+      -8.477838894852203,
+      -8.470840746266216,
+      -8.479202500113063,
+      -8.496840595332989,
+      -8.539198466040128,
+      -8.540459174274377,
+      -8.542675428207103,
+      -8.549913498784735,
+      -8.576404001570157,
+      -8.600281522416658,
+      -8.605355482936934,
+      -8.622181242245889,
+      -8.597806969208598,
+      -8.571050169172448,
+      -8.609266292997924,
+      -8.619782570131813,
+      -8.637454705054944,
+      -8.651245410829526,
+      -8.63545006106043,
+      -8.691114932043936,
+      -8.66252594401694,
+      -8.693110690055752,
+      -8.695641701038065,
+      -8.69497519288308,
+      -8.729152616883946,
+      -8.697003765289612,
+      -8.701670956255027,
+      -8.703311920522616,
+      -8.757379066638453,
+      -8.71831601063411,
+      -8.755279973021944,
+      -8.764256343780419,
+      -8.756535425247288,
+      -8.778082993703011,
+      -8.806118020033223,
+      -8.802030087026772,
+      -8.790422684107075,
+      -8.827815474200454,
+      -8.827415256826287,
+      -8.788371637947531,
+      -8.807510222838475,
+      -8.815038907935477,
+      -8.838455163617416,
+      -8.836638889964835,
+      -8.874543673360455,
+      -8.898043274268124,
+      -8.891177387675672,
+      -8.883867200928881,
+      -8.90945321688285,
+      -8.892611440124677,
+      -8.90081204899356,
+      -8.974737249582242,
+      -8.995877582089511,
+      -8.978011010039573,
+      -8.964765571121479,
+      -8.94833901672281,
+      -8.994697517003772,
+      -9.007170154090632,
+      -9.004492255051932,
+      -9.03283164898554,
+      -9.010524407003674,
+      -9.029367656432662,
+      -9.008184736916148,
+      -9.081891494021455,
+      -9.031249945082209,
+      -9.01580849060645,
+      -9.05710388768433,
+      -9.092489253150092,
+      -9.102805704642568,
+      -9.079633062415649,
+      -9.122125975392832,
+      -9.102067274415598,
+      -9.122229276763068,
+      -9.113823464894915,
+      -9.1651386264043,
+      -9.138773479115246,
+      -9.12129195194978,
+      -9.144455577369428,
+      -9.15912618983505,
+      -9.099569693589826,
+      -9.179516656785948,
+      -9.159938207039469,
+      -9.163810334552053,
+      -9.16628072485964,
+      -9.170814365708935,
+      -9.18028780578548,
+      -9.171771639330785,
+      -9.176540715266494,
+      -9.218504903357264,
+      -9.205242362898641,
+      -9.266685975718703,
+      -9.23191982140907,
+      -9.279066171605356,
+      -9.251621839123908,
+      -9.304323559744754,
+      -9.28162112195268,
+      -9.273420568307248,
+      -9.262948013778427
+    ],
+    "gp": [
+      0.5177698262991051,
+      0.32633995146323475,
+      0.3352730232076001,
+      0.33070357247040816,
+      0.3239095399363176,
+      0.30454278661208806,
+      0.2779746312504777,
+      0.2639767948378864,
+      0.25723152828967977,
+      0.252030570951537,
+      0.24532575991928074,
+      0.24000449798300733,
+      0.23491869099183477,
+      0.22872555979933512,
+      0.22469883727498802,
+      0.220064683505294,
+      0.21621901580952418,
+      0.21379376741556042,
+      0.2097984680476095,
+      0.20661427188728362,
+      0.20377136689539144,
+      0.20342468525736754,
+      0.20043841858641198,
+      0.19893635901598591,
+      0.19794144260131905,
+      0.1943051668409355,
+      0.19465573021075414,
+      0.19184494037658734,
+      0.1914997584369575,
+      0.18998065067248207,
+      0.1899716500574962,
+      0.18859995196963492,
+      0.18787671985407156,
+      0.18748284899112272,
+      0.18725816535675882,
+      0.1872154682619957,
+      0.1853518881142522,
+      0.1849283434386946,
+      0.18445746381568082,
+      0.18389124695657405,
+      0.18332112415765348,
+      0.1837224014389973,
+      0.1841424989522013,
+      0.18398757278282435,
+      0.18406817878349707,
+      0.18342032233874023,
+      0.1843030681913226,
+      0.18425153753377957,
+      0.1846592110892137,
+      0.1856171176315118,
+      0.18453090630599064,
+      0.18509404775487548,
+      0.18647892560459603,
+      0.18637962277787606,
+      0.18475312971406518,
+      0.18427798755021182,
+      0.18496978683635976,
+      0.1846530988430365,
+      0.18474420231695365,
+      0.18577300554985138,
+      0.1852100855742509,
+      0.18612100577348045,
+      0.18495473497921328,
+      0.18559822938285564,
+      0.18602798696862072,
+      0.18625599825993558,
+      0.1866001136218891,
+      0.18649549230169032,
+      0.1861988907632155,
+      0.18649297679941637,
+      0.1870393530616903,
+      0.1861895910360747,
+      0.1866517885914469,
+      0.1863860412158518,
+      0.18728497976230252,
+      0.18858985906450917,
+      0.18852617581430672,
+      0.18753511904078163,
+      0.1890222365514208,
+      0.1899827563609833,
+      0.18920631618835987,
+      0.1891061297211891,
+      0.19059290648079838,
+      0.18992164304368525,
+      0.19108287151266132,
+      0.19222259860581314,
+      0.19121399836089376,
+      0.1904158524500253,
+      0.19051466597896854,
+      0.1912798127589318,
+      0.1919709156052424,
+      0.1917199712724257,
+      0.19173092738303343,
+      0.19258493237579477,
+      0.19325704786639952,
+      0.1940552502456637,
+      0.19389212638076045,
+      0.19462779848876166,
+      0.1945625018392107,
+      0.19347729461633742,
+      0.19455591976387904,
+      0.19560000822584825,
+      0.19542550470075035,
+      0.19546171316924774,
+      0.19573546171697792,
+      0.19611502263026365,
+      0.19738496275156983,
+      0.1983034890877387,
+      0.19725983886318846,
+      0.1979625270057182,
+      0.19858253714111118,
+      0.19945179645258643,
+      0.19973617394287593,
+      0.20025815809639105,
+      0.20051568513815754,
+      0.19859617654329703,
+      0.200360570781124,
+      0.2000493614687623,
+      0.2011680348497681,
+      0.20098042626284116,
+      0.20147523988579585,
+      0.20246979043715513,
+      0.20191304580841818,
+      0.20317427212229158,
+      0.20241337561199785,
+      0.203481401128965,
+      0.2042487176246623,
+      0.2033050367656424,
+      0.20330401931244588,
+      0.20448722128366306,
+      0.2056017825173008,
+      0.2044978688892901,
+      0.20509594100662792,
+      0.20612474274304185,
+      0.20609815314921567,
+      0.20594562079725612,
+      0.20677235599957489,
+      0.20683692295709247,
+      0.20680182855098683,
+      0.20762673772744122,
+      0.20906888082559794,
+      0.20729609510710098,
+      0.20717196234653149,
+      0.2078074644168471,
+      0.20861478004381695,
+      0.20803600462774433,
+      0.2101863404146881,
+      0.21134370377876308,
+      0.2105731888268238,
+      0.2113221597849814,
+      0.21105930607797752,
+      0.21121820924310106,
+      0.20985344312448287,
+      0.2128460541781453,
+      0.21422657355602462,
+      0.21399138209083665,
+      0.2134430699750907,
+      0.21320015956512378,
+      0.21464992287831455,
+      0.2151960316344968,
+      0.21466421451833512,
+      0.21605063033664346,
+      0.21490918992039493,
+      0.21623234108217754,
+      0.21466783753827084,
+      0.21653188908966184,
+      0.21584904650783443,
+      0.2152465660729978,
+      0.21701501436595239,
+      0.216723620130593,
+      0.21781324946727515,
+      0.2178196125448898,
+      0.21889628737082348,
+      0.21748662124523233,
+      0.21992958603251714,
+      0.2182036766288879,
+      0.220356303650854,
+      0.22012058855313,
+      0.2193760652700042,
+      0.21903627296734568,
+      0.22013778298392775,
+      0.21882927882938807,
+      0.2201864162380371,
+      0.22024842281308432,
+      0.22094041131054754,
+      0.22142581061866037,
+      0.22130067795037464,
+      0.22150409784581954,
+      0.22141598366901388,
+      0.22127340938418338,
+      0.2220577540942748,
+      0.22185662393577582,
+      0.22370209006799607,
+      0.22256857702492652,
+      0.2245172336219978,
+      0.22359555840301212,
+      0.22524864284847032,
+      0.22536892518082757,
+      0.22420646066053046,
+      0.2245028651359245
+    ],
+    "w_dist": [
+      16.272186049787162,
+      16.23231835446806,
+      16.396632681952592,
+      16.03023941414989,
+      15.62012077396751,
+      14.729475824649505,
+      13.684632797730291,
+      13.164812359850632,
+      12.91544673259442,
+      12.678062830215845,
+      12.427979311168688,
+      12.212854902968441,
+      11.99774168087887,
+      11.775050203209247,
+      11.59955698323046,
+      11.446541406761883,
+      11.299999834533452,
+      11.182219430320284,
+      11.000701170701243,
+      10.903381411234552,
+      10.794259293874099,
+      10.75145520593366,
+      10.644222415206784,
+      10.584524100866084,
+      10.534128125508632,
+      10.406899771731114,
+      10.402355284568582,
+      10.309767122350188,
+      10.270235024965727,
+      10.257349597898305,
+      10.249263062436343,
+      10.188605735648379,
+      10.170149900159256,
+      10.168871663574476,
+      10.147526990450348,
+      10.145032993544877,
+      10.061204280201189,
+      10.04200475643843,
+      10.037539231063981,
+      10.000808772877756,
+      9.980311978984082,
+      9.995669125287963,
+      10.012001891829009,
+      10.006361390790369,
+      10.018456371013942,
+      9.991537891290138,
+      9.997413560264127,
+      10.025125219883067,
+      10.033579638880536,
+      10.07634543883495,
+      9.992753319047457,
+      10.02993255680443,
+      10.060629701206823,
+      10.06401895376352,
+      10.019778240236464,
+      10.023016981793269,
+      10.022957552396333,
+      10.043482159345585,
+      10.05353976355658,
+      10.077239784827595,
+      10.042815731733274,
+      10.067305592593984,
+      10.038258795452933,
+      10.056606745923679,
+      10.07766933278141,
+      10.101308327047224,
+      10.099383861183105,
+      10.10175304575863,
+      10.079342282939164,
+      10.080477448813927,
+      10.109023923955412,
+      10.089535202939286,
+      10.10103121374407,
+      10.102250537709292,
+      10.151439376570227,
+      10.154694320809128,
+      10.159495730277815,
+      10.12898824642867,
+      10.1818388816638,
+      10.211829946795074,
+      10.158808282705458,
+      10.166707291562327,
+      10.22416786454681,
+      10.230521383040998,
+      10.26026351472251,
+      10.284287584744975,
+      10.258873768341848,
+      10.242509786491718,
+      10.249079161423904,
+      10.280399348593164,
+      10.296813586634446,
+      10.255336950579254,
+      10.311150676368644,
+      10.334645745693102,
+      10.317341701800999,
+      10.37176748748519,
+      10.38404320610894,
+      10.393270195855035,
+      10.380075213440469,
+      10.344908605070191,
+      10.427236706986383,
+      10.452310255653847,
+      10.432093941452157,
+      10.425457884307596,
+      10.43655711116954,
+      10.457990819164827,
+      10.513048097007294,
+      10.523494061853132,
+      10.515273819621806,
+      10.529538778973428,
+      10.56222937168218,
+      10.594799491686699,
+      10.60271722516443,
+      10.624762817121976,
+      10.602963824149885,
+      10.557011941763081,
+      10.612872007973161,
+      10.620276191907045,
+      10.649135054889902,
+      10.661049671662152,
+      10.650202459351625,
+      10.715812833085014,
+      10.681656411977906,
+      10.724853417812241,
+      10.719775449312664,
+      10.729789199177022,
+      10.771639792939533,
+      10.730054134792756,
+      10.734711149818866,
+      10.748184132372211,
+      10.813396890754383,
+      10.763294702513612,
+      10.806239384056159,
+      10.825503761747969,
+      10.817516952905907,
+      10.837539203350364,
+      10.873841577513609,
+      10.870399312076422,
+      10.858440966483881,
+      10.904082847660419,
+      10.918104064158898,
+      10.861332585261424,
+      10.87922984718257,
+      10.893113558516541,
+      10.924602958483574,
+      10.916998936579782,
+      10.976407082060462,
+      11.011480321639626,
+      10.996909273587743,
+      10.997088799109823,
+      11.020046273255952,
+      11.00479352250059,
+      10.999346480410328,
+      11.103197792248835,
+      11.138143314459388,
+      11.117924838595908,
+      11.099196266109107,
+      11.080340608776119,
+      11.141196748130346,
+      11.159130481165697,
+      11.151134406195748,
+      11.193337940965971,
+      11.159616310576084,
+      11.191691067687465,
+      11.15486309997037,
+      11.247210380358577,
+      11.18974041408963,
+      11.168274154011005,
+      11.227254029624483,
+      11.259725456563835,
+      11.28093820555597,
+      11.257829186855218,
+      11.311088848929126,
+      11.276933490720557,
+      11.32152514335437,
+      11.295860225318833,
+      11.368701667459614,
+      11.339979364118022,
+      11.315052599784654,
+      11.334818306947367,
+      11.360504019973623,
+      11.287862481011286,
+      11.381380821293243,
+      11.362422434488947,
+      11.373214447803981,
+      11.380538829575231,
+      11.383821145082136,
+      11.395328782562515,
+      11.385931468213734,
+      11.389274818876869,
+      11.439082440759393,
+      11.423808608095877,
+      11.50370687501044,
+      11.457605593428662,
+      11.524238508175593,
+      11.487577428573214,
+      11.556809991037749,
+      11.535310363769538,
+      11.515485166892036,
+      11.507976668920278
+    ]
+  }
+}
\ No newline at end of file
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0010.png b/generator/outputs/samples/p0_ddpm/epoch_0010.png
new file mode 100644
index 0000000..a0ca83f
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0010.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0020.png b/generator/outputs/samples/p0_ddpm/epoch_0020.png
new file mode 100644
index 0000000..5810fcc
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0020.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0030.png b/generator/outputs/samples/p0_ddpm/epoch_0030.png
new file mode 100644
index 0000000..1d42413
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0030.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0040.png b/generator/outputs/samples/p0_ddpm/epoch_0040.png
new file mode 100644
index 0000000..30f04fe
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0040.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0050.png b/generator/outputs/samples/p0_ddpm/epoch_0050.png
new file mode 100644
index 0000000..ac87077
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0050.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0060.png b/generator/outputs/samples/p0_ddpm/epoch_0060.png
new file mode 100644
index 0000000..7aaafb5
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0060.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0070.png b/generator/outputs/samples/p0_ddpm/epoch_0070.png
new file mode 100644
index 0000000..5223092
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0070.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0080.png b/generator/outputs/samples/p0_ddpm/epoch_0080.png
new file mode 100644
index 0000000..a6c8ae9
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0080.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0090.png b/generator/outputs/samples/p0_ddpm/epoch_0090.png
new file mode 100644
index 0000000..3ba7980
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0090.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0100.png b/generator/outputs/samples/p0_ddpm/epoch_0100.png
new file mode 100644
index 0000000..ca7e523
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0100.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0110.png b/generator/outputs/samples/p0_ddpm/epoch_0110.png
new file mode 100644
index 0000000..db7a887
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0110.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0120.png b/generator/outputs/samples/p0_ddpm/epoch_0120.png
new file mode 100644
index 0000000..fc86332
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0120.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0130.png b/generator/outputs/samples/p0_ddpm/epoch_0130.png
new file mode 100644
index 0000000..6d7085b
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0130.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0140.png b/generator/outputs/samples/p0_ddpm/epoch_0140.png
new file mode 100644
index 0000000..52a0f8e
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0140.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0150.png b/generator/outputs/samples/p0_ddpm/epoch_0150.png
new file mode 100644
index 0000000..b31d93f
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0150.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0160.png b/generator/outputs/samples/p0_ddpm/epoch_0160.png
new file mode 100644
index 0000000..0a09b1b
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0160.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0170.png b/generator/outputs/samples/p0_ddpm/epoch_0170.png
new file mode 100644
index 0000000..6d7f0e5
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0170.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0180.png b/generator/outputs/samples/p0_ddpm/epoch_0180.png
new file mode 100644
index 0000000..945c1e4
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0180.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0190.png b/generator/outputs/samples/p0_ddpm/epoch_0190.png
new file mode 100644
index 0000000..e160e65
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0190.png differ
diff --git a/generator/outputs/samples/p0_ddpm/epoch_0200.png b/generator/outputs/samples/p0_ddpm/epoch_0200.png
new file mode 100644
index 0000000..e4e70ec
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm/epoch_0200.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0010.png b/generator/outputs/samples/p0_ddpm_small/epoch_0010.png
new file mode 100644
index 0000000..6a60abd
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0010.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0020.png b/generator/outputs/samples/p0_ddpm_small/epoch_0020.png
new file mode 100644
index 0000000..5e0ac12
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0020.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0030.png b/generator/outputs/samples/p0_ddpm_small/epoch_0030.png
new file mode 100644
index 0000000..d1cbd6e
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0030.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0040.png b/generator/outputs/samples/p0_ddpm_small/epoch_0040.png
new file mode 100644
index 0000000..13ffc43
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0040.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0050.png b/generator/outputs/samples/p0_ddpm_small/epoch_0050.png
new file mode 100644
index 0000000..be57788
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0050.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0060.png b/generator/outputs/samples/p0_ddpm_small/epoch_0060.png
new file mode 100644
index 0000000..1a0e29e
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0060.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0070.png b/generator/outputs/samples/p0_ddpm_small/epoch_0070.png
new file mode 100644
index 0000000..58e7d85
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0070.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0080.png b/generator/outputs/samples/p0_ddpm_small/epoch_0080.png
new file mode 100644
index 0000000..a9d11ff
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0080.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0090.png b/generator/outputs/samples/p0_ddpm_small/epoch_0090.png
new file mode 100644
index 0000000..2802381
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0090.png differ
diff --git a/generator/outputs/samples/p0_ddpm_small/epoch_0100.png b/generator/outputs/samples/p0_ddpm_small/epoch_0100.png
new file mode 100644
index 0000000..4fb73c3
Binary files /dev/null and b/generator/outputs/samples/p0_ddpm_small/epoch_0100.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0005.png b/generator/outputs/samples/p0_vae/epoch_0005.png
new file mode 100644
index 0000000..2195d13
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0005.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0010.png b/generator/outputs/samples/p0_vae/epoch_0010.png
new file mode 100644
index 0000000..277c9fc
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0010.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0015.png b/generator/outputs/samples/p0_vae/epoch_0015.png
new file mode 100644
index 0000000..3869a3f
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0015.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0020.png b/generator/outputs/samples/p0_vae/epoch_0020.png
new file mode 100644
index 0000000..2c348bc
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0020.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0025.png b/generator/outputs/samples/p0_vae/epoch_0025.png
new file mode 100644
index 0000000..300bd8d
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0025.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0030.png b/generator/outputs/samples/p0_vae/epoch_0030.png
new file mode 100644
index 0000000..ca47b45
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0030.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0035.png b/generator/outputs/samples/p0_vae/epoch_0035.png
new file mode 100644
index 0000000..3789be2
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0035.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0040.png b/generator/outputs/samples/p0_vae/epoch_0040.png
new file mode 100644
index 0000000..1d98d53
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0040.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0045.png b/generator/outputs/samples/p0_vae/epoch_0045.png
new file mode 100644
index 0000000..89783fe
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0045.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0050.png b/generator/outputs/samples/p0_vae/epoch_0050.png
new file mode 100644
index 0000000..f6379c2
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0050.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0055.png b/generator/outputs/samples/p0_vae/epoch_0055.png
new file mode 100644
index 0000000..2a70fcd
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0055.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0060.png b/generator/outputs/samples/p0_vae/epoch_0060.png
new file mode 100644
index 0000000..4d945b9
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0060.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0065.png b/generator/outputs/samples/p0_vae/epoch_0065.png
new file mode 100644
index 0000000..8076438
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0065.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0070.png b/generator/outputs/samples/p0_vae/epoch_0070.png
new file mode 100644
index 0000000..59fcdfd
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0070.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0075.png b/generator/outputs/samples/p0_vae/epoch_0075.png
new file mode 100644
index 0000000..e02ec59
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0075.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0080.png b/generator/outputs/samples/p0_vae/epoch_0080.png
new file mode 100644
index 0000000..3d3496d
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0080.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0085.png b/generator/outputs/samples/p0_vae/epoch_0085.png
new file mode 100644
index 0000000..7765e91
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0085.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0090.png b/generator/outputs/samples/p0_vae/epoch_0090.png
new file mode 100644
index 0000000..179d16b
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0090.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0095.png b/generator/outputs/samples/p0_vae/epoch_0095.png
new file mode 100644
index 0000000..c88b6d8
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0095.png differ
diff --git a/generator/outputs/samples/p0_vae/epoch_0100.png b/generator/outputs/samples/p0_vae/epoch_0100.png
new file mode 100644
index 0000000..4c0dd7a
Binary files /dev/null and b/generator/outputs/samples/p0_vae/epoch_0100.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0010.png b/generator/outputs/samples/p0_wgan/epoch_0010.png
new file mode 100644
index 0000000..b98ad13
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0010.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0020.png b/generator/outputs/samples/p0_wgan/epoch_0020.png
new file mode 100644
index 0000000..1ec87bf
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0020.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0030.png b/generator/outputs/samples/p0_wgan/epoch_0030.png
new file mode 100644
index 0000000..9bf605b
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0030.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0040.png b/generator/outputs/samples/p0_wgan/epoch_0040.png
new file mode 100644
index 0000000..bb4e258
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0040.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0050.png b/generator/outputs/samples/p0_wgan/epoch_0050.png
new file mode 100644
index 0000000..637ddc4
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0050.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0060.png b/generator/outputs/samples/p0_wgan/epoch_0060.png
new file mode 100644
index 0000000..6e2e4a9
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0060.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0070.png b/generator/outputs/samples/p0_wgan/epoch_0070.png
new file mode 100644
index 0000000..6f3d4f7
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0070.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0080.png b/generator/outputs/samples/p0_wgan/epoch_0080.png
new file mode 100644
index 0000000..a634b6c
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0080.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0090.png b/generator/outputs/samples/p0_wgan/epoch_0090.png
new file mode 100644
index 0000000..70378ad
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0090.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0100.png b/generator/outputs/samples/p0_wgan/epoch_0100.png
new file mode 100644
index 0000000..39a76db
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0100.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0110.png b/generator/outputs/samples/p0_wgan/epoch_0110.png
new file mode 100644
index 0000000..3bb827a
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0110.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0120.png b/generator/outputs/samples/p0_wgan/epoch_0120.png
new file mode 100644
index 0000000..0db25d9
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0120.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0130.png b/generator/outputs/samples/p0_wgan/epoch_0130.png
new file mode 100644
index 0000000..9d45db7
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0130.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0140.png b/generator/outputs/samples/p0_wgan/epoch_0140.png
new file mode 100644
index 0000000..51caef5
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0140.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0150.png b/generator/outputs/samples/p0_wgan/epoch_0150.png
new file mode 100644
index 0000000..ab80ef0
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0150.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0160.png b/generator/outputs/samples/p0_wgan/epoch_0160.png
new file mode 100644
index 0000000..7b46155
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0160.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0170.png b/generator/outputs/samples/p0_wgan/epoch_0170.png
new file mode 100644
index 0000000..b0b0500
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0170.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0180.png b/generator/outputs/samples/p0_wgan/epoch_0180.png
new file mode 100644
index 0000000..404069d
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0180.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0190.png b/generator/outputs/samples/p0_wgan/epoch_0190.png
new file mode 100644
index 0000000..048546b
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0190.png differ
diff --git a/generator/outputs/samples/p0_wgan/epoch_0200.png b/generator/outputs/samples/p0_wgan/epoch_0200.png
new file mode 100644
index 0000000..bb4b164
Binary files /dev/null and b/generator/outputs/samples/p0_wgan/epoch_0200.png differ