create_quant_retrain_model
功能说明
量化感知训练接口,根据用户设置的量化配置文件对图结构进行量化处理,该函数在config_file指定的层插入数据和weights伪量化层,将修改后的网络存为新的模型文件。
约束说明
无。
函数原型
create_quant_retrain_model(model_file, weights_file, config_file, modified_model_file, modified_weights_file, scale_offset_record_file)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
model_file |
输入 |
用户Caffe模型的定义文件,格式为.prototxt。 |
数据类型:string |
weights_file |
输入 |
用户训练好的的Caffe模型权重文件,格式为.caffemodel。 |
数据类型:string |
config_file |
输入 |
量化配置文件。 |
数据类型:string |
modified_model_file |
输入 |
文件名,用于存储插入量化感知训练层的Caffe模型定义文件(格式为.prototxt)。 |
数据类型:string |
modified_weights_file |
输入 |
文件名,用于存储插入量化感知训练层的Caffe模型权重文件(格式为.caffemodel)。 |
数据类型:string |
scale_offset_record_file |
输入 |
存储量化因子的文件。 |
数据类型:string |
返回值说明
无。
函数输出
- modified_model_file:修改后模型的定义文件,在原始模型上插入了量化感知训练层。
- modified_weights_file:修改后模型的权重文件,在原始模型上插入了量化感知训练层。
调用示例
1 2 3 4 5 6 7 8 |
from amct_caffe import amct model_file = 'resnet50_train.prototxt' weights_file = 'ResNet-50-model.caffemodel' modified_model_file = './tmp/modified_model.prototxt' modified_weights_file = './tmp/modified_model.caffemodel' config_json_file = './config.json' scale_offset_record_file = './record.txt' amct.create_quant_retrain_model( model_file, weights_file, config_json_file, modified_model_file, modified_weights_file, scale_offset_record_file) |
父主题: 量化感知训练