File size: 1,564 Bytes
ff66cf3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
#!/bin/bash
DATA_DIR=$1
DISP=False
echo "Generating dataset... Folder: $DATA_DIR"
# You can parallelize these depending on how much resources you have
#############################
## Language-Conditioned Tasks
LANG_TASKS=$2
for task in $LANG_TASKS
do
python cliport/demos.py n=100 task=$task mode=train data_dir=$DATA_DIR disp=$DISP &
python cliport/demos.py n=100 task=$task mode=val data_dir=$DATA_DIR disp=$DISP &
python cliport/demos.py n=100 task=$task mode=test data_dir=$DATA_DIR disp=$DISP
done
echo "Finished Language Tasks."
#########################
## Demo-Conditioned Tasks
# LANG_TASKS='align-rope assembling-kits-seq-seen-colors assembling-kits-seq-unseen-colors packing-shapes packing-boxes-pairs-seen-colors packing-boxes-pairs-unseen-colors packing-seen-google-objects-seq packing-unseen-google-objects-seq packing-seen-google-objects-group packing-unseen-google-objects-group put-block-in-bowl-seen-colors put-block-in-bowl-unseen-colors stack-block-pyramid-seq-seen-colors stack-block-pyramid-seq-unseen-colors separating-piles-seen-colors separating-piles-unseen-colors towers-of-hanoi-seq-seen-colors towers-of-hanoi-seq-unseen-colors'
DEMO_TASKS=$2
for task in $DEMO_TASKS
do
python cliport/demos.py n=100 task=$task mode=train data_dir=$DATA_DIR disp=$DISP &
python cliport/demos.py n=100 task=$task mode=val data_dir=$DATA_DIR disp=$DISP &
python cliport/demos.py n=100 task=$task mode=test data_dir=$DATA_DIR disp=$DISP
done
echo "Finished Demo Tasks."
|