Fix the bug that tensorRT batch_size does not take effect (ultralytic…

…s#11672)

* Fix the bug that tensorRT batch_size does not take effect

Signed-off-by: wuhongsheng <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Signed-off-by: wuhongsheng <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

benchmarks.py

@@ -76,7 +76,12 @@ def run(
  if f == '-':
  w = weights # PyTorch format
  else:
- w = export.run(weights=weights, imgsz=[imgsz], include=[f], device=device, half=half)[-1] # all others
+ w = export.run(weights=weights,
+ imgsz=[imgsz],
+ include=[f],
+ batch_size=batch_size,
+ device=device,
+ half=half)[-1] # all others
  assert suffix in str(w), 'export failed'
 
  # Validate