diff --git a/infra/k8s/argo/gpu-test-pipeline-template.yaml b/infra/k8s/argo/gpu-test-pipeline-template.yaml
index 59633f5f0..eec65000b 100644
--- a/infra/k8s/argo/gpu-test-pipeline-template.yaml
+++ b/infra/k8s/argo/gpu-test-pipeline-template.yaml
@@ -433,7 +433,6 @@ spec:
               --symbol ES.FUT \
               --epochs 5 \
               --max-steps-per-epoch 100 \
-              --batch-size 1024 \
               --train-months 3 --val-months 1 --test-months 1 --step-months 3 \
               2>&1)