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