vae/config.json

{
  "_class_name": "OobleckVAE",
  "io_channels": 2,
  "latent_dim": 64,
  "downsampling_ratio": 2048,
  "encoder_config": {
    "in_channels": 2,
    "channels": 128,
    "c_mults": [
      1,
      2,
      4,
      8,
      16
    ],
    "strides": [
      2,
      4,
      4,
      8,
      8
    ],
    "latent_dim": 128,
    "use_snake": true
  },
  "decoder_config": {
    "out_channels": 2,
    "channels": 128,
    "c_mults": [
      1,
      2,
      4,
      8,
      16
    ],
    "strides": [
      2,
      4,
      4,
      8,
      8
    ],
    "latent_dim": 64,
    "use_snake": true,
    "final_tanh": false
  }
}