This is an automated email from the ASF dual-hosted git repository.
damccorm 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 71f2278c79f Update
beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt
(#31341)
71f2278c79f is described below
commit 71f2278c79fbf64408f4cf6b1b1f09227498bb3a
Author: liferoad <huxiangq...@gmail.com>
AuthorDate: Sat May 18 07:35:43 2024 -0400
Update
beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt
(#31341)
Fix #30644
---
...taflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt
index e79369befad..ce67b4e116f 100644
---
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt
+++
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Imagenet_Classification_Resnet_152_Tesla_T4_GPU.txt
@@ -29,8 +29,8 @@
--influx_measurement=torch_inference_imagenet_resnet152_tesla_t4
--pretrained_model_name=resnet152
--device=GPU
---experiments=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver
+--experiments=worker_accelerator=type:nvidia-tesla-t4;count:1;install-nvidia-driver:5xx
--sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
--input_file=gs://apache-beam-ml/testing/inputs/openimage_50k_benchmark.txt
--model_state_dict_path=gs://apache-beam-ml/models/torchvision.models.resnet152.pth
---runner=DataflowRunner
\ No newline at end of file
+--runner=DataflowRunner
