dense/vae/config.json

{
  "model_type": "vae",
  "model_class": "pixal3d.models.autoencoders.dense_vae.DenseShapeVAE",
  "use_checkpoint": true,
  "embed_dim": 8,
  "in_channels": 1,
  "out_channels": 1,
  "model_channels_encoder": [
    32,
    128,
    512
  ],
  "num_res_blocks_encoder": 2,
  "num_res_blocks_middle_encoder": 2,
  "model_channels_decoder": [
    512,
    128,
    64
  ],
  "num_res_blocks_decoder": 4,
  "num_res_blocks_middle_decoder": 4,
  "use_fp16": true,
  "latents_scale": 1.0,
  "latents_shift": 0.0
}