NVIDIA-AI-IOT / redtail

Perception and AI components for autonomous mobile robotics.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Regarding tensorRT of TrailNet_SResNet-18

MSP93 opened this issue · comments

commented

Is the TrailNet_SResNet-18 Pretrained model an optimized ones that built using TensorRT? If so can you please share pretrained model without optimized ones.

TrailNet Caffe model was originally trained in Caffe using FP32 precision.
The model is used by our ROS node which uses TensorRT to load and optimize it for fast inference.
The target precision FP16/FP32 is selected based on user-provided parameters, see launch files for more details