anhtld commited on
Commit
6fa9659
·
verified ·
1 Parent(s): ac75a98

sync rgb ref export slurm wrapper 2026-07-03

Browse files
workspace/scripts/slurm/reexport_rgb_ref_cil_charts.sbatch ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ #SBATCH --job-name=cil_rgb_ref_export
3
+ #SBATCH --account=def-yalda
4
+ #SBATCH --nodes=1
5
+ #SBATCH --ntasks=1
6
+ #SBATCH --cpus-per-task=2
7
+ #SBATCH --mem=8G
8
+ #SBATCH --time=01:00:00
9
+ #SBATCH --output=outputs/hpc/logs/%x_%j.out
10
+ #SBATCH --error=outputs/hpc/logs/%x_%j.err
11
+
12
+ set -euo pipefail
13
+
14
+ PROJECT_DIR="${PROJECT_DIR:-$SLURM_SUBMIT_DIR}"
15
+ PYTHON="${PYTHON:-$PROJECT_DIR/.venv/bin/python}"
16
+ DATASET="${DATASET:-/scratch/$USER/dovla/experiments/six_task_h16_collection}"
17
+ OUT_ROOT="${OUT_ROOT:-data/cil_charts_rgb_refs}"
18
+ SPLIT_FRACTIONS="${SPLIT_FRACTIONS:-0.70,0.15,0.15}"
19
+ SPLIT_SEED="${SPLIT_SEED:-0}"
20
+ EPSILON="${EPSILON:-0.05}"
21
+ BASE_CANDIDATE_TYPES="${BASE_CANDIDATE_TYPES:-policy,policy_residual,anchor,base,expert}"
22
+
23
+ cd "$PROJECT_DIR"
24
+ mkdir -p outputs/hpc/logs "$OUT_ROOT" runs/leakage_audit_rgb_refs runs/chart_feature_audit_rgb_refs
25
+
26
+ common_args=(
27
+ --dataset "$DATASET"
28
+ --split-policy stable-hash
29
+ --split-fractions "$SPLIT_FRACTIONS"
30
+ --split-seed "$SPLIT_SEED"
31
+ --epsilon "$EPSILON"
32
+ --base-candidate-types "$BASE_CANDIDATE_TYPES"
33
+ )
34
+
35
+ "$PYTHON" scripts/export_cil_charts.py \
36
+ "${common_args[@]}" \
37
+ --split train \
38
+ --out-dir "$OUT_ROOT/train"
39
+
40
+ "$PYTHON" scripts/export_cil_charts.py \
41
+ "${common_args[@]}" \
42
+ --split val \
43
+ --include-outcomes \
44
+ --out-dir "$OUT_ROOT/val"
45
+
46
+ "$PYTHON" scripts/export_cil_charts.py \
47
+ "${common_args[@]}" \
48
+ --split test \
49
+ --include-outcomes \
50
+ --out-dir "$OUT_ROOT/test"
51
+
52
+ "$PYTHON" scripts/audit_cil_charts.py \
53
+ --chart-root "$OUT_ROOT" \
54
+ --out-dir runs/leakage_audit_rgb_refs
55
+
56
+ "$PYTHON" scripts/audit_chart_feature_sources.py \
57
+ --indexes "$OUT_ROOT/train/index.json" "$OUT_ROOT/val/index.json" "$OUT_ROOT/test/index.json" \
58
+ --out-dir runs/chart_feature_audit_rgb_refs