Merge pull request #12 from tfjgeorge/get_n_params

removes legacy function
tfjgeorge · Mar 3, 2021 · 50c1e21 · 50c1e21
2 parents 665c0b1 + 02ee956
commit 50c1e21
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/nngeometry/utils.py b/nngeometry/utils.py
@@ -2,10 +2,6 @@
 import torch.nn.functional as F
 
 
-def get_n_parameters(model):
-    return sum([p.numel() for p in model.parameters()])
-
-
 def per_example_grad_conv(mod, x, gy):
     ks = (mod.weight.size(2), mod.weight.size(3))
     gy_s = gy.size()