config.json

{
  "activation_op": "prelu",
  "activation_params": null,
  "architectures": [
    "cceVAE"
  ],
  "auto_map": {
    "AutoConfig": "cceVAEConfig.cceVAEConfig",
    "AutoModel": "cceVAE.cceVAE"
  },
  "block_op": null,
  "block_params": null,
  "conv_params": null,
  "d": 2,
  "fmap_sizes": [
    16,
    64,
    256,
    1024
  ],
  "input_size": [
    1,
    256,
    256
  ],
  "model_type": "cceVAE",
  "normalization_op": null,
  "normalization_params": null,
  "tconv_params": null,
  "to_1x1": true,
  "torch_dtype": "float32",
  "transformers_version": "4.40.1",
  "z_dim": 1024
}