removes useless loc

tfjgeorge · Nov 20, 2023 · 0b6e9ff · 0b6e9ff
1 parent cb4343f
commit 0b6e9ff
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nngeometry/generator/jacobian/grads.py b/nngeometry/generator/jacobian/grads.py
@@ -155,7 +155,6 @@ def kfac_xx(cls, buffer, mod, layer, x, gy):
 
     @classmethod
     def kfac_gg(cls, buffer, mod, layer, x, gy):
-        spatial_locations = gy.size(2) * gy.size(3)
         os = gy.size(1)
         # DS_tilda in KFC
         DS_tilda = gy.permute(0, 2, 3, 1).contiguous().view(-1, os)