File size: 2,147 Bytes
920e526 | 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 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | #!/bin/bash
# Master script: Run all multi-seed experiments (120 epochs, 3 seeds each)
# 8 variants: baseline + mobilenet×2 + tinycnn×2 + unet×3
#
# Estimated: ~28 hrs on H100
set -e
cd /workspace/dloc-code
echo "============================================"
echo " MULTI-SEED PIPELINE START: $(date)"
echo "============================================"
# Helper to update config and run student models
run_student() {
local STYPE=$1
local MODE=$2
local SCHED=$3
echo ""
echo "============================================"
echo " Starting: ${STYPE} / ${MODE} / sched=${SCHED}"
echo " Time: $(date)"
echo "============================================"
sed -i "s/STUDENT_TYPE = .*/STUDENT_TYPE = '${STYPE}'/" run_multiple_seeds.py
sed -i "s/MODE = .*/MODE = '${MODE}'/" run_multiple_seeds.py
sed -i "s/SCHEDULER_TYPE = .*/SCHEDULER_TYPE = '${SCHED}'/" run_multiple_seeds.py
sed -i "s/EPOCHS = .*/EPOCHS = 120/" run_multiple_seeds.py
python3 run_multiple_seeds.py 2>&1 | tee "/tmp/multiseed_${STYPE}_${MODE}_${SCHED}.log"
}
# 1. Baseline (separate script, ~12.8 hrs)
echo ""
echo "============================================"
echo " Starting: BASELINE"
echo " Time: $(date)"
echo "============================================"
python3 run_multiseed_baseline.py 2>&1 | tee /tmp/multiseed_baseline.log
# 2. MobileNet standalone (~2.5 hrs)
run_student mobilenet standalone none
# 3. MobileNet distill (~2.5 hrs)
run_student mobilenet distill none
# 4. TinyCNN standalone (~2.1 hrs)
run_student tinycnn standalone none
# 5. TinyCNN distill (~2.1 hrs)
run_student tinycnn distill none
# 6. U-Net cosine standalone (~2.7 hrs)
run_student unet standalone cosine
# 7. U-Net cosine distill (~2.7 hrs)
run_student unet distill cosine
echo ""
echo "============================================"
echo " ALL MULTI-SEED RUNS COMPLETE: $(date)"
echo "============================================"
echo ""
# Print all reports
echo "=== FINAL SUMMARY ==="
for f in /workspace/dloc-code/runs/multiseed_*/report.txt; do
echo ""
echo "--- $(basename $(dirname $f)) ---"
cat "$f"
done
|