anhtld commited on
Commit
d4996e0
·
verified ·
1 Parent(s): ddf773a

auto-sync 2026-07-04T07:17:20Z workspace (part 4)

Browse files
workspace/scripts/slurm/train_ctt_no_cycle_proxy.sbatch ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ #SBATCH --job-name=ctt_no_cycle_proxy
3
+ #SBATCH --account=def-yalda
4
+ #SBATCH --nodes=1
5
+ #SBATCH --ntasks=1
6
+ #SBATCH --cpus-per-task=4
7
+ #SBATCH --mem=16G
8
+ #SBATCH --time=04:00:00
9
+ #SBATCH --array=0-2
10
+ #SBATCH --output=outputs/hpc/logs/%x_%A_%a.out
11
+ #SBATCH --error=outputs/hpc/logs/%x_%A_%a.err
12
+
13
+ set -euo pipefail
14
+
15
+ PROJECT_DIR="${PROJECT_DIR:-$SLURM_SUBMIT_DIR}"
16
+ cd "$PROJECT_DIR"
17
+ mkdir -p outputs/hpc/logs
18
+
19
+ export OMP_NUM_THREADS=1
20
+ export OPENBLAS_NUM_THREADS=1
21
+ export MKL_NUM_THREADS=1
22
+ export DOVLA_TORCH_THREADS=1
23
+ export PYTHONDONTWRITEBYTECODE=1
24
+
25
+ PYTHON="${PYTHON:-$PROJECT_DIR/.venv/bin/python}"
26
+ SEED="${SLURM_ARRAY_TASK_ID:-0}"
27
+ NAME="ctt_residual_no_cycle_full_seed${SEED}"
28
+ TRAIN_OUT="runs/${NAME}"
29
+ VAL_OUT="runs/${NAME}_val_proxy"
30
+
31
+ "$PYTHON" scripts/train_ctt.py \
32
+ --dataset data/cil_charts/train/index.json \
33
+ --out-dir "$TRAIN_OUT" \
34
+ --variant residual \
35
+ --epochs "${CTT_EPOCHS:-5}" \
36
+ --max-charts "${CTT_MAX_CHARTS:-100000}" \
37
+ --neighbors "${CTT_NEIGHBORS:-8}" \
38
+ --lr "${CTT_LR:-0.001}" \
39
+ --seed "$SEED" \
40
+ --pos-alignment "${CTT_POS_ALIGNMENT:-1.0}" \
41
+ --ot-alignment "${CTT_OT_ALIGNMENT:-0.0}" \
42
+ --negative-boundary "${CTT_NEGATIVE_BOUNDARY:-0.25}" \
43
+ --pairwise-rank "${CTT_PAIRWISE_RANK:-1.0}" \
44
+ --listwise-rank "${CTT_LISTWISE_RANK:-0.5}" \
45
+ --cycle 0.0 \
46
+ --diversity "${CTT_DIVERSITY:-0.05}" \
47
+ --negative-margin "${CTT_NEGATIVE_MARGIN:-0.2}" \
48
+ --ot-epsilon "${CTT_OT_EPSILON:-0.05}" \
49
+ --ot-iterations "${CTT_OT_ITERATIONS:-30}" \
50
+ --transport-samples-per-pair "${CTT_TRANSPORT_SAMPLES_PER_PAIR:-4}" \
51
+ --diversity-temperature "${CTT_DIVERSITY_TEMPERATURE:-1.0}" \
52
+ --no-markdown-report
53
+
54
+ "$PYTHON" scripts/eval_ctt_proxy.py \
55
+ --checkpoint "$TRAIN_OUT/model.pt" \
56
+ --source-index data/cil_charts/train/index.json \
57
+ --target-index data/cil_charts/val/index.json \
58
+ --out-dir "$VAL_OUT" \
59
+ --k "${CTT_PROXY_K:-16}" \
60
+ --max-target-charts "${CTT_MAX_TARGET_CHARTS:-100000}" \
61
+ --neighbors "${CTT_PROXY_NEIGHBORS:-8}" \
62
+ --thresholds "${CTT_PROXY_THRESHOLDS:-0.20,0.40}" \
63
+ --no-markdown-report
64
+
65
+ "$PYTHON" scripts/build_ctt_proxy_comparison.py \
66
+ --out-dir runs/ctt_val_proxy_comparison \
67
+ --no-markdown-report
68
+
69
+ "$PYTHON" scripts/summarize_ctt_runs.py \
70
+ --run-root runs \
71
+ --out-csv runs/summary_ctt.csv \
72
+ --out-md runs/summary_ctt.md \
73
+ --no-markdown-report