1 files changed, 61 insertions, 0 deletions
diff --git a/research/flossing/flossing_suite/smoke_test.sh b/research/flossing/flossing_suite/smoke_test.sh
new file mode 100755
index 0000000..8b766b5
--- /dev/null
+++ b/research/flossing/flossing_suite/smoke_test.sh
@@ -0,0 +1,61 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+GPU="${1:-0}"
+ROOT="/home/yurenh2/rrm"
+PY="/home/yurenh2/miniconda3/envs/rrm/bin/python"
+OUT_DIR="${ROOT}/research/flossing/flossing_suite/results/smoke"
+mkdir -p "${OUT_DIR}"
+cd "${ROOT}"
+
+echo "[smoke] toy RNN faithful port"
+CUDA_VISIBLE_DEVICES="${GPU}" PYTHONUNBUFFERED=1 "${PY}" research/flossing/engelken_python_flossing.py \
+  --hidden-size 16 \
+  --n-lyap 4 \
+  --train-epochs 3 \
+  --pre-epochs 2 \
+  --inter-epochs 0 \
+  --max-inter-episodes 0 \
+  --batch-size 4 \
+  --train-steps 40 \
+  --lyap-steps 30 \
+  --floss-input-steps 40 \
+  --eval-every 1 \
+  --eval-batches 1 \
+  --device cuda \
+  --out "${OUT_DIR}/toy_smoke.json" \
+  > "${OUT_DIR}/toy_smoke.log" 2>&1
+
+echo "[smoke] TRM interfloss analogue"
+CUDA_VISIBLE_DEVICES="${GPU}" PYTHONUNBUFFERED=1 "${PY}" research/flossing/step7_interfloss.py \
+  --model trm \
+  --ckpt-root "${ROOT}/trm/checkpoints/Sudoku-extreme-1k-aug-1000-ACT-torch/pretrain_mlp_t_sudoku_official_gbs768_repro" \
+  --ckpt-name __random__ \
+  --init-seed 123 \
+  --train-steps 1 \
+  --batch-size 2 \
+  --train-lr 1e-4 \
+  --floss-lr 1e-4 \
+  --floss-steps 1 \
+  --interfloss-at 0 \
+  --floss-mode engelken_l2 \
+  --lambda-star 0 \
+  --k-lyap 1 \
+  --lyap-act-steps 1 \
+  --seed 42 \
+  --eval-every 1 \
+  --eval-n 8 \
+  --eval-batch-size 8 \
+  --floss-log-every 1 \
+  --no-eval-after-floss \
+  --train-puzzle-emb \
+  --puzzle-emb-lr 1e-4 \
+  --puzzle-emb-weight-decay 1.0 \
+  --out "${OUT_DIR}/trm_step7_smoke.json" \
+  > "${OUT_DIR}/trm_step7_smoke.log" 2>&1
+
+echo "[smoke] summarizer"
+"${PY}" research/flossing/flossing_suite/summarize_flossing.py \
+  > "${OUT_DIR}/summarize_smoke.log" 2>&1
+
+echo "smoke OK: ${OUT_DIR}"