Download code/scripts/eval_5k_demo.sh from teawhite/ActionRoPE: direct link, hf CLI and curl.
- Browser
- Download file 1.3 kB
-
https://huggingface.co/teawhite/ActionRoPE/resolve/main/code/scripts/eval_5k_demo.sh
- Command line
-
hf download hf://teawhite/ActionRoPE/code/scripts/eval_5k_demo.sh
-
curl -L -o eval_5k_demo.sh https://huggingface.co/teawhite/ActionRoPE/resolve/main/code/scripts/eval_5k_demo.sh
1.3 kB
| # 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 | |