config.json

{
  "blocks_at_neck": 1,
  "blocks_per_stage": 2,
  "dim_in": 9,
  "dim_out": 2,
  "gradient_checkpointing": false,
  "init_features": 42,
  "n_spatial_dims": 2,
  "spatial_resolution": [
    1024,
    256
  ],
  "stages": 4
}