qsin_qat
功能说明
模型量化接口，对用户提供的模型根据配置的量化参数进行量化，并返回插入量化感知训练算子的模型和插入量化感知训练算子的loss。
函数原型
qsin_qat(model, loss)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
|---|---|---|---|
model |
输入 |
待量化模型实例。 |
必选。 数据类型：PyTorch模型。 |
loss |
输入 |
模型损失函数。 |
必选。 数据类型：function。 |
调用示例
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)
父主题： 量化感知训练
