diff options
Diffstat (limited to 'ep_run/runs/bp_sweep.sh')
| -rwxr-xr-x | ep_run/runs/bp_sweep.sh | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ep_run/runs/bp_sweep.sh b/ep_run/runs/bp_sweep.sh new file mode 100755 index 0000000..e98c79c --- /dev/null +++ b/ep_run/runs/bp_sweep.sh @@ -0,0 +1,8 @@ +#!/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 |