TensorRT 5.1.5.0入门 Pytorch & ONNXTensorRT中的pytorchnetwork_api_pytorch示例demoTensorRT中的ONNX问题
目录demo介绍流程问题demo介绍这个demo是在线训练了mnist的网络，然后直接用torch的nn.Module.state_dict()方法把weights取出来，填充给builder创建的trt格式的network，然后利用这个被填充完weights的network创建engine，进行推断。这个demo没有涉及到pytorch从文件中读取weights的问题。流程 # ...
