Changed batch size to be 16 to decrease memory load

brainglobe · Oct 25, 2024 · d6ddc52 · d6ddc52
1 parent 2a0583c
commit d6ddc52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brainglobe_stitch/image_mosaic.py b/brainglobe_stitch/image_mosaic.py
@@ -1040,7 +1040,7 @@ def _fuse_to_2d_tiff(
             The shape of the fused image.
         """
         z_size, y_size, x_size = self.tiles[0].data_pyramid[0].shape
-        batch_size = self.tiles[0].data_pyramid[0].chunksize[0]
+        batch_size = 16
 
         batched_image_shape = (batch_size, *fused_image_shape[1:])