| #!/usr/bin/env bash |
| |
| |
| set -euo pipefail |
|
|
| REPO="/home/mikeb/work/feather" |
| cd "$REPO" |
|
|
| |
| pkill -9 -f "python.*train\.py" 2>/dev/null || true |
| sleep 1 |
|
|
| HF_TOKEN_VAL=$(grep -ohP 'hf_[A-Za-z0-9_-]+' ~/.bashrc 2>/dev/null | head -1 || true) |
| TICK="${1:-0}" |
| LOG="${REPO}/run_loop_t${TICK}.log" |
|
|
| echo "[loop] tick-${TICK} starting $(date +%H:%M:%S)" > "${LOG}" |
|
|
| setsid -f /usr/bin/env \ |
| LD_LIBRARY_PATH=/usr/lib/wsl/lib:/usr/local/cuda/lib64 \ |
| PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True \ |
| HF_TOKEN="${HF_TOKEN_VAL}" \ |
| HUGGINGFACE_HUB_TOKEN="${HF_TOKEN_VAL}" \ |
| WANDB_DISABLED=true \ |
| HYDRA_USE_NEMOTRON=1 \ |
| HYDRA_USE_FULL_BLEND=1 \ |
| HYDRA_SAMPLED_SOFTMAX=256 \ |
| HYDRA_SOFTCAP_CLAMP=1 \ |
| HYDRA_SEQ_LEN=1024 \ |
| HYDRA_HEADDIM=32 \ |
| HYDRA_D_STATE=64 \ |
| HYDRA_TIME_BUDGET=300 \ |
| HYDRA_ENGRAM_TOPK=64 \ |
| HYDRA_CANTOR_DISABLE=0 \ |
| HYDRA_CANTOR_LEARNABLE=1 \ |
| HYDRA_CANTOR_SCORE_GRAD=1 \ |
| HYDRA_ENGRAM_ROUTING=auto \ |
| HYDRA_REALITY_BRIDGE=1 \ |
| HYDRA_SEMANTIC_SMOOTH_STD=0.01 \ |
| HYDRA_SLOW_FAST_ORTHO_METRICS=1 \ |
| HYDRA_SLOW_FAST_ORTHO_LAMBDA=1e-4 \ |
| HYDRA_GDN_LAYERS= \ |
| HYDRA_MTP_K=1 \ |
| HYDRA_USE_MDLM=0 \ |
| HYDRA_MUON_COMPILE=0 \ |
| HYDRA_MUON_NS_STEPS=2 \ |
| HYDRA_MATRIX_LR="${2:-0.01}" \ |
| HYDRA_EMBED_LR="${3:-0.20}" \ |
| HYDRA_UNEMBED_LR="${4:-0.001}" \ |
| HYDRA_DT_BIAS_LR="${5:-0.05}" \ |
| HYDRA_SCALAR_LR="${6:-0.01}" \ |
| HYDRA_WARMUP_RATIO=0.01 \ |
| HYDRA_LR_MIN_MULT=0.10 \ |
| HYDRA_DOC_SEP_MASK=1 \ |
| HYDRA_STREAM_SHUFFLE_BUFFER=4096 \ |
| HYDRA_LOCAL_SHARDS_ONLY=0 \ |
| HYDRA_BACKGROUND_PREFETCH=0 \ |
| HYDRA_STREAM_PREFETCH=16 \ |
| HYDRA_TOKEN_PREFETCH=4 \ |
| HYDRA_TOKEN_CACHE_GB=1 \ |
| HYDRA_CKPT_INTERVAL=2000 \ |
| HYDRA_MID_VAL_INTERVAL=0 \ |
| HYDRA_EVAL_BATCH=1 \ |
| HYDRA_EVAL_TOKENS=51200 \ |
| HYDRA_CE_CHUNK=16 \ |
| HYDRA_SKIP_FACTUAL_EVAL=1 \ |
| HYDRA_N_LAYER=6 \ |
| HYDRA_D_MODEL=192 \ |
| HYDRA_EXPAND=3 \ |
| HYDRA_BATCH_SIZE=16 \ |
| HYDRA_TOTAL_BATCH=32768 \ |
| HYDRA_HYENA_LAYERS= \ |
| HYDRA_HTM_SUBSAMPLE=16 \ |
| UV_PYTHON=/usr/bin/python3 \ |
| taskset -c 0-15 "${REPO}/.venv/bin/python" -u train.py \ |
| </dev/null >>"${LOG}" 2>&1 |
|
|
| sleep 2 |
| TPID=$(pgrep -n -f 'python -u train\.py' || echo "") |
| if [ -z "${TPID}" ]; then |
| TPID=$(pgrep -n -f 'train\.py' || echo "0") |
| fi |
| echo "${TPID}" > /home/mikeb/.cache/autoresearch/train_pid |
| echo "[loop] tick-${TICK} PID=${TPID} PPID=$(ps -o ppid= -p "${TPID}" 2>/dev/null || echo '?')" >> "${LOG}" |
|
|