模型量化(4): Pytorch 量化工具包介绍

本文介绍了PyTorch的量化工具包,包括量化基础、描述符和量化器、量化模块,以及训练后量化(PTQ)和量化感知训练(QAT)。此外,还讲解了如何进行模型校准和导出ONNX模型,为深度学习模型的部署提供了量化解决方案。
摘要由CSDN通过智能技术生成

1. pytorch quantization介绍

在实际开发过程中,单独开发量化的工具进行PTQ或者QAT量化,同时去适配TensorRT, onnxruntime,openvion等推理引擎。Pytorch官方推出了量化工具:Pytorch Quantization库,方便大家使用。

在这里插入图片描述
文档: pytorch-quantization’s documentation:https://docs.nvidia.com/deeplearning/tensorrt/pytorch-quantization-toolkit/docs/index.html

  • PyTorch Quantization是一个工具包,用于训练和评估具有模拟量化的PyTorch模型。PyTorch Quantization API支持将 PyT
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

@BangBang

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值