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

create_quant_config

功能说明

训练后量化接口,根据图的结构找到所有可量化的层,自动生成量化配置文件,并将可量化层的量化配置信息写入配置文件。

函数原型

create_quant_config(config_file, model)

参数说明

参数名

输入/返回值

含义

使用限制

config_file

输入

待生成的量化配置文件存放路径及名称。

必选。

数据类型:string。

配置文件必须为.json后缀的文件。如果存放路径下已经存在该文件,则调用该接口时会覆盖已有文件。

model

输入

待量化的模型实例。

必选。

数据类型:MindSpore模型。

调用示例

from modelslim.mindspore.quant.ptq_quant.create_config import create_quant_config
model = SampleModel()
config_file = "./test_config_file.json"
create_quant_config(config_file, model)
搜索结果
找到“0”个结果

当前产品无相关内容

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