dloc-code / scripts /run_all_multiseed.sh

Upload scripts/run_all_multiseed.sh with huggingface_hub

920e526 verified 5 months ago

2.15 kB

#!/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

	#!/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