reproduce_jmvae / model_config.json
asenella's picture
Uploading JMVAE in asenella/reproduce_jmvae
bbe586c
raw
history blame contribute delete
365 Bytes
{"name": "JMVAEConfig", "n_modalities": 2, "latent_dim": 64, "input_dims": {"images": [1, 28, 28], "labels": [1]}, "uses_likelihood_rescaling": false, "rescale_factors": null, "decoders_dist": {"images": "bernoulli", "labels": "categorical"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders", "joint_encoder"], "alpha": 0.1, "warmup": 200}