chore: lower bsize, resultd unaffected

AnswerDotAI · Mar 18, 2024 · 27f2f23 · 27f2f23
1 parent e6aabb9
commit 27f2f23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ragatouille/models/torch_kmeans.py b/ragatouille/models/torch_kmeans.py
@@ -27,7 +27,7 @@ def compute_pytorch_kmeans(
     num_partitions,
     kmeans_niters,
     use_gpu,
-    batch_size=256000,
+    batch_size=64000,
     verbose=1,
     seed=123,
     max_points_per_centroid=256,