We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e9644da commit 0b7b03dCopy full SHA for 0b7b03d
torchao/prototype/sparsity/pruner/saliency_pruner.py
@@ -11,7 +11,7 @@ class SaliencyPruner(BaseStructuredSparsifier):
11
Prune rows based on the saliency (L1 norm) of each row.
12
13
This pruner works on N-Dimensional weight tensors.
14
- For each row, we will calculate the saliency, whic is the sum the L1 norm of all weights in that row.
+ For each row, we will calculate the saliency, which is the sum the L1 norm of all weights in that row.
15
We expect that the resulting saliency vector has the same shape as our mask.
16
We then pick elements to remove until we reach the target sparsity_level.
17
"""
0 commit comments