diff --git a/infra/k8s/argo/train-multi-seed-template.yaml b/infra/k8s/argo/train-multi-seed-template.yaml
index 8ee36257c..eee064b67 100644
--- a/infra/k8s/argo/train-multi-seed-template.yaml
+++ b/infra/k8s/argo/train-multi-seed-template.yaml
@@ -353,10 +353,10 @@ spec:
         resources:
           requests:
             cpu: "4"
-            memory: 16Gi
+            memory: 32Gi
           limits:
             cpu: "28"
-            memory: 56Gi
+            memory: 96Gi
         volumeMounts:
           - name: training-data
             mountPath: /data