reproduce_mvae_mnist / model_config.json
asenella's picture
Uploading MVAE in asenella/reproduce_mvae_mnist
f679b80
{"name": "MVAEConfig", "n_modalities": 2, "latent_dim": 64, "input_dims": {"images": [1, 28, 28], "labels": [1]}, "uses_likelihood_rescaling": true, "rescale_factors": {"images": 1, "labels": 50}, "decoders_dist": {"images": "bernoulli", "labels": "categorical"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders"], "k": 0, "warmup": 200}