第一次完整测例跑完
This commit is contained in:
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -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 "========================================"
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Reference in New Issue
Block a user