通道稀疏或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模型。
1 | amct.save_prune_retrain_model(masked_pb_path, [operation_name_1, operation_name_2], './tmp/record.txt', './pb_model/final_model') |