torchao.sparsity¶
使用 apply_tensor_subclass 转换模型中线性模块的权重。 |
|
别名 |
|
此函数模拟模型中所有线性层的 2:4 稀疏性。 |
|
Wanda 稀疏器 |
|
一个自定义观察器,计算每个通道的 L2 范数并将其存储在缓冲区中。 |
使用 apply_tensor_subclass 转换模型中线性模块的权重。 |
|
别名 |
|
此函数模拟模型中所有线性层的 2:4 稀疏性。 |
|
Wanda 稀疏器 |
|
一个自定义观察器,计算每个通道的 L2 范数并将其存储在缓冲区中。 |