更新时间:2021-03-18 GMT+08:00
分享

init

功能说明

初始化模型小型化工具,记录存储量化因子的文件,解析用户模型为图结构graph,供quantize_model和save_model使用。

函数原型

graph = init(config_file, model_file, weights_file, scale_offset_record_file)

参数说明

表1

参数名

输入/输出

含义

使用限制

config_file

输入

用户生成的量化配置文件

数据类型:string

model_file

输入

用户Caffe模型的定义文件,格式为.prototxt,与create_quant_config中的model_file保持一致

数据类型:string

weights_file

输入

用户训练好的Caffe模型文件,格式为.caffemodel,与create_quant_config中的weights_file保持一致

数据类型:string

scale_offset_record_file

输入

存储量化因子的文件,如果该文件存在,会被重写

数据类型:string

graph

输出

用户模型解析出来的图结构。

数据类型:工具自定义的数据结构Graph

返回值说明

graph: 用户模型解析出来的图结构。

函数输出

scale_offset_record_file: 存储量化因子的文件,文件如果不存在,则会被创建,否则会被清空。

重新执行量化时,该接口输出的存储量化因子文件将会被覆盖。

调用示例

from amct_caffe import init
# 初始化工具
graph = init(config_file="./configs/config.json",
             model_file="./pretrained_model/model.prototxt",
             weights_file="./pretrained_model/model.caffemodel",
             scale_offset_record_file="./recording.txt")
分享:

    相关文档

    相关产品

close