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

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)
搜索结果
找到“0”个结果

当前产品无相关内容

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