diff --git a/infra/k8s/argo/alpha-perception-template.yaml b/infra/k8s/argo/alpha-perception-template.yaml
index 1580eaad0..f9d487a68 100644
--- a/infra/k8s/argo/alpha-perception-template.yaml
+++ b/infra/k8s/argo/alpha-perception-template.yaml
@@ -62,6 +62,10 @@ spec:
         value: "1000"
       - name: seed
         value: "16962"
+      - name: batch-size
+        value: "1"
+      - name: auto-horizon-weights
+        value: "false"
 
   volumes:
     - name: git-ssh-key
@@ -409,6 +413,11 @@ spec:
             echo "  predecoded:    $PREDECODED_DIR"
             echo "  out:           $OUT_DIR"
 
+            EXTRA_FLAGS=""
+            if [ "{{workflow.parameters.auto-horizon-weights}}" = "true" ]; then
+              EXTRA_FLAGS="$EXTRA_FLAGS --auto-horizon-weights"
+            fi
+
             "$BIN" \
               --mbp10-data-dir   "$MBP10_DIR" \
               --predecoded-dir   "$PREDECODED_DIR" \
@@ -420,7 +429,9 @@ spec:
               --lr-mamba2        {{workflow.parameters.lr-mamba2}} \
               --n-train-seqs     {{workflow.parameters.n-train-seqs}} \
               --n-val-seqs       {{workflow.parameters.n-val-seqs}} \
-              --seed             {{workflow.parameters.seed}}
+              --seed             {{workflow.parameters.seed}} \
+              --batch-size       {{workflow.parameters.batch-size}} \
+              $EXTRA_FLAGS
 
             echo "=== Training complete ==="
             ls -lh "$OUT_DIR/"
diff --git a/scripts/argo-alpha-perception.sh b/scripts/argo-alpha-perception.sh
index ab3c46ea4..0e9c5b7b1 100755
--- a/scripts/argo-alpha-perception.sh
+++ b/scripts/argo-alpha-perception.sh
@@ -27,6 +27,8 @@ LR_MAMBA2=1e-3
 N_TRAIN_SEQS=8000
 N_VAL_SEQS=1000
 SEED=16962
+BATCH_SIZE=1
+AUTO_HORIZON_WEIGHTS=false
 WATCH=false
 
 usage() {
@@ -60,6 +62,8 @@ while [[ $# -gt 0 ]]; do
     --n-train-seqs)      N_TRAIN_SEQS="$2"; shift 2 ;;
     --n-val-seqs)        N_VAL_SEQS="$2"; shift 2 ;;
     --seed)              SEED="$2"; shift 2 ;;
+    --batch-size)        BATCH_SIZE="$2"; shift 2 ;;
+    --auto-horizon-weights) AUTO_HORIZON_WEIGHTS=true; shift ;;
     --watch)             WATCH=true; shift ;;
     -h|--help)           usage; exit 0 ;;
     *) echo "Unknown option: $1"; usage; exit 1 ;;
@@ -123,4 +127,6 @@ argo submit -n foxhunt --from=wftmpl/alpha-perception \
   -p n-train-seqs="$N_TRAIN_SEQS" \
   -p n-val-seqs="$N_VAL_SEQS" \
   -p seed="$SEED" \
+  -p batch-size="$BATCH_SIZE" \
+  -p auto-horizon-weights="$AUTO_HORIZON_WEIGHTS" \
   $WATCH_FLAG