convert
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
√ |
|
√ |
|
x |
|
x |
|
x |
功能说明
基于torch module的训练后量化接口，将量化校准模型转换为量化部署模型。
函数原型
1 | convert(model) |
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
model |
输入/输出 |
输入含义：量化校准模型 输出含义：量化部署模型 数据类型：torch.nn.Module |
返回值说明
量化部署模型。
约束说明
无
调用示例
1 2 3 4 | # 调用前序接口生成量化校准模型 quantize(model, cfg) # 调用接口将量化校准模型转换为量化部署模型 convert(model) |
父主题： 基于torch module的量化接口