run()

功能说明

运行权重压缩算法，初始化Compressor之后，通过run()函数来执行权重压缩。

函数原型

compress_result_weight, compress_result_index, compress_result_info = compressor.run(weight_transpose=False)

参数说明

参数名	输入/返回值	含义	使用限制
weight_transpose	输入	待压缩权重是否需要转置。	可选。数据类型：bool。默认为False，为不需要转置。可以设置为True，为需要转置。
compress_result_weight	返回值	压缩后的权重结果。	数据类型：dict。
compress_result_index	返回值	压缩后的索引结果。	数据类型：dict。
compress_result_info	返回值	压缩信息结果。	数据类型：dict。

调用示例

from modeslim.pytorch.weight_compression import CompressConfig, Compressor
compress_config = CompressConfig(do_pseudo_sparse=False, sparse_ratio=1, is_debug=True, compress_disable_layer=None, record_detail_root=save_root)
weight_save_path = ‘./quant_weight.npy’  # 根据实际情况修改带压缩的权重文件路径
compressor = Compressor(compress_config, weight_path)
compress_result_weight, compress_result_index, compress_result_info = compressor.run()

父主题： 权重压缩接口