Add model configuration

config.json

@@ -0,0 +1,185 @@
+{
+  "architectures": [
+    "MPPDESolver"
+  ],
+  "framework": "pytorch",
+  "model_name": "MP_PDE",
+  "model_type": "mp_pde",
+  "source_config": "config/config.yaml",
+  "experiment": {
+    "name": "E3",
+    "model": "MP-PDE",
+    "paper": "https://arxiv.org/abs/2202.03376",
+    "evidence_priority": "paper_then_official_repo_for_missing_details",
+    "ambiguity_policy": "paper_strict"
+  },
+  "paths": {
+    "data": "data/e3.h5",
+    "checkpoint": "weight/best_model.pth",
+    "results": "results",
+    "train_history": "results/train_history.json",
+    "predictions": "results/predictions.npz",
+    "metrics": "results/metrics.json"
+  },
+  "data": {
+    "domain_length": 16.0,
+    "final_time": 4.0,
+    "num_time_points": 250,
+    "high_resolution_nx": 200,
+    "resolution": 100,
+    "supported_resolutions": [
+      100,
+      50,
+      40
+    ],
+    "sample_fraction": 0.3333333333333333,
+    "full_experiment_samples": {
+      "train": 2096,
+      "valid": 128,
+      "test": 128
+    },
+    "train_samples": 698,
+    "valid_samples": 43,
+    "test_samples": 43,
+    "seed": 0,
+    "equation": {
+      "alpha_range": [
+        0.0,
+        3.0
+      ],
+      "beta_range": [
+        0.0,
+        0.4
+      ],
+      "gamma_range": [
+        0.0,
+        1.0
+      ]
+    },
+    "forcing": {
+      "terms": 5,
+      "amplitude_range": [
+        -0.5,
+        0.5
+      ],
+      "paper_omega_range": [
+        -0.4,
+        -0.4
+      ],
+      "official_consistency_omega_range": [
+        -0.4,
+        0.4
+      ],
+      "modes": [
+        1,
+        2,
+        3
+      ],
+      "phase_range": [
+        0.0,
+        6.283185307179586
+      ]
+    },
+    "generation": {
+      "cfl": 0.2,
+      "min_dt": 1e-08,
+      "max_substeps": 10000,
+      "weno_epsilon": 1e-06,
+      "compression": "gzip",
+      "compression_level": 4
+    },
+    "parallel_generation": {
+      "workers": 8,
+      "max_in_flight": 16,
+      "flush_every": 8,
+      "resume_partial": true
+    }
+  },
+  "model": {
+    "time_window": 25,
+    "hidden_dim": 164,
+    "message_passing_layers": 6,
+    "neighbor_offsets": [
+      -3,
+      -2,
+      -1,
+      1,
+      2,
+      3
+    ],
+    "aggregation": "sum",
+    "instance_norm_affine": false,
+    "decoder": {
+      "middle_channels": 8,
+      "kernel_sizes": [
+        16,
+        26
+      ],
+      "strides": [
+        3,
+        1
+      ]
+    },
+    "scaling": {
+      "coordinates": true,
+      "parameters": true,
+      "solution": "none"
+    }
+  },
+  "training": {
+    "epochs": 20,
+    "batch_size": 16,
+    "num_workers": 0,
+    "optimizer": "AdamW",
+    "learning_rate": 0.0001,
+    "weight_decay": 1e-08,
+    "loss": "rmse",
+    "loss_epsilon": 1e-12,
+    "max_no_grad_pushes": 1,
+    "scheduler": {
+      "enabled": true,
+      "milestones": [
+        1,
+        5,
+        10,
+        15
+      ],
+      "gamma": 0.4
+    },
+    "precision": "float32",
+    "amp": false,
+    "gradient_clip": null,
+    "seed": 0,
+    "deterministic": true,
+    "device": "auto",
+    "print_interval": 10,
+    "validation_rollout_samples": null,
+    "test_rollout_samples": null
+  },
+  "inference": {
+    "batch_size": 16,
+    "split": "test",
+    "max_samples": null,
+    "device": "auto",
+    "forecast_start_index": 25
+  },
+  "visualization": {
+    "sample_index": 0,
+    "time_indices": [
+      25,
+      75,
+      150,
+      249
+    ],
+    "dpi": 160
+  },
+  "paper_reference": {
+    "metric": "forecast_accumulated_mse",
+    "values": {
+      "100": 4.26,
+      "50": 3.74,
+      "40": 3.7
+    },
+    "note": "provenance_only_not_an_assertion"
+  }
+}