Add model configuration

config.json

@@ -0,0 +1,69 @@
+{
+  "architectures": [
+    "ONO"
+  ],
+  "framework": "pytorch",
+  "model_name": "ONO",
+  "model_type": "ono",
+  "source_config": "conf/config.yaml",
+  "root": {
+    "common": {
+      "device": "auto",
+      "seed": 0
+    },
+    "datapipe": {
+      "source": {
+        "data_dir": "/public/share/sugonhpcapp01/onestore/onedatasets/CFD_Benchmark/ns",
+        "file_name": "NavierStokes_V1e-5_N1200_T20.mat"
+      },
+      "data": {
+        "ntrain": 1000,
+        "ntest": 200,
+        "t_in": 10,
+        "t_out": 10,
+        "out_dim": 1,
+        "downsamplex": 1,
+        "downsampley": 1,
+        "normalize": false
+      },
+      "dataloader": {
+        "batch_size": 8,
+        "num_workers": 4,
+        "pin_memory": true
+      }
+    },
+    "model": {
+      "in_dim": 10,
+      "out_dim": 1,
+      "hidden_dim": 64,
+      "num_layers": 4,
+      "num_heads": 4,
+      "space_dim": 2,
+      "include_pos": true,
+      "dropout": 0.0,
+      "activation": "gelu",
+      "mlp_ratio": 1,
+      "attn_type": "nystrom",
+      "psi_dim": 8
+    },
+    "training": {
+      "epochs": 500,
+      "lr": 0.001,
+      "weight_decay": 1e-05,
+      "step_size": 100,
+      "gamma": 0.5,
+      "eval_interval": 1,
+      "patience": 100,
+      "teacher_forcing": false,
+      "max_grad_norm": 0.1,
+      "max_train_batches": null,
+      "max_eval_batches": null,
+      "weight_dir": "weight",
+      "checkpoint_name": "ono_navier_stokes.pt"
+    },
+    "inference": {
+      "result_dir": "result",
+      "num_samples": 4
+    }
+  }
+}