diff --git a/distiller/utils.py b/distiller/utils.py index 510a481d276509548c200710dfa77fb55d40a6c9..220ba3ce8a5befa8fbf4565b461cfafeca2f3a39 100755 --- a/distiller/utils.py +++ b/distiller/utils.py @@ -287,9 +287,6 @@ def sparsity_blocks(tensor, block_shape): ) view1 = tensor.view(*view_dims) - # Next, compute the sums of each column (block) - block_sums = view1.abs().sum(dim=1) - # Next, compute the sums of each column (block) block_sums = view1.abs().sum(dim=1) nonzero_blocks = len(torch.nonzero(block_sums))