ms_config_1_JNFGMC_irqwi18o / model_config.json
asenella's picture
Uploading JNFGMC in asenella/ms_config_1_JNFGMC_irqwi18o
1206d52 verified
raw
history blame contribute delete
818 Bytes
{"name": "JNFGMCConfig", "n_modalities": 2, "latent_dim": 20, "input_dims": {"mnist": [1, 28, 28], "svhn": [3, 32, 32]}, "uses_likelihood_rescaling": true, "rescale_factors": null, "decoders_dist": {"mnist": "laplace", "svhn": "laplace"}, "decoder_dist_params": {"mnist": {"scale": 0.75}, "svhn": {"scale": 0.75}}, "logits_to_std": "standard", "custom_architectures": ["gmc_model", "decoders", "joint_encoder"], "warmup": 100, "annealing": false, "alpha": 0.1, "nb_epochs_gmc": 150, "beta": 2.5, "gmc_config": {"name": "GMCConfig", "n_modalities": 2, "input_dims": {"mnist": [1, 28, 28], "svhn": [3, 32, 32]}, "common_dim": 10, "latent_dim": 10, "temperature": 0.1, "custom_architectures": ["processors", "shared_encoder"], "loss": "between_modality_pairs"}, "add_reconstruction_terms": false, "divide_by_prior": true}