model_config.json

{"name": "JNFConfig", "n_modalities": 2, "latent_dim": 20, "input_dims": {"mnist": [1, 28, 28], "svhn": [3, 32, 32]}, "uses_likelihood_rescaling": false, "rescale_factors": null, "decoders_dist": {"mnist": "laplace", "svhn": "laplace"}, "decoder_dist_params": {"mnist": {"scale": 0.75}, "svhn": {"scale": 0.75}}, "custom_architectures": ["encoders", "decoders"], "warmup": 100, "beta": 0.5}