早停特征验证，早停不通

scripts/run_world_model_interaction.sh

@@ -3,6 +3,8 @@ ckpt=/path/to/model/checkpoint
 config=configs/inference/world_model_interaction.yaml
 seed=123
 res_dir="/path/to/result/directory"
+analysis_reference_steps=50
+# analysis_psnr_path="/path/to/psnr_full50.csv"
 
 datasets=(
     "unitree_z1_stackbox"
@@ -19,6 +21,13 @@ for i in "${!datasets[@]}"; do
     dataset=${datasets[$i]}
     n_iter=${n_iters[$i]}
     fs=${fses[$i]}
+    analysis_args=(
+        --analysis_log_metrics
+        --analysis_reference_steps ${analysis_reference_steps}
+    )
+    if [ -n "${analysis_psnr_path:-}" ]; then
+        analysis_args+=(--analysis_psnr_path "${analysis_psnr_path}")
+    fi
 
     CUDA_VISIBLE_DEVICES=0 python3 scripts/evaluation/world_model_interaction.py \
     --seed ${seed} \
@@ -38,5 +47,6 @@ for i in "${!datasets[@]}"; do
     --n_iter ${n_iter} \
     --timestep_spacing 'uniform_trailing' \
     --guidance_rescale 0.7 \
-    --perframe_ae
+    --perframe_ae \
+    "${analysis_args[@]}"
 done