reproduced_hvae / model_config.json
clementchadebec's picture
Uploading HVAE in clementchadebec/reproduced_hvae
a3e16b2
{"name": "HVAEConfig", "input_dim": [1, 28, 28], "latent_dim": 64, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce", "n_lf": 4, "eps_lf": 0.05, "beta_zero": 1.0, "learn_eps_lf": false, "learn_beta_zero": false}