This is an automated email from the ASF dual-hosted git repository. vterentev pushed a commit to branch oss-image-cpu in repository https://gitbox.apache.org/repos/asf/beam.git
commit bd5c9da59da097e2ef2021d7ffbbd8d45f9eaed8 Author: Vitaly Terentyev <[email protected]> AuthorDate: Wed Jan 21 22:00:17 2026 +0400 Test with cpu --- ...am_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt | 2 +- ...Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt | 2 +- ...erence_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt index d37e91abbc2..07917a797c0 100644 --- a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt +++ b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt @@ -25,7 +25,7 @@ --metrics_table=result_torch_inference_image_captioning_batch --input_options={} --influx_measurement=result_torch_inference_image_captioning_batch ---device=GPU +--device=CPU --mode=batch --input=gs://apache-beam-ml/testing/inputs/openimage_50k_benchmark.txt --blip_model_name=Salesforce/blip-image-captioning-base diff --git a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt index 7920a3a69ab..82f5d6e2325 100644 --- a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt +++ b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt @@ -27,7 +27,7 @@ --metrics_table=torch_inference_imagenet_results_stream_rightfit --influx_measurement=torch_inference_imagenet_stream_rightfit --pretrained_model_name=efficientnet_b0 ---device=GPU +--device=CPU --input_file=gs://apache-beam-ml/testing/inputs/openimage_50k_benchmark.txt --runner=DataflowRunner --mode=streaming diff --git a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt index 0bd0778224e..f92b22dd86f 100644 --- a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt +++ b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt @@ -28,7 +28,7 @@ --input_options={} --influx_measurement=result_torch_inference_image_object_detection_batch --pretrained_model_name=fasterrcnn_resnet50_fpn ---device=GPU +--device=CPU --mode=batch --inference_batch_size=8 --resize_shorter_side=800
