vae/config.json

{
  "embed_dim": 4,
  "monitor": "val/rec_loss",
  "ddconfig": {
    "double_z": true,
    "z_channels": 4,
    "resolution": 256,
    "in_channels": 3,
    "out_ch": 3,
    "ch": 128,
    "ch_mult": [
      1,
      2,
      4,
      4
    ],
    "num_res_blocks": 2,
    "attn_resolutions": [],
    "dropout": 0.0
  },
  "lossconfig": {
    "target": "torch.nn.Identity"
  },
  "_target": "crs_core.autoencoder.AutoencoderKL"
}