config.json

{
  "dim_in": 24,
  "dim_out": 6,
  "gradient_checkpointing": false,
  "hidden_channels": 128,
  "modes1": 16,
  "modes2": 16,
  "modes3": 16,
  "n_spatial_dims": 3,
  "spatial_resolution": [
    64,
    64,
    64
  ]
}