upload real-robot deployment code
This commit is contained in:
26
scripts/run_real_eval_server.sh
Normal file
26
scripts/run_real_eval_server.sh
Normal file
@@ -0,0 +1,26 @@
|
||||
model_name=testing
|
||||
ckpt=/path/to/model/checkpoint
|
||||
config=configs/inference/world_model_decision_making.yaml
|
||||
seed=123
|
||||
res_dir="path/to/results/directory"
|
||||
datasets=(
|
||||
"unitree_g1_pack_camera"
|
||||
)
|
||||
|
||||
|
||||
for dataset in "${datasets[@]}"; do
|
||||
CUDA_VISIBLE_DEVICES=0 python3 scripts/evaluation/real_eval_server.py \
|
||||
--seed ${seed} \
|
||||
--ckpt_path $ckpt \
|
||||
--config $config \
|
||||
--savedir "${res_dir}/${dataset}/${model_name}/videos" \
|
||||
--bs 1 --height 320 --width 512 \
|
||||
--unconditional_guidance_scale 1.0 \
|
||||
--ddim_steps 16 \
|
||||
--ddim_eta 1.0 \
|
||||
--video_length 16 \
|
||||
--frame_stride 2 \
|
||||
--timestep_spacing 'uniform_trailing' \
|
||||
--guidance_rescale 0.7 \
|
||||
--perframe_ae
|
||||
done
|
||||
Reference in New Issue
Block a user