ActionRoPE / code /scripts /eval_5k_demo.sh
teawhite's picture
add docs+code
880dff9 verified
Raw History Blame Contribute Delete
1.3 kB
#!/usr/bin/env bash
# step-5000 效果演示:同一首帧,一组自造指令 + replay,8 卡并行各跑一条(50 步)。
set -euo pipefail
ROOT=/opt/dlami/nvme/zhiyangdeng/ActionRoPE
export DIFFSYNTH_SKIP_DOWNLOAD=True PYTHONPATH=$ROOT
CK=${CK:-$ROOT/outputs/arope_5k/step-5000.safetensors}
OUT=${OUT:-$ROOT/outputs/samples/arope_5k}; mkdir -p "$OUT"
FF=clip_Eybx_200000958_000273; SCENE=tidal_flats
run() { local gpu=$1 name=$2; shift 2
CUDA_VISIBLE_DEVICES=$gpu "$ROOT/.venv/bin/python" -m actionrope.infer --ckpt "$CK" --steps 50 --out "$OUT/$name.mp4" "$@" \
> "$OUT/$name.log" 2>&1 && grep -E "实测位移|PSNR" "$OUT/$name.log" | sed "s/^/[$name] /" || echo "[$name] FAILED"; }
run 0 replay --replay $FF &
run 1 right_x1 --first_frame $FF --scene $SCENE --actions "right:1.0:21" &
run 2 left_x1 --first_frame $FF --scene $SCENE --actions "left:1.0:21" &
run 3 up_x1 --first_frame $FF --scene $SCENE --actions "up:1.0:21" &
run 4 down_x1 --first_frame $FF --scene $SCENE --actions "down:1.0:21" &
run 5 right_x0.5 --first_frame $FF --scene $SCENE --actions "right:0.5:21" &
run 6 right_x1.5 --first_frame $FF --scene $SCENE --actions "right:1.5:21" &
run 7 there_back --first_frame $FF --scene $SCENE --actions "right:1.0:10,left:1.0:11" &
wait
echo DONE