| { | |
| "model_defaults": { | |
| "architecture": "deeponet", | |
| "activation": "relu", | |
| "initializer": "xavier_normal", | |
| "dense_bias": true, | |
| "branch_depth": 2, | |
| "trunk_depth": 3, | |
| "width": 40, | |
| "latent_dim": 40 | |
| }, | |
| "variants": { | |
| "unstacked_bias": { | |
| "architecture": "deeponet", | |
| "stacked": false, | |
| "branch_output_bias": true, | |
| "global_bias": true | |
| }, | |
| "unstacked_no_bias": { | |
| "architecture": "deeponet", | |
| "stacked": false, | |
| "branch_output_bias": false, | |
| "global_bias": false | |
| }, | |
| "stacked_bias": { | |
| "architecture": "deeponet", | |
| "stacked": true, | |
| "branch_output_bias": true, | |
| "global_bias": true | |
| }, | |
| "stacked_no_bias": { | |
| "architecture": "deeponet", | |
| "stacked": true, | |
| "branch_output_bias": false, | |
| "global_bias": false | |
| }, | |
| "fnn": { | |
| "architecture": "fnn", | |
| "depth": 3, | |
| "width": 40, | |
| "output_bias": true | |
| } | |
| }, | |
| "experiments": { | |
| "antiderivative": { | |
| "sensor_points": 100, | |
| "trunk_dim": 1, | |
| "default_variant": "unstacked_bias" | |
| }, | |
| "nonlinear_ode": { | |
| "sensor_points": 100, | |
| "trunk_dim": 1, | |
| "default_variant": "unstacked_bias" | |
| }, | |
| "pendulum": { | |
| "sensor_points": 100, | |
| "trunk_dim": 1, | |
| "default_variant": "unstacked_bias" | |
| }, | |
| "diffusion_reaction": { | |
| "sensor_points": 100, | |
| "trunk_dim": 2, | |
| "default_variant": "unstacked_bias", | |
| "branch_depth": 2, | |
| "trunk_depth": 3, | |
| "width": 100, | |
| "latent_dim": 100 | |
| } | |
| } | |
| } | |