set_steps
功能说明
PruneConfig类方法,根据自定义参数配置模型剪枝的步骤。
函数原型
set_steps(steps)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
steps |
输入 |
权重剪枝的步骤。 |
必选。 数据类型:list。 取值如下:
|
调用示例
from msmodelslim.common.prune.transformer_prune.prune_model import PruneConfig prune_config = PruneConfig() prune_config.set_steps([ 'prune_bert_intra_block'])
父主题: PruneConfig