下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

set_steps

功能说明

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

函数原型

set_steps(steps)

参数说明

参数名

输入/返回值

含义

使用限制

steps

输入

权重剪枝的步骤。

必选。

数据类型:list。

取值如下:

  • “prune_bert_intra_block”:根据输入的模型进行预训练权重裁剪。剪裁预训练权重与模型中同名但是shape不同的权重,从而使预训练权重的shape与模型一致,可单独指定。

    数据类型:string。

  • “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'])
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词