文档
注册
评分
提单
论坛
小AI

create_quant_retrain_model

功能说明

量化感知训练接口,根据用户设置的量化配置文件对图结构进行量化处理,该函数在config_file指定的层插入数据和weights伪量化层,将修改后的网络存为新的模型文件。

约束说明

无。

函数原型

retrain_ops = create_quant_retrain_model(graph, config_file, record_file)

参数说明

参数名

输入/返回值

含义

使用限制

graph

输入

用户传入的待量化模型的tf.Graph图。

数据类型:tf.Graph

config_file

输入

用户生成的量化感知训练配置文件,用于指定模型tf.Graph图中量化感知训练层的配置情况。

数据类型:string

record_file

输入

量化因子记录文件路径。

数据类型:string

retrain_ops

返回值

量化感知训练新增的算子变量列表。

数据类型:list,列表中元素类型为tf.Variable

返回值说明

量化感知训练新增的层名变量列表。

函数输出

无。

调用示例

1
retrain_ops = amct.create_quant_retrain_model(graph, config_path, record_file)
搜索结果
找到“0”个结果

当前产品无相关内容

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