improve name

quintenroets · Apr 9, 2024 · 548cc16 · 548cc16
1 parent 68a7f45
commit 548cc16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/revnets/evaluations/weights/standardize/order.py b/src/revnets/evaluations/weights/standardize/order.py
@@ -13,12 +13,12 @@ class Standardizer:
     neurons: InternalNeurons
 
     def run(self) -> None:
-        sort_indices = calculate_sort_indices(self.neurons.incoming)
+        sort_indices = calculate_sort_order(self.neurons.incoming)
         permute_output_weights(self.neurons.incoming, sort_indices)
         permute_input_weights(self.neurons.outgoing, sort_indices)
 
 
-def calculate_sort_indices(layer: nn.Module) -> torch.Tensor:
+def calculate_sort_order(layer: nn.Module) -> torch.Tensor:
     weights = extract_linear_layer_weights(layer)
     p = 1  # use l1-norm because l2-norm is already standardized
     sort_values = weights.norm(dim=1, p=p)