Directly exploring pytorch model convert to TensorRT engine. - View it on GitHub