# 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_b1_tj*.json results-loop/train_carrycot_b1_*tj*_log.json results-loop/adapter_carrycot_b1_*tj*_e200.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
for FR in 0.0 5.0; do
  T=tjs50_$(echo $FR | tr -d .)
  $P train_carry_cot.py --drop-steps 1 --traj-span step --traj-tf 5.0 --traj-fr $FR \
     --pause-per-step 0 --base-pauses 0 --inner-iters 10 \
     --warm-start $LOOP_OUT/adapter_carrycot_e400.pt --steps 200 --lr 3e-4
  A=$LOOP_OUT/adapter_carrycot_b1_ii10_${T}_p0_e200.pt
  $P eval_carry_cot.py --adapter $A --tag gsm_carrycot_b1_$T --grid 0:0,2:0 --n 256 --inner-iters 10
  $P eval_carry_cot.py --adapter $A --tag gsm_carrycot_b1_${T}_ablate --grid 2:0 --n 256 --inner-iters 0
done