update

example/run_pipeline.py

@@ -5,6 +5,7 @@ import argparse
 import subprocess
 import sys
 from pathlib import Path
+import json
 
 from platform_utils import safe_path, is_windows
 
@@ -40,6 +41,13 @@ def parse_args():
 def main():
     args = parse_args()
     base_dir = Path(__file__).resolve().parent
+    config_path = Path(args.config)
+    with open(config_path, "r", encoding="utf-8") as f:
+        cfg = json.load(f)
+    timesteps = cfg.get("timesteps", 200)
+    seq_len = cfg.get("sample_seq_len", cfg.get("seq_len", 64))
+    batch_size = cfg.get("sample_batch_size", cfg.get("batch_size", 2))
+    clip_k = cfg.get("clip_k", 5.0)
     run([sys.executable, str(base_dir / "prepare_data.py")])
     run([sys.executable, str(base_dir / "train.py"), "--config", args.config, "--device", args.device])
     run(
@@ -49,6 +57,17 @@ def main():
             "--include-time",
             "--device",
             args.device,
+            "--config",
+            str(config_path),
+            "--timesteps",
+            str(timesteps),
+            "--seq-len",
+            str(seq_len),
+            "--batch-size",
+            str(batch_size),
+            "--clip-k",
+            str(clip_k),
+            "--use-ema",
         ]
     )
     run([sys.executable, str(base_dir / "evaluate_generated.py")])