larger batch size for predictions

sasha-scale · sasha-scale · commit 6236c7ba0527 · 2021-03-14T22:45:33.000-07:00
diff --git a/nucleus/__init__.py b/nucleus/__init__.py
@@ -657,7 +657,7 @@ def predict(
             Union[BoxPrediction, PolygonPrediction, SegmentationPrediction]
         ],
         update: bool,
-        batch_size: int = 100,
+        batch_size: int = 5000,
     ):
         """
         Uploads model outputs as predictions for a model_run. Returns info about the upload.
