diff --git a/infra/k8s/argo/train-multi-seed-template.yaml b/infra/k8s/argo/train-multi-seed-template.yaml index 983907bb4..0642fb753 100644 --- a/infra/k8s/argo/train-multi-seed-template.yaml +++ b/infra/k8s/argo/train-multi-seed-template.yaml @@ -365,6 +365,7 @@ spec: if [ ! -x "$BINARY" ]; then echo "ERROR: precompute_features not found at $BINARY"; exit 1 fi + export RAYON_NUM_THREADS=8 if $BINARY \ --data-dir /data/futures-baseline \ --mbp10-data-dir /data/futures-baseline-mbp10 \