Upload 6 files

decoders.pkl

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

encoders.pkl

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

metrics.log

@@ -0,0 +1,23 @@
+Uploading MMVAE model to asenella/reproducing_mmvae_4 repo in HF hub...
+Successfully uploaded MMVAE to asenella/reproducing_mmvae_4 repo in HF hub!
+Subset ['mnist'] accuracies 
+{'svhn': 0.8451766666666667}
+Mean subset ['mnist'] accuracies : 0.8451766666666667
+Subset ['svhn'] accuracies 
+{'mnist': 0.6358066666666666}
+Mean subset ['svhn'] accuracies : 0.6358066666666666
+Conditional accuracies for 1 modalities : 0.7404916666666667 +- 0.10468500000000003
+Subset ['mnist'] accuracies 
+{'svhn': 0.8448}
+Mean subset ['mnist'] accuracies : 0.8448
+Subset ['svhn'] accuracies 
+{'mnist': 0.63636}
+Mean subset ['svhn'] accuracies : 0.63636
+Conditional accuracies for 1 modalities : 0.74058 +- 0.10421999999999998
+Subset ['mnist'] accuracies 
+{'svhn': 0.84403}
+Mean subset ['mnist'] accuracies : 0.84403
+Subset ['svhn'] accuracies 
+{'mnist': 0.6360233333333334}
+Mean subset ['svhn'] accuracies : 0.6360233333333334
+Conditional accuracies for 1 modalities : 0.7400266666666666 +- 0.10400333333333328

model.pt

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

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}