olivame/unifolm-world-model-action
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

脚本参数暂时修改

Browse Source
This commit is contained in:
qhy
2026-02-07 21:28:54 +08:00
parent d5f6577fa8
commit bc78815acf
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
unitree_g1_pack_camera/case1/run_world_model_interaction.sh
View File

@@ -2,7 +2,7 @@ res_dir="unitree_g1_pack_camera/case1"
dataset="unitree_g1_pack_camera" 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 \ --seed 123 \
--ckpt_path ckpts/unifolm_wma_dual_mixbf16.ckpt \ --ckpt_path ckpts/unifolm_wma_dual_mixbf16.ckpt \
--config configs/inference/world_model_interaction.yaml \ --config configs/inference/world_model_interaction.yaml \
@@ -23,5 +23,7 @@ dataset="unitree_g1_pack_camera"
--perframe_ae \ --perframe_ae \
--diffusion_dtype bf16 \ --diffusion_dtype bf16 \
--projector_mode bf16_full \ --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" } 2>&1 | tee "${res_dir}/output.log"