config.json

{
  "aeconfig": {
    "z_channels": 4,
    "resolution": 128,
    "in_channels": 96,
    "out_ch": 96,
    "ch": 256,
    "ch_mult": [4, 4, 2, 1],
    "res_updown_level": [4],
    "num_res_blocks": 2,
    "dropout": 0.1
  },
  "embed_dim": 4,
  "latent_scale_factor": null,
  "kl_weight": 1e-06,
  "geom_weight": 1.0,
  "recon_loss_type": "l2"
}