Update model configuration

config.json

@@ -1,42 +1,4 @@
 {
-  "architectures": [
-    "DeepONet"
-  ],
-  "framework": "pytorch",
-  "model_name": "DeepONet",
-  "model_type": "deeponet",
-  "source_config": "config/config.yaml",
-  "project": {
-    "name": "DeepONet",
-    "paper": "DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operators",
-    "arxiv": "1910.03193v3",
-    "paper_url": "https://arxiv.org/abs/1910.03193",
-    "implementation_policy": "paper_first_independent_reimplementation",
-    "paper_scale": true
-  },
-  "paths": {
-    "checkpoint": "weight/best_model.pth",
-    "results": "results",
-    "cache": "results/cache"
-  },
-  "runtime": {
-    "seed": 0,
-    "device": "auto",
-    "dtype": "float32",
-    "num_workers": 0
-  },
-  "function_space": {
-    "type": "grf",
-    "length_scale": 0.2,
-    "grf_grid_size": 1000,
-    "jitter": 1e-13,
-    "interpolation": "cubic",
-    "generation_chunk_size": 128,
-    "chebyshev": {
-      "degree": 10,
-      "coefficient_bound": 1.0
-    }
-  },
   "model_defaults": {
     "architecture": "deeponet",
     "activation": "relu",
@@ -79,222 +41,30 @@
       "output_bias": true
     }
   },
-  "training_defaults": {
-    "optimizer": "adam",
-    "learning_rate": 0.001,
-    "batch_size": null,
-    "evaluation_batch_size": 8192,
-    "weight_decay": 0.0,
-    "print_every": 1000,
-    "evaluate_every": 1000,
-    "checkpoint_metric": "test_mse",
-    "resume": false
-  },
-  "solver_defaults": {
-    "method": "RK45",
-    "rtol": 1e-07,
-    "atol": 1e-09,
-    "pde_newton_tolerance": 1e-10,
-    "pde_newton_max_iterations": 20
-  },
   "experiments": {
     "antiderivative": {
-      "equation": "ds/dx = u(x), s(0) = 0",
       "sensor_points": 100,
       "trunk_dim": 1,
-      "domain_end": 1.0,
-      "train_size": 10000,
-      "test_size": 100000,
-      "iterations": 50000,
-      "default_variant": "unstacked_bias",
-      "sweeps": {
-        "architectures": [
-          "fnn",
-          "unstacked_no_bias",
-          "unstacked_bias",
-          "stacked_no_bias",
-          "stacked_bias"
-        ],
-        "fnn_depths": [
-          2,
-          3,
-          4
-        ],
-        "fnn_widths": [
-          20,
-          40,
-          100
-        ],
-        "learning_rates": [
-          0.0001,
-          0.001,
-          0.01
-        ]
-      }
+      "default_variant": "unstacked_bias"
     },
     "nonlinear_ode": {
-      "equation": "ds/dx = -s^2 + u(x), s(0) = 0",
       "sensor_points": 100,
       "trunk_dim": 1,
-      "domain_end": 1.0,
-      "train_size": 10000,
-      "test_size": 100000,
-      "iterations": 100000,
-      "default_variant": "unstacked_bias",
-      "trim_fraction": 0.001,
-      "ood_functions": [
-        "linear",
-        "sin_pi",
-        "sin_2pi"
-      ],
-      "sweeps": {
-        "architectures": [
-          "stacked_no_bias",
-          "stacked_bias",
-          "unstacked_no_bias",
-          "unstacked_bias"
-        ]
-      }
+      "default_variant": "unstacked_bias"
     },
     "pendulum": {
-      "equation": "ds1/dt = s2; ds2/dt = -k*sin(s1) + u(t); s(0) = (0,0)",
       "sensor_points": 100,
       "trunk_dim": 1,
-      "domain_end": 1.0,
-      "k": 1.0,
-      "train_size": 10000,
-      "test_size": 100000,
-      "iterations": 100000,
-      "default_variant": "unstacked_bias",
-      "convergence_iterations": 500000,
-      "convergence_test_size": 1000000,
-      "sweeps": {
-        "sensor_points": [
-          10,
-          20,
-          50,
-          100,
-          200
-        ],
-        "domain_end": [
-          1.0,
-          2.0,
-          3.0,
-          4.0
-        ],
-        "length_scale": [
-          0.1,
-          0.2,
-          0.5,
-          1.0
-        ],
-        "k": [
-          0.5,
-          1.0,
-          2.0
-        ],
-        "width": [
-          20,
-          40,
-          100
-        ],
-        "train_size": [
-          1000,
-          5000,
-          10000
-        ],
-        "chebyshev_degree": [
-          5,
-          10,
-          20
-        ]
-      }
+      "default_variant": "unstacked_bias"
     },
     "diffusion_reaction": {
-      "equation": "s_t = D*s_xx + k*s^2 + u(x)",
       "sensor_points": 100,
       "trunk_dim": 2,
-      "domain_end": 1.0,
-      "train_functions": 100,
-      "points_per_function": 1000,
-      "test_functions": 1000,
-      "test_points_per_function": 1000,
-      "test_size": 1000000,
-      "iterations": 500000,
       "default_variant": "unstacked_bias",
-      "diffusion": 0.01,
-      "reaction": 0.01,
-      "space_points": 100,
-      "time_points": 100,
       "branch_depth": 2,
       "trunk_depth": 3,
       "width": 100,
-      "latent_dim": 100,
-      "sweeps": {
-        "train_functions": [
-          50,
-          100,
-          200,
-          400
-        ],
-        "points_per_function": [
-          10,
-          100,
-          1000,
-          10000
-        ]
-      }
-    }
-  },
-  "inference": {
-    "batch_size": 8192,
-    "ood_query_points": 100,
-    "save_inputs": true
-  },
-  "smoke_test": {
-    "project": {
-      "paper_scale": false
-    },
-    "function_space": {
-      "grf_grid_size": 32,
-      "generation_chunk_size": 16
-    },
-    "training_defaults": {
-      "batch_size": 16,
-      "evaluation_batch_size": 32,
-      "print_every": 1,
-      "evaluate_every": 1
-    },
-    "experiments": {
-      "antiderivative": {
-        "sensor_points": 16,
-        "train_size": 32,
-        "test_size": 24,
-        "iterations": 2
-      },
-      "nonlinear_ode": {
-        "sensor_points": 16,
-        "train_size": 24,
-        "test_size": 16,
-        "iterations": 2
-      },
-      "pendulum": {
-        "sensor_points": 16,
-        "train_size": 24,
-        "test_size": 16,
-        "iterations": 2
-      },
-      "diffusion_reaction": {
-        "sensor_points": 16,
-        "train_functions": 4,
-        "points_per_function": 8,
-        "test_functions": 2,
-        "test_points_per_function": 8,
-        "test_size": 16,
-        "space_points": 12,
-        "time_points": 12,
-        "iterations": 2
-      }
+      "latent_dim": 100
     }
   }
 }