create_quant_retrain_model
功能说明
根据用户设置的重训练配置文件对图结构进行量化处理,该函数在config_file指定的层完成参数重训练,并插入数据和重训练层,将修改后的网络存为新的模型文件。
约束说明
无。
函数原型
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 |
返回值 |
重训练的operation。 |
数据类型:list,列表中元素类型为tf.Operation |
返回值说明
重训练的operation。
函数输出
无。
调用示例
retrain_ops = amct.create_quant_retrain_model(graph, config_path, record_file)