本文主要记录pytorch模型转onnx,在用onnx-tensorrt转tensorrt,并用tensorrt文件做前向推理的过程。 一、pytorch2onnx. pytorch对onnx支持比较好,能直接 ...
確定! 回上一頁