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_all_full.py
+++ b/example/run_all_full.py
@@ -67,7 +67,7 @@ def main():
    clip_k = cfg.get("clip_k", 5.0)

    if not args.skip_prepare:
-        run([sys.executable, str(base_dir / "prepare_data.py")])
+        run([sys.executable, str(base_dir / "prepare_data.py"), "--config", str(config_path)])
    if not args.skip_train:
        run([sys.executable, str(base_dir / "train.py"), "--config", str(config_path), "--device", args.device])
    if not args.skip_export: