rpotter6298
80 lines
2.3 KiB
Bash
Executable File
80 lines
2.3 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
set -euo pipefail
|
||
|
||
# Usage:
|
||
# bash scripts/run_gradual_thaw_top5.sh --epochs 20 --n-splits 5 --batch-size 8 [extra args]
|
||
#
|
||
# Runs the full gradual-thaw grid aligned with the SE sweep (bridge/tower/both × R=8/16/32 × norm vs nonorm).
|
||
|
||
ARGS=("$@")
|
||
|
||
run() {
|
||
local SHORT="$1"; shift
|
||
echo "=== Running: $SHORT ==="
|
||
if ls "analysis_data/${SHORT}_"*.md >/dev/null 2>&1; then
|
||
echo "… skipping ${SHORT} (summary already present)"
|
||
return 0
|
||
fi
|
||
python3 scripts/run_multifold.py \
|
||
--shortname "$SHORT" \
|
||
--gradual-thaw --thaw-phase-duration 5 --thaw-ratio 0.33 --thaw-start-epoch 5 \
|
||
--early-stop --early-patience 5 \
|
||
"$@" \
|
||
"${ARGS[@]}" || true
|
||
}
|
||
|
||
# Bridge-only thaw runs
|
||
for R in 8 16 32; do
|
||
for MODE in norm nonorm; do
|
||
if [[ "$MODE" == "norm" ]]; then
|
||
FLAGS=(--se-where bridge --se-reduction "$R" --se-pre-norm --checkpoint-best)
|
||
else
|
||
FLAGS=(--se-where bridge --se-reduction "$R" --no-se-pre-norm --checkpoint-best)
|
||
fi
|
||
run "thaw_se_bridge_R${R}_${MODE}" "${FLAGS[@]}"
|
||
run "thawbin_se_bridge_R${R}_${MODE}" "${FLAGS[@]}" --eval_mode binary
|
||
done
|
||
done
|
||
|
||
# Tower-only thaw runs
|
||
for R in 8 16 32; do
|
||
for MODE in norm nonorm; do
|
||
if [[ "$MODE" == "norm" ]]; then
|
||
FLAGS=(--se-where tower --se-reduction-tower "$R" --se-pre-norm-tower --checkpoint-best)
|
||
else
|
||
FLAGS=(--se-where tower --se-reduction-tower "$R" --no-se-pre-norm-tower --checkpoint-best)
|
||
fi
|
||
run "thaw_se_tower_R${R}_${MODE}" "${FLAGS[@]}"
|
||
run "thawbin_se_tower_R${R}_${MODE}" "${FLAGS[@]}" --eval_mode binary
|
||
done
|
||
done
|
||
|
||
# Tower+bridge thaw runs
|
||
for R in 8 16 32; do
|
||
for MODE in norm nonorm; do
|
||
if [[ "$MODE" == "norm" ]]; then
|
||
FLAGS=(
|
||
--se-where both
|
||
--se-reduction "$R"
|
||
--se-reduction-tower "$R"
|
||
--se-pre-norm
|
||
--se-pre-norm-tower
|
||
--checkpoint-best
|
||
)
|
||
else
|
||
FLAGS=(
|
||
--se-where both
|
||
--se-reduction "$R"
|
||
--se-reduction-tower "$R"
|
||
--no-se-pre-norm
|
||
--no-se-pre-norm-tower
|
||
--checkpoint-best
|
||
)
|
||
fi
|
||
run "thaw_se_tower_bridge_R${R}_${MODE}" "${FLAGS[@]}"
|
||
run "thawbin_se_tower_bridge_R${R}_${MODE}" "${FLAGS[@]}" --eval_mode binary
|
||
done
|
||
done
|
||
|
||
echo "Gradual thaw grid submitted. Check analysis_data/* and models/* for outputs."
|