experiments/oral_a_apical_screen.sh


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

#!/usr/bin/env bash
set -euo pipefail

device="${1:-cuda}"
mkdir -p results/oral_a_apical_screen

for vectorizer in spatial_template channel_gated; do
  for a_scale in 0 1; do
    for eta_a in 0.001 0.01 0.1; do
      tag="${vectorizer}_a${a_scale}_etaA${eta_a}"
      python3 experiments/conv_run.py \
        --mode sdil --device "${device}" --depth 20 --width 16 \
        --seed 0 --loader_seed 0 --batch_size 128 --epochs 0 \
        --train_limit 10000 --val_examples 5000 --split_seed 2027 \
        --eval_split validation --eval_every 0 --augment_train 1 \
        --lr 0.03 --output_lr 0.1 --lr_schedule constant --warmup_epochs 0 \
        --momentum 0.9 --weight_decay 1e-4 --normalization batchnorm \
        --vectorizer_mode "${vectorizer}" --a_scale "${a_scale}" \
        --eta_A "${eta_a}" --a_warmup_steps 100 \
        --pert_sigma 0.01 --pert_directions 1 --pert_every 4 \
        --alignment_probe 32 \
        --out "results/oral_a_apical_screen/${tag}.json"
    done
  done
done