resnet_vae_celeba / model_config.json
onlytojay's picture
Uploading VAE in onlytojay/resnet_vae_celeba
f8b1b9c verified
{"name": "VAEConfig", "input_dim": [3, 64, 64], "latent_dim": 64, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse"}