下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

qsin_qat

功能说明

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

函数原型

qsin_qat(model, quant_config, quant_logger)

参数说明

参数名

输入/返回值

含义

使用限制

model

输入

待量化模型实例。

必选。

数据类型:PyTorch模型。

quant_config

输入

量化参数配置。

必选。

数据类型:config。

quant_logger

输入

量化输出日志。

必选。

数据类型:log。

调用示例

from msmodelslim.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)
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词