# gpuq-in:  results-loop/star_data.json results-loop/gsm_cot_data.json results-loop/adapter_carrycot_e400.pt
# gpuq-out: results-loop/eval_gsm_carrycot_b*.json results-loop/train_carrycot_b*_log.json results-loop/adapter_carrycot_b*.pt
git pull origin main -q 2>/dev/null
P=/home/nils/jspace/.venv/bin/python
export JLENS_MODEL=google/gemma-4-E2B-it LOOP_OUT=/home/nils/jspace/results-loop
cd /home/nils/jspace/scripts
WARM=$LOOP_OUT/adapter_carrycot_e400.pt
for D in 1 2 3; do
  $P train_carry_cot.py --drop-steps $D --warm-start $WARM --steps 200 --lr 3e-4
  WARM=$LOOP_OUT/adapter_carrycot_b${D}_e200.pt
  $P eval_carry_cot.py --adapter $WARM --tag gsm_carrycot_b$D \
     --grid 0:0,2:$((2+10*D)),2:$((6+10*D)) --n 256
done