diff options
| author | YurenHao0426 <blackhao0426@gmail.com> | 2026-06-13 12:35:36 -0500 |
|---|---|---|
| committer | YurenHao0426 <blackhao0426@gmail.com> | 2026-06-13 12:35:36 -0500 |
| commit | 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a (patch) | |
| tree | c29cba61124018755a19b02c9d33e3ad5f2e05cc /research/flossing/launch_engelken_python_official_queue.sh | |
Curated export for clone-and-run Maze training (2x A6000) + diagnostics.
trm/hrm pretrain.py carry trajectory-augmentation code (backward-compatible).
Heavy artifacts (checkpoints/wandb/npz) gitignored; see PROVENANCE.md.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Diffstat (limited to 'research/flossing/launch_engelken_python_official_queue.sh')
| -rwxr-xr-x | research/flossing/launch_engelken_python_official_queue.sh | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/research/flossing/launch_engelken_python_official_queue.sh b/research/flossing/launch_engelken_python_official_queue.sh new file mode 100755 index 0000000..d1b3faf --- /dev/null +++ b/research/flossing/launch_engelken_python_official_queue.sh @@ -0,0 +1,74 @@ +#!/usr/bin/env bash +set -euo pipefail + +cd /home/yurenh2/rrm + +PY="/home/yurenh2/miniconda3/envs/rrm/bin/python" +SCRIPT="research/flossing/engelken_python_flossing.py" +OUT_DIR="research/flossing/engelken_python" +mkdir -p "${OUT_DIR}" + +common=( + --hidden-size 80 + --n-lyap 40 + --train-epochs 1000 + --inter-period 100 + --inter-epochs 100 + --batch-size 16 + --input-dim 1 + --train-steps 300 + --lyap-steps 55 + --floss-input-steps 300 + --seed-ic 1 + --seed-input 1 + --seed-net 1 + --seed-ons 1 + --lr 0.001 + --beta1 0.9 + --beta2 0.999 + --init-type 1 + --recurrent-gain 1.0 + --recurrent-mean-gain 0.0 + --input-scale 1.0 + --delay 10 + --ws-std 1.0 + --ws-mean 0.0 + --wr-std 1.0 + --wr-mean 0.0 + --b-std 0.1 + --b-mean 0.0 + --task -1 + --lyap-target 0.0 + --eval-every 100 + --eval-batches 4 + --log-every-floss 25 + --device cuda +) + +run_case() { + local gpu="$1" + local name="$2" + shift 2 + CUDA_VISIBLE_DEVICES="${gpu}" PYTHONUNBUFFERED=1 "${PY}" "${SCRIPT}" \ + "${common[@]}" "$@" \ + --out "${OUT_DIR}/${name}.json" \ + > "${OUT_DIR}/${name}.log" 2>&1 +} + +( + run_case 2 "official_py_baseline_no_floss_N80_k40_E1000" \ + --pre-epochs 0 --max-inter-episodes 0 + run_case 2 "official_py_pre_inter_N80_k40_E1000" \ + --pre-epochs 100 --max-inter-episodes 2 +) & +echo $! > "${OUT_DIR}/gpu2_python_queue.pid" + +( + run_case 3 "official_py_prefloss_N80_k40_E1000" \ + --pre-epochs 100 --max-inter-episodes 0 +) & +echo $! > "${OUT_DIR}/gpu3_python_queue.pid" + +echo "queued Engelken Python official-analogue runs" +echo "gpu2 queue pid: $(cat "${OUT_DIR}/gpu2_python_queue.pid")" +echo "gpu3 queue pid: $(cat "${OUT_DIR}/gpu3_python_queue.pid")" |
