脚本参数暂时修改
This commit is contained in:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user