create_prune_retrain_model
功能说明
通道稀疏或4选2结构化稀疏接口,两种稀疏特性每次只能使能一个:
- 通道稀疏场景:根据用户设置的稀疏配置文件对图结构进行稀疏处理,该函数在图结构中插入通道稀疏mask算子,达到推理时伪稀疏的效果。
- 4选2结构化稀疏:根据用户设置的稀疏配置文件对图结构进行稀疏处理,该函数在图结构中插入4选2结构化稀疏算子,达到推理时伪稀疏的效果。
约束说明
无。
函数原型
create_prune_retrain_model(graph, outputs, record_file, config_defination)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
graph |
输入 |
用户传入的待稀疏模型的tf.Graph图。 |
数据类型:tf.Graph |
outputs |
输入 |
用户模型的输出。 |
数据类型:list,列表中元素类型为string,例如[output1,output2,...]。 |
record_file |
输入 |
记录稀疏信息的文件路径及名称,记录通道稀疏结点间的级联关系或记录4选2稀疏的节点。 |
数据类型:string |
config_defination |
输入 |
用户提供的稀疏配置文件,用于指定模型tf.Graph图中各层稀疏配置情况。 基于retrain_config_tf.proto文件生成的简易配置文件prune.cfg,retrain_config_tf.proto文件所在路径为:AMCT安装目录/amct_tensorflow/proto/retrain_config_tf.proto。 retrain_config_tf.proto文件参数解释以及生成的prune.cfg稀疏配置文件样例请参见量化感知训练简易配置文件说明。 |
数据类型:string |
返回值说明
无
函数输出
无。
调用示例
amct.create_prune_retrain_model(graph, [operation_name_1, operation_name_2], './tmp/record.txt', './tmp/sample_prune.cfg')
父主题: 通道稀疏