调整配置启动视频写入开关

2026-04-10 03:44:09 +00:00
parent 46cb2177bc
commit 75a5d86966
6 changed files with 67 additions and 3 deletions
--- a/eval.py
+++ b/eval.py
@@ -370,7 +370,7 @@ def run_eval_subset(
                        callables=OmegaConf.to_container(
                            local_cfg.eval.get("callables"), resolve=True
                        ),
-                        save_video=False,
+                        save_video=bool(local_cfg.eval.get("save_video", False)),
                        video_path=output_dir,
                    )
    if str(device).startswith("cuda") and torch.cuda.is_available():