Add model configuration

config.json

@@ -0,0 +1,148 @@
+{
+  "architectures": [
+    "SAPINN"
+  ],
+  "framework": "pytorch",
+  "model_name": "SA-PINNs",
+  "model_type": "sa_pinns",
+  "source_config": "conf/config.yaml",
+  "root": {
+    "common": {
+      "device": "auto",
+      "dtype": "float64",
+      "seed": 42,
+      "weight_dir": "weight",
+      "result_dir": "result"
+    },
+    "cases": {
+      "laplace": {
+        "model": {
+          "input_dim": 1,
+          "n_layers": 4,
+          "n_neurons": 50,
+          "activation": "tanh"
+        },
+        "attention": {
+          "enabled": true,
+          "lr": 0.01
+        },
+        "training": {
+          "epochs": 10000,
+          "lr": 0.001,
+          "lbfgs_iters": 2000,
+          "log_interval": 1000
+        },
+        "loss": {
+          "data": 10.0,
+          "pde": 1.0,
+          "boundary": 10.0
+        },
+        "data": {
+          "domain": [
+            0.0,
+            1.0
+          ],
+          "n_sol": 200,
+          "n_pde": 1000,
+          "n_bnd": 2,
+          "noise_std": 0.0,
+          "test_res": 200
+        },
+        "output": {
+          "checkpoint_name": "sapinn_laplace.pt",
+          "predictions_name": "sapinn_laplace.npz",
+          "figure_name": "sapinn_laplace.png",
+          "history_name": "sapinn_laplace_history.npz"
+        }
+      },
+      "helmholtz": {
+        "model": {
+          "input_dim": 2,
+          "n_layers": 4,
+          "n_neurons": 50,
+          "activation": "tanh"
+        },
+        "attention": {
+          "enabled": true,
+          "lr": 0.001
+        },
+        "training": {
+          "epochs": 10000,
+          "lr": 0.001,
+          "lbfgs_iters": 5000,
+          "log_interval": 1000
+        },
+        "loss": {
+          "data": 0.0,
+          "pde": 1.0,
+          "boundary": 10.0
+        },
+        "data": {
+          "domain_x": [
+            -1.0,
+            1.0
+          ],
+          "domain_y": [
+            -1.0,
+            1.0
+          ],
+          "n_sol": 200,
+          "n_pde": 10000,
+          "n_bnd": 400,
+          "test_res": 100,
+          "wave_number": 1.0
+        },
+        "output": {
+          "checkpoint_name": "sapinn_helmholtz.pt",
+          "predictions_name": "sapinn_helmholtz.npz",
+          "figure_name": "sapinn_helmholtz.png",
+          "history_name": "sapinn_helmholtz_history.npz"
+        }
+      },
+      "burgers": {
+        "model": {
+          "input_dim": 2,
+          "n_layers": 8,
+          "n_neurons": 20,
+          "activation": "tanh"
+        },
+        "attention": {
+          "enabled": true,
+          "lr": 0.005
+        },
+        "training": {
+          "epochs": 100,
+          "lr": 0.005,
+          "lbfgs_iters": 5000,
+          "log_interval": 10
+        },
+        "loss": {
+          "data": 10.0,
+          "pde": 1.0,
+          "boundary": 10.0
+        },
+        "data": {
+          "domain_x": [
+            -1.0,
+            1.0
+          ],
+          "domain_t": [
+            0.0,
+            1.0
+          ],
+          "n_sol": 100,
+          "n_pde": 10000,
+          "n_bnd": 50,
+          "test_res": 100,
+          "viscosity": 0.003183098861837907
+        },
+        "output": {
+          "checkpoint_name": "sapinn_burgers.pt",
+          "predictions_name": "sapinn_burgers.npz",
+          "figure_name": "sapinn_burgers.png",
+          "history_name": "sapinn_burgers_history.npz"
+        }
+      }
+    }
+  }
+}