# Standalone script, use scripts/rlbench/hiveformer_datagen.sh
# to generate and package as well
DATA_PATH=hiveformer_raw/

seed=0
variation=0
variation_count=1
tasks=(
    close_door
)

num_tasks=${#tasks[@]}
for ((i=0; i<$num_tasks; i++)); do
     xvfb-run -a python data_generation/generate.py \
          --save_path ${DATA_PATH} \
          --image_size 256,256 --renderer opengl \
          --episodes_per_task 100 \
          --tasks ${tasks[$i]} --variations ${variation_count} --offset ${variation} \
          --processes 1 --seed ${seed}
done