binary-vae-mnist / config.json
jolespin's picture
Upload model
c7e9699 verified
{
"learning_rate": 0.001,
"input_dim": 784,
"hidden_dims": [
512,
256,
128
],
"latent_dim": 24,
"activation_fn": "ReLU",
"beta": 1.0
}