Disentangled-BETA-VAE-TMNIST / model_config.json
Allayte's picture
Uploading DisentangledBetaVAE in Allayte/Disentangled-BETA-VAE-TMNIST
518adc6
raw
history blame
217 Bytes
{"name": "DisentangledBetaVAEConfig", "input_dim": [1, 28, 28], "latent_dim": 16, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse", "beta": 5.0, "C": 30.0, "warmup_epoch": 25}