tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

the command : ./yolov5_det -s yolov5s.wts yolov5s.engine s, error!!!

Open coding-and-coding opened this issue 9 months ago • 3 comments

Env

  • GPU: RTX3090
  • OS: Ubuntu 20.04
  • Cuda version: 11.4
  • TensorRT version: 8.5.2.2
  • torch: 1.12.0

About this repo

  • branch: tensorrtx(yolov5-v7.0)
  • model: yolov5s, v7.0, official weight

Your problem

  • my command: ./yolov5_det -s yolov5s.wts yolov5s.engine s

  • output: [E] [TRT] 9: [graphShapeAnalyzer.cpp::addVolumeCheck::739] Error Code 9: Internal Error (prob: dimension 0 never exceeds -5) [05/24/2024-19:01:38] [E] [TRT] 4: [network.cpp::validate::2980] Error Code 4: Internal Error (Could not compute dimensions for (Unnamed Layer* 0) [Convolution]_output, because the network is not valid.)

  • expect output: yolov5s.engine

coding-and-coding avatar May 24 '24 11:05 coding-and-coding