File size: 497 Bytes
0453c63 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | #!/bin/bash
# Batch generate affordance masks for all four LIBERO subsets sequentially.
SRC_ROOT="/gemini/space/wrz/libero_per_frame"
TGT_ROOT="/gemini/space/wrz/ragnet_results"
for ds in libero_object libero_goal libero_spatial libero_10; do
echo "========== Processing ${ds} =========="
CUDA_VISIBLE_DEVICES=0 python batch_generate.py \
--data_dir "${SRC_ROOT}/${ds}_converted" \
--save_dir "${TGT_ROOT}/${ds}"
echo "========== ${ds} done =========="
echo
done
|