Update profile.py

e-lab · Sep 7, 2018 · 27e4a39 · 27e4a39
1 parent 7fb29af
commit 27e4a39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/profiler/profile.py b/profiler/profile.py
@@ -19,7 +19,7 @@ def count_conv2d(m, x, y):
 
     # total ops
     num_out_elements = y.numel()
-    total_ops = num_out_elements * ops * m.groups
+    total_ops = num_out_elements * ops
 
     # incase same conv is used multiple times
     m.total_ops += torch.Tensor([int(total_ops)])