set_steps¶

功能说明¶

PruneConfig类方法，根据自定义参数配置模型剪枝的步骤。

函数原型¶

set_steps(steps)

参数说明¶

参数名	输入/返回值	含义	使用限制
steps	输入	权重剪枝的步骤。	必选。数据类型：list。取值如下： 1. “prune_bert_intra_block”：根据输入的模型进行预训练权重裁剪。裁剪预训练权重与模型中同名但是shape不同的权重，从而使预训练权重的shape与模型一致，可单独指定。数据类型：string。 2. “prune_blocks”：根据add_blocks_params()的参数进行预训练权重裁剪。将指定id的layer的权重保留到另一个layer，可单独指定。指定该步骤时，须同时配置add_blocks_params方可生效。数据类型：string。

调用示例¶

from msmodelslim.common.prune.transformer_prune.prune_model import PruneConfig
prune_config = PruneConfig()
prune_config.set_steps([ 'prune_bert_intra_block'])