search

linghu8812 / tensorrt_inference

705 stars 207 forks source link

Stucked when convert yolov5s.onnx to yolov5s.trt #67

Closed Zejun-Yang closed 3 years ago

Zejun-Yang commented 3 years ago

When I run './yolovs_trt ../config.yaml ../samples', this project would convert the onnx model to trt model with common.hpp. However, the program stucks with the log "strat building engine". Seems that something wrong with "engine = builder->buildEngineWithConfig(network, config);". in common.hpp.

Thank you for your attention.

Zejun-Yang commented 3 years ago

I have solved.