blob: e98c79c7f192584361f2005170d815577b1c5561 (
plain)
1
2
3
4
5
6
7
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
|