reproduced_vae / model_config.json
clementchadebec's picture
Uploading VAE in clementchadebec/reproduced_vae
06aeb60
{"name": "VAEConfig", "input_dim": [784], "latent_dim": 40, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce"}