search

Haiyang-W / DSVT

[CVPR2023] Official Implementation of "DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets"
https://arxiv.org/abs/2301.06051
Apache License 2.0
353 stars 28 forks source link

DSVT tensorrt model slower than pytorch #53

Closed realwenpeng closed 10 months ago

realwenpeng commented 10 months ago

Thanks for your great job!!! test on A30 GPU DSVT-pytroch 20.03ms, DSVT-trt 69ms I think it may because of the dynamic shape setting when tranfer onnx to engine. I want to know how to get the optimal dynamic shape of the trtexec operation on my own datasets.

chenshi3 commented 10 months ago

The dynamic shapes in this implementation are based on the statistics of the Waymo-validation-set, which may not be optimal for your own datasets. Additionally, I recommend testing on fixed shapes in a single scene to identify and eliminate any other potential bugs.

realwenpeng commented 10 months ago

Thanks, I would try....