Disentangled-BETA-VAE-TMNIST-15 / model_config.json

Uploading DisentangledBetaVAE in Allayte/Disentangled-BETA-VAE-TMNIST-15

550d718 over 1 year ago

No virus

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}