Failed covert onnx model to tensorrt engine ，when use convert_bevdet_to_TRT.py

HuangJunJie2017 / BEVDet

Code base of the BEVDet series .

Apache License 2.0

1.45k stars 265 forks source link

Open knavezl opened 2 months ago

knavezl commented 2 months ago

Describe the bug Failed covert onnx model to tensorrt engine ，when use convert_bevdet_to_TRT.py

Reproduction

python tools/convert_bevdet_to_TRT.py configs/bevdet/bevdet-r50.py ckpts/bevdet-r50.pth ckpts/bevdet-r50/ --fuse-conv-bn --fp16

Environment

Error traceback