reproducing_mopoe_2 / model_config.json
asenella's picture
Update model_config.json
2631005
{"name": "MoPoEConfig", "n_modalities": 5, "latent_dim": 512, "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, "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}}, "custom_architectures": ["encoders", "decoders"], "subsets": {"": [], "m0": ["m0"], "m1": ["m1"], "m2": ["m2"], "m3": ["m3"], "m4": ["m4"], "m0_m1": ["m0", "m1"], "m0_m2": ["m0", "m2"], "m0_m3": ["m0", "m3"], "m0_m4": ["m0", "m4"], "m1_m2": ["m1", "m2"], "m1_m3": ["m1", "m3"], "m1_m4": ["m1", "m4"], "m2_m3": ["m2", "m3"], "m2_m4": ["m2", "m4"], "m3_m4": ["m3", "m4"], "m0_m1_m2": ["m0", "m1", "m2"], "m0_m1_m3": ["m0", "m1", "m3"], "m0_m1_m4": ["m0", "m1", "m4"], "m0_m2_m3": ["m0", "m2", "m3"], "m0_m2_m4": ["m0", "m2", "m4"], "m0_m3_m4": ["m0", "m3", "m4"], "m1_m2_m3": ["m1", "m2", "m3"], "m1_m2_m4": ["m1", "m2", "m4"], "m1_m3_m4": ["m1", "m3", "m4"], "m2_m3_m4": ["m2", "m3", "m4"], "m0_m1_m2_m3": ["m0", "m1", "m2", "m3"], "m0_m1_m2_m4": ["m0", "m1", "m2", "m4"], "m0_m1_m3_m4": ["m0", "m1", "m3", "m4"], "m0_m2_m3_m4": ["m0", "m2", "m3", "m4"], "m1_m2_m3_m4": ["m1", "m2", "m3", "m4"], "m0_m1_m2_m3_m4": ["m0", "m1", "m2", "m3", "m4"]}, "beta": 2.5}