脚本参数暂时修改

This commit is contained in:
qhy
2026-02-07 21:28:54 +08:00
parent d5f6577fa8
commit bc78815acf

View File

@@ -2,7 +2,7 @@ res_dir="unitree_g1_pack_camera/case1"
dataset="unitree_g1_pack_camera"
{
time CUDA_VISIBLE_DEVICES=1 python3 scripts/evaluation/world_model_interaction.py \
time CUDA_VISIBLE_DEVICES=0 python3 scripts/evaluation/world_model_interaction.py \
--seed 123 \
--ckpt_path ckpts/unifolm_wma_dual_mixbf16.ckpt \
--config configs/inference/world_model_interaction.yaml \
@@ -23,5 +23,7 @@ dataset="unitree_g1_pack_camera"
--perframe_ae \
--diffusion_dtype bf16 \
--projector_mode bf16_full \
--encoder_mode bf16_full
--encoder_mode bf16_full \
--vae_dtype bf16 \
--vae_channels_last
} 2>&1 | tee "${res_dir}/output.log"