moved_repo_first_update
This commit is contained in:
Executable
+79
@@ -0,0 +1,79 @@
|
||||
#!/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."
|
||||
Reference in New Issue
Block a user