GuanLianzheng / pytorch_to_TensorRT5

通过pytorch搭建卷积神经网络完成手写识别任务,并将训练好的模型以多种方式部署到TensorRT中加速。(1)通过加载权重。(2)通过加载onnx。(3)通过加载engine。
45Updated 5 years ago

Alternatives and similar repositories for pytorch_to_TensorRT5:

Users that are interested in pytorch_to_TensorRT5 are comparing it to the libraries listed below