anhtld commited on
Commit
ac75a98
·
verified ·
1 Parent(s): 7406f6f

sync rgb render slurm wrapper 2026-07-03

Browse files
workspace/scripts/slurm/render_six_task_chart_observations.sbatch ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ #SBATCH --job-name=dovla_six_rgb
3
+ #SBATCH --account=def-yalda_cpu
4
+ #SBATCH --partition=cpubase_bycore_b1
5
+ #SBATCH --nodes=1
6
+ #SBATCH --ntasks=1
7
+ #SBATCH --cpus-per-task=8
8
+ #SBATCH --mem=24G
9
+ #SBATCH --time=03:00:00
10
+ #SBATCH --array=0-4%2
11
+ #SBATCH --output=outputs/hpc/logs/%x_%A_%a.out
12
+ #SBATCH --error=outputs/hpc/logs/%x_%A_%a.err
13
+
14
+ set -euo pipefail
15
+
16
+ PROJECT_DIR="${PROJECT_DIR:-$SLURM_SUBMIT_DIR}"
17
+ DATASET_ROOT="${DATASET_ROOT:-/scratch/$USER/dovla/experiments/six_task_h16_collection}"
18
+
19
+ TASKS=(
20
+ "LiftPegUpright-v1"
21
+ "PickCube-v1"
22
+ "PullCube-v1"
23
+ "PushCube-v1"
24
+ "StackCube-v1"
25
+ )
26
+
27
+ TASK_INDEX="${SLURM_ARRAY_TASK_ID:-0}"
28
+ if [[ "$TASK_INDEX" -lt 0 || "$TASK_INDEX" -ge "${#TASKS[@]}" ]]; then
29
+ echo "unsupported array index: $TASK_INDEX" >&2
30
+ exit 2
31
+ fi
32
+
33
+ export PROJECT_DIR
34
+ export DATASET="$DATASET_ROOT/${TASKS[$TASK_INDEX]}"
35
+ export IMAGE_QUALITY="${IMAGE_QUALITY:-85}"
36
+ export SEED="${SEED:-0}"
37
+ export OVERWRITE="${OVERWRITE:-0}"
38
+
39
+ exec bash "$PROJECT_DIR/scripts/slurm/render_maniskill_observations.sbatch"