Pytorch模型转onnx使用onnx runtime加速&Triton加速!
文章目录
- 一. ONNX简要介绍
- 二. 从PyTorch到ONNX的实战
* - 2.1. 例子1: Alexnet
– - 2.2. 例子2: SuperResolution模型
– - 三. torch.stack和torch.cat的函数区别
- 四. 参考文章
<
Original: https://blog.csdn.net/abc13526222160/article/details/123564875
Author: 布衣小张
Title: 『NLP学习笔记』Pytorch模型转onnx使用onnx runtime加速&Triton加速
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/532217/
转载文章受原作者版权保护。转载请注明原作者出处!