reproduced_beta_tc_vae / model_config.json
clementchadebec's picture
Uploading BetaTCVAE in clementchadebec/reproduced_beta_tc_vae
9765a0c
raw
history blame contribute delete
219 Bytes
{"name": "BetaTCVAEConfig", "input_dim": [64, 64], "latent_dim": 10, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce", "alpha": 1.0, "beta": 6.0, "gamma": 1.0, "use_mss": false}