qsin_qat

功能说明

模型量化接口，对用户提供的模型根据配置的量化参数进行量化，并返回插入量化感知训练算子的模型和插入量化感知训练算子的loss。

函数原型

qsin_qat(model, loss)

参数说明

调用示例

from modelslim.pytorch.quant.qat import qsin_qat
from torchvision.models import resnet50
import torch
model=resnet50()
criterion=torch.nn.CrossEntropyLoss()
model, criterion = qsin_qat(model, criterion)