jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
https://jkjung-avt.github.io/
MIT License
1.74k stars 545 forks source link

Xavier onnx to TensorRT error #600

Closed kaikkk123 closed 1 year ago

kaikkk123 commented 1 year ago

Hello! I encountered the following error while running onnx to tensorRT on Xavier

[TensorRT] WARNING: TensorRT was linked against cuDNN 8.0.2 but loaded cuDNN 8.0.0 [TensorRT] ERROR: ../builder/cudnnBuilderWeightConverters.cpp (541) - Misc Error in operator(): 1 (Weights are outside of fp16 range. A possible fix is to retrain the model with regularization to bring the magnitude of the weights down.) [TensorRT] ERROR: ../builder/cudnnBuilderWeightConverters.cpp (541) - Misc Error in operator(): 1 (Weights are outside of fp16 range. A possible fix is to retrain the model with regularization to bring the magnitude of the weights down.) ERROR: failed to build the TensorRT engine!

I would like to ask if you have relevant experience, and if retraining is the only solution? Could this be related to my TensorRT or cuDNN version? TensorRT version : 7.2.0 cuDNN version : 8.0.0

jkjung-avt commented 1 year ago

https://github.com/jkjung-avt/tensorrt_demos/issues/161#issuecomment-662473343