#!/bin/bash
cd /home/yurenh2/ept/ep_run
for cfg in "qk3 --qknorm --lr 3e-3" "qk1 --qknorm --lr 1e-3" "nq3 --lr 3e-3" "nq1 --lr 1e-3"; do
  set -- $cfg; name=$1; shift
  CUDA_VISIBLE_DEVICES=0 python3 bp_lm.py "$@" --steps 32000 --log 1000 --ckpt runs/bp_$name.pt > runs/bp_$name.log 2>&1
  grep "DONE" runs/bp_$name.log
done
echo BP_SWEEP_ALL_DONE