第一次完整测例跑完

This commit is contained in:
2026-01-18 00:30:10 +08:00
parent ca15cc593b
commit 25c6fc04db
180 changed files with 29305 additions and 0 deletions

View File

@@ -0,0 +1,34 @@
res_dir="unitree_g1_pack_camera/case1"
dataset="unitree_g1_pack_camera"
{
time CUDA_VISIBLE_DEVICES=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" \
--bs 1 --height 320 --width 512 \
--unconditional_guidance_scale 1.0 \
--ddim_steps 50 \
--ddim_eta 1.0 \
--prompt_dir "unitree_g1_pack_camera/case1/world_model_interaction_prompts" \
--dataset ${dataset} \
--video_length 16 \
--frame_stride 6 \
--n_action_steps 16 \
--exe_steps 16 \
--n_iter 11 \
--timestep_spacing 'uniform_trailing' \
--guidance_rescale 0.7 \
--perframe_ae \
--profile \
--profile_iterations 3
} 2>&1 | tee "${res_dir}/output_profile.log"
echo ""
echo "========================================"
echo "Profiling results saved to: ${res_dir}/output/profile_output/"
echo " - profiling_report.txt: Human-readable summary"
echo " - profiling_data.json: Detailed data for analysis"
echo " - trace_iter_*.json: Chrome trace files (open in chrome://tracing)"
echo "========================================"