config.json

{
  "name": "FNO2d",
  "input_channels": 10,
  "output_channels": 1,
  "use_grid": true,
  "grid_channels": 2,
  "grid_include_endpoint": false,
  "width": 32,
  "modes1": 12,
  "modes2": 12,
  "num_layers": 4,
  "projection_width": 128,
  "activation": "relu",
  "normalization": "batch_norm",
  "block_order": "relu(batch_norm(spectral_plus_pointwise))",
  "fft_norm": "backward",
  "spectral_init": "scaled_uniform_complex"
}