From 7bd5b3ec3b1ca1b78fa06e9dd7e006e8568ec5fc Mon Sep 17 00:00:00 2001 From: kkannan Date: Wed, 10 Apr 2024 09:06:53 +0000 Subject: [PATCH] Patch for yolov5m (640x640) onnx model's CI job failure (cherry picked from commit ddcb47ea7974e737039079ff3da46e07d940b892) --- .../model_demos/high_prio/cnn/onnx/test_yolo_v5.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pybuda/test/model_demos/high_prio/cnn/onnx/test_yolo_v5.py b/pybuda/test/model_demos/high_prio/cnn/onnx/test_yolo_v5.py index a9d88cc8..938cd6b9 100644 --- a/pybuda/test/model_demos/high_prio/cnn/onnx/test_yolo_v5.py +++ b/pybuda/test/model_demos/high_prio/cnn/onnx/test_yolo_v5.py @@ -82,7 +82,7 @@ def test_yolo_v5_320x320_onnx(test_device, variant): model_name = f"{variant}_{input_size}_onnx" # Load data sample - url = "https://ultralytics.com/images/zidane.jpg" + url = "http://images.cocodataset.org/val2017/000000397133.jpg" image = Image.open(requests.get(url, stream=True).raw).convert("RGB") # Data preprocessing on Host @@ -157,7 +157,7 @@ def test_yolo_v5_480x480_onnx(test_device, variant): model_name = f"{variant}_{input_size}_onnx" # Load data sample - url = "https://ultralytics.com/images/zidane.jpg" + url = "http://images.cocodataset.org/val2017/000000397133.jpg" image = Image.open(requests.get(url, stream=True).raw).convert("RGB") # Data preprocessing on Host @@ -203,10 +203,10 @@ def test_yolo_v5_640x640_onnx(test_device, variant): if variant == "yolov5m": compiler_cfg.balancer_op_override( - " concatenate_332.dc.concatenate.7", "grid_shape", (1, 1) + "concatenate_332.dc.concatenate.7", "grid_shape", (1, 1) ) compiler_cfg.balancer_op_override( - " concatenate_332.dc.concatenate.7", "t_stream_shape", (1, 1) + "concatenate_332.dc.concatenate.7", "t_stream_shape", (1, 1) ) os.environ["TT_BACKEND_OVERLAY_MAX_EXTRA_BLOB_SIZE"] = f"{112*1024}" @@ -260,7 +260,7 @@ def test_yolo_v5_640x640_onnx(test_device, variant): model_name = f"{variant}_{input_size}_onnx" # Load data sample - url = "https://ultralytics.com/images/zidane.jpg" + url = "http://images.cocodataset.org/val2017/000000397133.jpg" image = Image.open(requests.get(url, stream=True).raw).convert("RGB") # Data preprocessing on Host