search

jmtomczak / vae_vampprior

Code for the paper "VAE with a VampPrior", J.M. Tomczak & M. Welling
https://jmtomczak.github.io/deebmed.html
MIT License
223 stars 50 forks source link

Why use GatedDense in VAE? #6

Open ShellingFord221 opened 3 years ago

ShellingFord221 commented 3 years ago

I notice that you use GatedDense in self.q_z_layers to encode q(z | x):

image

Is there any reason to use GatedDense rather than a simple MLP? Thanks!