How to export to tensorrt

[RainRoboforce]

Thanks @wenbowen123 for the great work! I have tested it on my scenarios and works fantastic!
Just want to ask if there are any instructions for exporting the model to tensorrt? Or any insights would be helpful!

[wenbowen123]

this will soon be released in NVIDIA Isaac SDK https://developer.nvidia.com/isaac/manipulator

[RainRoboforce]

Thanks for the reply! I am new to pytorch. A naive question: if I want to test the model in c++ programs. Can I use torchscript to save the weights and load them in c++?

[monajalal]

@wenbowen123 I am also interested in learning how to run your model on Jetson. If you may be able to share instructions for running this on Jetson or converting it to trt would be great.