File size: 939 Bytes
20c251e | 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 | #!/bin/bash
#SBATCH --job-name=dovla_ms_multi_rgb
#SBATCH --account=def-yalda_cpu
#SBATCH --partition=cpubase_bycore_b1
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=8
#SBATCH --mem=24G
#SBATCH --time=03:00:00
#SBATCH --array=0-4%2
#SBATCH --output=outputs/hpc/logs/%x_%A_%a.out
#SBATCH --error=outputs/hpc/logs/%x_%A_%a.err
set -euo pipefail
PROJECT_DIR="${PROJECT_DIR:-$SLURM_SUBMIT_DIR}"
MULTITASK_OUT_ROOT="${MULTITASK_OUT_ROOT:?Set MULTITASK_OUT_ROOT}"
case "${SLURM_ARRAY_TASK_ID:-0}" in
0) ENV_ID="PushCube-v1" ;;
1) ENV_ID="PullCube-v1" ;;
2) ENV_ID="StackCube-v1" ;;
3) ENV_ID="LiftPegUpright-v1" ;;
4) ENV_ID="PegInsertionSide-v1" ;;
*) echo "unsupported array index" >&2; exit 2 ;;
esac
export PROJECT_DIR
export DATASET="$MULTITASK_OUT_ROOT/$ENV_ID"
export IMAGE_QUALITY="${IMAGE_QUALITY:-85}"
export SEED="${SEED:-0}"
exec bash "$PROJECT_DIR/scripts/slurm/render_maniskill_observations.sbatch"
|