CausalGrok / code /scripts /run_sanity_check.sh

Upload code/scripts

42c0d23 verified 6 months ago

701 Bytes

#!/usr/bin/env bash
# scripts/run_sanity_check.sh
# Fires both the standard control AND the full grokking recipe, in parallel,
# each detached under nohup. Use this as the Week 2 go/no-go signal.

set -euo pipefail

ROOT="$(cd "$(dirname "$0")/.." && pwd)"
cd "${ROOT}"

GPU_STD="${GPU_STD:-0}"
GPU_GROK="${GPU_GROK:-1}"
N_TRAIN="${N_TRAIN:-500}"
SEED="${SEED:-42}"

echo "Sanity check — both conditions, detached"
bash scripts/launch.sh standard "${N_TRAIN}" "${SEED}" "${GPU_STD}"
bash scripts/launch.sh grokking "${N_TRAIN}" "${SEED}" "${GPU_GROK}"

echo
echo "Both runs detached. Watch progress with:"
echo "  bash scripts/list_runs.sh"
echo "  tail -f experiments/runs/<run_id>/logs/train.log"

	#!/usr/bin/env bash
	# scripts/run_sanity_check.sh
	# Fires both the standard control AND the full grokking recipe, in parallel,
	# each detached under nohup. Use this as the Week 2 go/no-go signal.

	set -euo pipefail

	ROOT="$(cd "$(dirname "$0")/.." && pwd)"
	cd "${ROOT}"

	GPU_STD="${GPU_STD:-0}"
	GPU_GROK="${GPU_GROK:-1}"
	N_TRAIN="${N_TRAIN:-500}"
	SEED="${SEED:-42}"

	echo "Sanity check — both conditions, detached"
	bash scripts/launch.sh standard "${N_TRAIN}" "${SEED}" "${GPU_STD}"
	bash scripts/launch.sh grokking "${N_TRAIN}" "${SEED}" "${GPU_GROK}"

	echo
	echo "Both runs detached. Watch progress with:"
	echo " bash scripts/list_runs.sh"
	echo " tail -f experiments/runs/<run_id>/logs/train.log"