(beam) branch master updated: [Python] Honor fixed object detection batch size (#39953)

[jrmccluskey]

This is an automated email from the ASF dual-hosted git repository.

jrmccluskey pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new 4855a35e92d [Python] Honor fixed object detection batch size (#39953)
4855a35e92d is described below

commit 4855a35e92d86f13fab50a77c66df35df3f631cb
Author: Bruno Volpato <brunocvcu...@gmail.com>
AuthorDate: Tue Sep 1 10:31:54 2026 -0400

    [Python] Honor fixed object detection batch size (#39953)
---
 .../apache_beam/examples/inference/pytorch_image_object_detection.py   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git 
a/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py 
b/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py
index a8eef5a7d5e..3279aaf4b4e 100644
--- 
a/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py
+++ 
b/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py
@@ -431,7 +431,8 @@ def run(
       model_params={},
       state_dict_path=known_args.model_state_dict_path,
       device=device,
-      inference_batch_size=batch_size,
+      min_batch_size=batch_size,
+      max_batch_size=batch_size,
       inference_fn=_torchvision_detection_inference_fn,
   )