config.json

{
  "dim_in": 16,
  "dim_out": 4,
  "gradient_checkpointing": false,
  "init_features": 48,
  "n_spatial_dims": 3,
  "spatial_resolution": [
    128,
    128,
    128
  ]
}