qsin_qat

功能说明

模型量化接口，对用户提供的模型根据配置的量化参数进行量化。

函数原型

qsin_qat(model, quant_config, quant_logger)

参数说明

参数名	输入/返回值	含义	使用限制
model	输入	待量化模型实例。	必选。数据类型：PyTorch模型。
quant_config	输入	量化参数配置。	必选。数据类型：config。
quant_logger	输入	量化输出日志。	必选。数据类型：log。

参数名

输入/返回值

含义

使用限制

model

输入

待量化模型实例。

必选。

数据类型：PyTorch模型。

quant_config

输入

量化参数配置。

必选。

数据类型：config。

quant_logger

输入

量化输出日志。

必选。

数据类型：log。

调用示例

from modelslim.pytorch.quant.qat_tools import qsin_qat, QatConfig, get_logger
from torchvision.models import resnet50
import torch
model=resnet50()
quant_config = QatConfig()
quant_logger = get_logger()
model = qsin_qat(model, quant_config, quant_logger)

父主题： 量化感知训练