Add comprehensive evaluation and ablation runner

2026-03-25 22:20:43 +08:00
parent f1afd4bf38
commit 957b010ea1
8 changed files with 1730 additions and 30 deletions
--- a/example/run_pipeline.py
+++ b/example/run_pipeline.py
@@ -50,7 +50,7 @@ def main():
    clip_k = cfg.get("clip_k", 5.0)
    data_glob = cfg.get("data_glob", "")
    data_path = cfg.get("data_path", "")
-    run([sys.executable, str(base_dir / "prepare_data.py")])
+    run([sys.executable, str(base_dir / "prepare_data.py"), "--config", str(config_path)])
    run([sys.executable, str(base_dir / "train.py"), "--config", args.config, "--device", args.device])
    run(
        [