Upload 6 files

decoders.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bfef2d35aa132bac638ddeba8fcc562436f36464b1d769a8b104aa8b7548defc
+size 2143155

encoders.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:745f81d7c0b5b0956471b3bd4b29b640619b61f861569374aca7d511eb6beb2f
+size 2337564

metrics.log

@@ -0,0 +1,7 @@
+Subset ['mnist'] accuracies 
+{'svhn': 0.8311866666666666}
+Mean subset ['mnist'] accuracies : 0.8311866666666666
+Subset ['svhn'] accuracies 
+{'mnist': 0.6183066666666667}
+Mean subset ['svhn'] accuracies : 0.6183066666666667
+Conditional accuracies for 1 modalities : 0.7247466666666667 +- 0.10643999999999998

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f6ff4e2ddd6dc041b91e2dcbc7b45f499cb2db553f1cdd496d18b71b48cfbfdd
+size 4437067

model_config.json

@@ -0,0 +1 @@
+{"name": "MMVAEConfig", "n_modalities": 2, "latent_dim": 20, "input_dims": {"mnist": [1, 28, 28], "svhn": [3, 32, 32]}, "uses_likelihood_rescaling": true, "decoders_dist": {"mnist": "laplace", "svhn": "laplace"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders"], "K": 30, "prior_and_posterior_dist": "laplace_with_softmax", "learn_prior": true}

training_config.json

@@ -0,0 +1 @@
+{"name": "BaseTrainerConfig", "output_dir": "dummy_output_dir", "per_device_train_batch_size": 128, "per_device_eval_batch_size": 128, "num_epochs": 30, "train_dataloader_num_workers": 2, "train_dataloader_pin_memory": true, "eval_dataloader_num_workers": 0, "optimizer_cls": "Adam", "optimizer_params": {"amsgrad": true}, "scheduler_cls": null, "scheduler_params": null, "learning_rate": 0.001, "steps_saving": null, "steps_predict": 1, "keep_best_on_train": false, "seed": 8, "no_cuda": false, "world_size": -1, "local_rank": -1, "rank": -1, "dist_backend": "nccl", "master_addr": "localhost", "master_port": "12345", "start_keep_best_epoch": 30, "drop_last": false}