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