save_compressed_retrain_model
功能说明
静态组合压缩接口，根据用户最终重训练好的模型，生成最终静态组合压缩模型。
约束说明
无。
函数原型
save_compressed_retrain_model(pb_model, outputs, record_file, save_path)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
|---|---|---|---|
pb_model |
输入 |
带稀疏与量化算子的推理模式pb模型。 |
数据类型：string |
outputs |
输入 |
graph中输出算子的列表。 |
数据类型：list，列表中元素类型为string |
record_file |
输入 |
稀疏和量化因子记录文件路径及名称。 |
数据类型：string |
save_path |
输入 |
保存压缩模型的路径。 该路径需要包含模型名前缀，例如./compressed_model/*model。 |
数据类型：string |
返回值说明
无。
函数输出
压缩后的模型，该模型既可以在TensorFlow环境进行精度仿真又可以在昇腾AI处理器做推理。
调用示例
1 2 |
import amct_tensorflow as amct amct.save_compressed_retrain_model(masked_pb_path, [operation_name_1, operation_name_2], './tmp/record.txt', './pb_model/final_model') |
父主题： 组合压缩接口