Add model configuration

config.json

@@ -0,0 +1,75 @@
+{
+  "architectures": [
+    "FactFormer2D"
+  ],
+  "framework": "pytorch",
+  "model_name": "FactFormer",
+  "model_type": "factformer",
+  "source_config": "conf/config.yaml",
+  "root": {
+    "common": {
+      "device": "auto",
+      "seed": 1234
+    },
+    "datapipe": {
+      "source": {
+        "data_dir": "/public/share/sugonhpcapp01/onestore/onedatasets/Kolmogorov_flow_2d",
+        "file_name": "kf_2d_re1000_256_120seed.npy"
+      },
+      "data": {
+        "train_num": 100,
+        "test_num": 20,
+        "resolution": 128,
+        "interval": 2,
+        "t_in": 10,
+        "t_out": 16,
+        "out_dim": 1,
+        "normalize": true,
+        "val_mode": false,
+        "split_seed": 1234,
+        "stats_file": "data/km2d_stat.npz",
+        "stats_max_trajectories": 8
+      },
+      "dataloader": {
+        "batch_size": 4,
+        "num_workers": 2,
+        "pin_memory": true
+      }
+    },
+    "model": {
+      "in_dim": 10,
+      "out_dim": 1,
+      "hidden_dim": 128,
+      "depth": 4,
+      "heads": 8,
+      "mlp_ratio": 2,
+      "dropout": 0.0,
+      "activation": "gelu",
+      "include_pos": true,
+      "space_dim": 2,
+      "latent_multiplier": 2.0,
+      "max_latent_steps": 4
+    },
+    "training": {
+      "epochs": 50,
+      "lr": 0.0005,
+      "weight_decay": 0.0001,
+      "step_size": 20,
+      "gamma": 0.5,
+      "eval_interval": 1,
+      "patience": 20,
+      "train_latent_steps": 4,
+      "max_grad_norm": 1.0,
+      "max_train_batches": null,
+      "max_eval_batches": null,
+      "weight_dir": "weight",
+      "checkpoint_name": "factformer_kolmogorov.pt"
+    },
+    "inference": {
+      "result_dir": "result",
+      "num_samples": 4,
+      "figure_name": "prediction_sample.png",
+      "tensor_name": "prediction_sample.pt"
+    }
+  }
+}