benchmark_VAE
benchmark_VAE copied to clipboard
Add context to VAE-IAF
New features
- Add the possibility to add a context vector
htoMADE. This is made to fit to Fig.2 in https://arxiv.org/abs/1606.04934 where the IAF are parametrized by a context vectorh. - Add a
context_dimargument in theMADEConfig,IAFConfig,MAFConfig,IAFSamplerConfig,MAFSamplerConfigandVAE_IAF_Config. - :warning: if the argument
context_dimis different fromNonetheVAE_IAFmodel expects the encoder to output a context vectorhin addition to the parameters of the posterior distribution.
Minor changes
- rename
n_made_blocks-->n_blocksinIAFandMAFconfigs - add automatic context layer in benchmarks neural nets.