模型量化(4): Pytorch 量化工具包介绍_股票证券

创始人

2025-05-30 08:46:02

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支持将 PyTorch 模块自动转换为其量化版本。转换也可以使用 API 手动完成，这允许在不想量化所有模块的情况下进行部分量化。例如，一些层可能对量化比较敏感，对其不进行量化可提高任务精度。
PyTorch Quantization的量化模型可以直接导出到ONNX，并由TensorRT 8.0或者更高版本导入进行