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) |
父主题: 量化感知训练接口