[REF] Improve name of input sizes

f-dangel · Nov 3, 2023 · 75070f1 · 75070f1
1 parent e6cff8f
commit 75070f1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/singd/optim/utils.py b/singd/optim/utils.py
@@ -101,8 +101,9 @@ def _extract_averaged_patches(
 
     # compute index pattern tensors, average output dimension
     patterns = []
+    input_sizes = x.shape[-2:]
     for i, k, s, p, d in zip(
-        x.shape[-2:],
+        input_sizes,
         _pair(kernel_size),
         _pair(stride),
         (padding, padding) if isinstance(padding, str) else _pair(padding),