torchao.sparsity¶
使用 apply_tensor_subclass 將模型中的線性模組的權重進行稀疏化。 |
|
是 |
|
此函式模擬模型中所有線性層的 2:4 稀疏性。 |
|
Wanda 稀疏化器 |
|
一個自定義觀察器,它計算每個通道的 L2 範數並將其儲存在緩衝區中。 |
使用 apply_tensor_subclass 將模型中的線性模組的權重進行稀疏化。 |
|
是 |
|
此函式模擬模型中所有線性層的 2:4 稀疏性。 |
|
Wanda 稀疏化器 |
|
一個自定義觀察器,它計算每個通道的 L2 範數並將其儲存在緩衝區中。 |