model_config.json

{"name": "PoincareVAEConfig", "input_dim": [1, 28, 28], "latent_dim": 10, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce", "prior_distribution": "wrapped_normal", "posterior_distribution": "wrapped_normal", "curvature": 0.7}