config.json

{
  "input_dim": 128,
  "hidden_dim": 512,
  "num_layers": 3,
  "z_dim": 20
}