config.json

{
  "dim": 2,
  "modality": "image",
  "input_channels": 3,
  "ps": [
    32,
    32,
    32,
    16,
    16,
    16,
    16,
    16,
    16,
    8,
    8,
    8,
    8,
    8,
    8,
    4,
    4,
    4,
    4,
    4,
    4,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2,
    2
  ],
  "decoder_ps": 8,
  "decoder_dim": 768,
  "decoder_kernel_size": 3,
  "decoder_arch": "CCCCCCCCCCCC",
  "decoder_mlp_ratio": 4.0,
  "decoder_layerscale": true,
  "decoder_layerscale_init": 1e-06,
  "encoder_arch": "SC8",
  "linear_input": false,
  "cumulative_channels": [
    3,
    6,
    9,
    12,
    15,
    18,
    21,
    24,
    30
  ],
  "n_trained": 30
}