config.json

{
  "name": "fno",
  "num_label_samples": 64,
  "in_chan": 2,
  "out_chan": 2,
  "ffn_depth": 2,
  "ffn_width": 64,
  "autoffn_depth": 2,
  "autoffn_width": 64,
  "deeponet_width": 64,
  "branch_depth": 2,
  "trunk_depth": 2,
  "act_fn": "relu",
  "act_scale_invariant": true,
  "act_on_output": false,
  "autoedeeponet_width": 64,
  "autoedeeponet_depth": 2,
  "autoedeeponet_act_fn": "relu",
  "fno_depth": 2,
  "fno_hidden_dim": 16,
  "fno_modes_x": 4,
  "fno_modes_y": 4,
  "unet_dim": 8,
  "unet_insert_case_params_at": "input",
  "resnet_depth": 2,
  "resnet_hidden_chan": 16,
  "resnet_kernel_size": 7,
  "resnet_padding": 3
}