blob: c5ec6118b015756a0c34072626352e3ec66a8aa6 (
plain)
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#!/usr/bin/env bash
# Development-only feedback-first timescale screen. Task seed 0 is reused;
# untouched task seeds 6--8 remain reserved for independent confirmation.
#
# The joint phase is the original context-SDIL K1/e4 simultaneous protocol.
# Before it, A receives 0/100/400 layerwise K4 calibration steps while every W
# and bias (including the readout) is frozen. See the analyzer for the frozen
# eligibility and smaller-prefix-within-one-point selection rule.
# Usage: c2_feedback_warmup_pilot.sh <gpu> "<model seeds>"
set -eu
cd "$(dirname "$0")/.."
GPU="${1:?GPU index required}"
MODEL_SEEDS="${2:-0 1 2}"
PYTHON="${PYTHON:-/home/yurenh2/miniconda3/envs/ep_pascal/bin/python3}"
for warmup_steps in 0 100 400; do
for model_seed in $MODEL_SEEDS; do
for depth in 1 4; do
tag="c2_awarmdev_v1_tent_l2_w8_context_a${warmup_steps}_d${depth}_t0_s${model_seed}"
out="results/${tag}.json"
if [ -s "$out" ]; then
echo "[$tag] exists; skipping"
continue
fi
lesion=0
if [ "$depth" -eq 4 ]; then
lesion=0.3333333333333333
fi
CUDA_VISIBLE_DEVICES="$GPU" "$PYTHON" experiments/run.py \
--mode sdil --dataset tentmap --device cuda \
--depth "$depth" --width 8 --act relu --residual 1 \
--residual_lesion_fraction "$lesion" \
--vectorizer_mode context_gated \
--epochs 80 --batch_size 256 --eta 0.03 --momentum 0.9 \
--eta_A 0.01 --eta_P 0.002 \
--pert_sigma 0.01 --pert_every 4 --pert_ndirs 1 \
--pert_mode simultaneous \
--a_warmup_steps "$warmup_steps" \
--a_warmup_ndirs 4 --a_warmup_mode layerwise \
--traffic_mode none --nuis_rho 0 \
--use_residual 1 --learn_A 1 --learn_P 1 --p_neutral 1 \
--task_train_examples 10000 --task_test_examples 5000 \
--task_levels 2 --task_n_in 1 --task_seed 0 \
--val_examples 2000 --split_seed 2027 --eval_split validation --eval_every 0 \
--diagnostics alignment --diagnostics_schedule final --probe_bs 512 \
--seed "$model_seed" --log_every 100000 --tag "$tag"
done
done
done
|
