多卡流水导出

unitree_z1_dual_arm_stackbox_v2/case2/run_world_model_interaction.sh

@@ -2,11 +2,11 @@ res_dir="unitree_z1_dual_arm_stackbox_v2/case2"
 dataset="unitree_z1_dual_arm_stackbox_v2"
 
 {
-    time CUDA_VISIBLE_DEVICES=0 "${PYTHON_BIN:-python}" scripts/evaluation/world_model_interaction.py \
+    time CUDA_VISIBLE_DEVICES=0,1 python3 scripts/evaluation/world_model_interaction.py \
         --seed 123 \
         --ckpt_path ckpts/unifolm_wma_dual.ckpt \
         --config configs/inference/world_model_interaction.yaml \
-        --savedir "${res_dir}/output/sparse_8" \
+        --savedir "${res_dir}/output" \
         --bs 1 --height 320 --width 512 \
         --unconditional_guidance_scale 1.0 \
         --ddim_steps 50 \
@@ -21,9 +21,7 @@ dataset="unitree_z1_dual_arm_stackbox_v2"
         --timestep_spacing 'uniform_trailing' \
         --guidance_rescale 0.7 \
         --perframe_ae \
-        --analysis_log_metrics \
-        --analysis_reference_steps 50 \
-        --head_schedule_steps 0 7 14 21 28 35 42 49 \
-        --head_skip_mode reuse_prediction \
-        --head_log_steps 40 43 46 47 48 49
+        --pipeline_split_step 30 \
+        --pipeline_multi_gpu \
+        --pipeline_gpu_ids 0 1
 } 2>&1 | tee "${res_dir}/output.log"