From 0f107d1ee4e750421a0e95f8bd06bf0f629d2fa7 Mon Sep 17 00:00:00 2001
From: YurenHao0426 <Blackhao0426@gmail.com>
Date: Tue, 21 Jul 2026 07:02:04 -0500
Subject: feat: multiplex causal probes across hidden layers

---
 experiments/narrow_depth_sweep.sh | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'experiments/narrow_depth_sweep.sh')

diff --git a/experiments/narrow_depth_sweep.sh b/experiments/narrow_depth_sweep.sh
index 8f62a68..3e4e6b2 100755
--- a/experiments/narrow_depth_sweep.sh
+++ b/experiments/narrow_depth_sweep.sh
@@ -3,7 +3,7 @@
 #
 # Usage:
 #   bash experiments/narrow_depth_sweep.sh \
-#     <gpu> "<depths>" "<modes>" <width> <seed> <epochs> <prefix>
+#     <gpu> "<depths>" "<modes>" <width> <seed> <epochs> <prefix> [pert_mode] [n_dirs]
 #
 # Example:
 #   bash experiments/narrow_depth_sweep.sh 5 "30 60" "bp dfa sdil" 64 0 5 pilot_jac
@@ -18,6 +18,8 @@ WIDTH="${4:-64}"
 SEED="${5:-0}"
 EPOCHS="${6:-5}"
 PREFIX="${7:-narrow}"
+PERT_MODE="${8:-layerwise}"
+N_DIRS="${9:-8}"
 
 export CUDA_VISIBLE_DEVICES="$GPU"
 export OMP_NUM_THREADS=2
@@ -41,7 +43,7 @@ run_one() {
   "$PY" experiments/run.py \
     --mode "$mode" --dataset cifar10 --depth "$depth" --width "$WIDTH" \
     --act tanh --residual 1 --epochs "$EPOCHS" --seed "$SEED" \
-    --eta "$eta" --eta_A 0.02 --pert_ndirs 8 \
+    --eta "$eta" --eta_A 0.02 --pert_ndirs "$N_DIRS" --pert_mode "$PERT_MODE" \
     --log_every 500 --probe_bs 256 --tag "$tag" --outdir results \
     > "$log" 2>&1
   grep -h DONE "$log"
-- 
cgit v1.2.3