Acceleration of Pytorch Model by TensorRT

autograd
1

Now I’m doing TensorRT to speed up the Pytorch model, and you need to get the input and output relationships of each layer of the model and the parameters of each layer, thus manually building this