CUB_JNFGMC_v9wdsb19 / model_config.json
asenella's picture
Uploading JNFGMC in asenella/CUB_JNFGMC_v9wdsb19
ecef50f verified
{"name": "JNFGMCConfig", "n_modalities": 2, "latent_dim": 64, "input_dims": null, "uses_likelihood_rescaling": true, "rescale_factors": {"image": 0.0026041666666666665, "text": 5.0}, "decoders_dist": {"image": "laplace", "text": "categorical"}, "decoder_dist_params": {"image": {"scale": 0.01}}, "logits_to_std": "softplus", "custom_architectures": ["gmc_model", "encoders", "decoders", "joint_encoder", "flows"], "warmup": 150, "annealing": false, "alpha": 0.1, "nb_epochs_gmc": 80, "beta": 1.0, "gmc_config": {"name": "GMCConfig", "n_modalities": 2, "input_dims": null, "common_dim": 32, "latent_dim": 32, "temperature": 0.1, "custom_architectures": ["processors", "shared_encoder"], "loss": "between_modality_pairs"}, "add_reconstruction_terms": false, "divide_by_prior": true}