Skip to content

Commit

Permalink
Patch for yolov5m (640x640) onnx model's CI job failure
Browse files Browse the repository at this point in the history
(cherry picked from commit ddcb47ea7974e737039079ff3da46e07d940b892)
  • Loading branch information
kamalrajkannan78 authored and vmilosevic committed Apr 11, 2024
1 parent 42d2815 commit 7bd5b3e
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions pybuda/test/model_demos/high_prio/cnn/onnx/test_yolo_v5.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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}"

Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 7bd5b3e

Please sign in to comment.