Upload 6 files

decoders.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e2a28d6477ebd1b86b5b85fdb6ba4a999047849e94fcbb839ed30bc9a5163590
-size 22915318
+oid sha256:2e95ed9d72fb19db8d4bf45421a4455c1c2c8edad3a134b2f40393d113d374e2
+size 22915312

encoders.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d2538d8aa67b784961be2c2165941e4f48b3dcc0dcd84ca0d911e26d52808340
-size 43873160
+oid sha256:21142df1a5a44ffe3c772ad7cef73c4ddf5da6ff71d0ef47011fa9611c36ab24
+size 43873109

model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f42398cbecf8832ff6acfd3f24b3735518640479a621743658b2fcc42b7b8d83
-size 66735765
+oid sha256:6d5f96646703b5dc4a49affa45284ce359e70e632c8e00ea90046031ef0550b6
+size 66735829

model_config.json

@@ -1 +1 @@
-{"name": "MMVAEConfig", "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, "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}}, "custom_architectures": ["encoders", "decoders"], "K": 10, "prior_and_posterior_dist": "laplace_with_softmax", "learn_prior": false, "beta": 1.0}
+{"name": "MMVAEConfig", "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, "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}}, "custom_architectures": ["encoders", "decoders"], "K": 10, "prior_and_posterior_dist": "normal", "learn_prior": false, "beta": 1.0}

training_config.json

@@ -0,0 +1 @@
+{"name": "BaseTrainerConfig", "output_dir": "compare_on_mmnist/config2/MMVAE/seed_2/missing_ratio_0.2/", "per_device_train_batch_size": 256, "per_device_eval_batch_size": 64, "num_epochs": 100, "train_dataloader_num_workers": 0, "eval_dataloader_num_workers": 0, "optimizer_cls": "Adam", "optimizer_params": {}, "scheduler_cls": "ReduceLROnPlateau", "scheduler_params": {"patience": 30}, "learning_rate": 0.001, "steps_saving": null, "steps_predict": 5, "keep_best_on_train": false, "seed": 2, "no_cuda": false, "world_size": -1, "local_rank": -1, "rank": -1, "dist_backend": "nccl", "master_addr": "localhost", "master_port": "12345", "start_keep_best_epoch": 0, "drop_last": false}

wandb_info.json

@@ -0,0 +1 @@
+{"entity_name": "multimodal_vaes", "project_name": "compare_on_mmnist", "id": "1ft1lh4r", "path": "multimodal_vaes/compare_on_mmnist/1ft1lh4r"}