| { | |
| "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" | |
| } | |