save_prune_retrain_model

通道稀疏接口：通道稀疏场景：根据用户最终的重训练好的带mask算子的模型，生成最终实现通道裁剪的稀疏模型，并删除mask算子。

无。

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')

父主题： 通道稀疏