huhuhu12138 / torchtrtz Goto Github PK
View Code? Open in Web Editor NEWThis project forked from makaveli10/torchtrtz
Pytorch models built and saved to generate tensorrt weights file. Sibling project to use the generated trt weights file and create tensorrt engine using TensorRT C++ api.