{"name": "JNFConfig", "n_modalities": 5, "latent_dim": 200, "input_dims": {"m0": [3, 28, 28], "m1": [3, 28, 28], "m2": [3, 28, 28], "m3": [3, 28, 28], "m4": [3, 28, 28]}, "uses_likelihood_rescaling": false, "rescale_factors": null, "decoders_dist": {"m0": "laplace", "m1": "laplace", "m2": "laplace", "m3": "laplace", "m4": "laplace"}, "decoder_dist_params": {"m0": {"scale": 0.75}, "m1": {"scale": 0.75}, "m2": {"scale": 0.75}, "m3": {"scale": 0.75}, "m4": {"scale": 0.75}}, "logits_to_std": "softplus", "custom_architectures": ["encoders", "decoders", "joint_encoder", "flows"], "warmup": 150, "two_steps_training": true, "alpha": 0.1, "beta": 0.5} |