
TensorRT
文章平均质量分 94
呆呆的猫
爱发呆 爱抹茶 一直在路上
展开
-
【TensorRT】2、神经网络中的量化
本文主要介绍神经网络在部署时的量化原创 2022-11-18 16:51:35 · 4555 阅读 · 0 评论 -
【TensorRT】1、TensorRT、ONNX、PyTorch 互相转换
在深度学习模型落地的过程中,会面临将模型部署到边端设备的问题,模型训练使用不同的框架,则推理的时候也需要使用相同的框架,但不同类型的平台,调优和实现起来非常困难,因为每个平台都有不同的功能和特性。如果需要在该平台上运行多种框架,则会增加复杂性,所以 ONNX 便派上了用场。可以通过将不同框架训练的模型转换成通用的 ONNX 模型,再进而转换成各个平台支持的格式,就可以实现简化部署。一、什么是 ONNXONNX 是 Open Neural Network Exchange 的简称,也叫开放神经网络交换,.原创 2022-03-21 15:32:29 · 11331 阅读 · 3 评论