下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

save_prune_retrain_model

功能说明

通道稀疏或4选2结构化稀疏接口,两种稀疏特性每次只能使能一个:

  • 通道稀疏场景:根据用户最终的重训练好的带mask算子的模型,生成最终实现通道裁剪的稀疏模型,并删除mask算子。
  • 4选2结构化稀疏场景:根据用户最终的重训练好的带4选2结构化稀疏算子的模型,生成最终的稀疏模型,并删除结构化稀疏算子。

约束说明

无。

函数原型

save_prune_retrain_model(pb_model, outputs, record_file, save_path)

参数说明

参数名

输入/返回值

含义

使用限制

pb_model

输入

带有稀疏算子的推理pb模型,其参数由重训练checkpoint恢复。

数据类型:string

outputs

输入

用户模型的输出。

数据类型:list,列表中元素类型为string,例如[output1,output2,...]。

record_file

输入

记录稀疏信息的文件路径及名称。

数据类型:string

save_path

输入

模型存放路径。

该路径需要包含模型名前缀,例如./prune_model/*model

数据类型:string

返回值说明

无。

函数输出

实现稀疏的pb模型。

调用示例

amct.save_prune_retrain_model(masked_pb_path, [operation_name_1, operation_name_2], './tmp/record.txt', './pb_model/final_model')
搜索结果
找到“0”个结果

当前产品无相关内容

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