from_fixed
功能说明
Decompose类方法,模型低秩分解各层的分解率配置,以固定值的方式指定各层分解后中间层的channel数,返回自身用于链式调用。
函数原型
from_fixed(channel_fixed, excludes=None, divisor=64)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
channel_fixed |
输入 |
各层的固定分解channel数。 |
必选。 数据类型:整数,且大于0。 |
excludes |
输入 |
指定不分解的层名称。 |
可选。 数据类型:None或列表或元组。 默认值为None。 |
divisor |
输入 |
指定分解后channel的倍率,如指定16,则分解后的channel数为16的倍数。 |
可选。 数据类型:整数,且大于0,默认值为64。 说明:
divisor设置为1时,表示禁用此功能。 |
调用示例
from msmodelslim.pytorch import low_rank_decompose decomposer = low_rank_decompose.Decompose(model) # 调用__init__初始化类 decomposer = decomposer.from_fixed(64,divisor=16) # 按照fixed方式计算分解信息
父主题: Decompose