Finally fix transposed conv

Author: sovrasov

ptflops/pytorch_ops.py

@@ -85,8 +85,7 @@ def conv_flops_counter_hook(conv_module, input, output, extra_per_position_flops
     bias_flops = 0
 
     if conv_module.bias is not None:
-
-        bias_flops = out_channels * active_elements_count
+        bias_flops = batch_size * int(np.prod(list(output.shape[1:]), dtype=np.int64))
 
     overall_flops = overall_conv_flops + bias_flops
 

tests/common_test.py

@@ -33,6 +33,17 @@ def test_conv(self, default_input_image_size, backend: FLOPS_BACKEND):
         assert params == 3 * 3 * 2 * 3 + 2
         assert macs == 2759904
 
+    @pytest.mark.parametrize("backend", [FLOPS_BACKEND.PYTORCH, FLOPS_BACKEND.ATEN])
+    def test_conv_t(self, default_input_image_size, backend: FLOPS_BACKEND):
+        net = nn.ConvTranspose2d(3, 2, 3, stride=(2, 2), bias=True)
+        macs, params = get_model_complexity_info(net, default_input_image_size,
+                                                 as_strings=False,
+                                                 print_per_layer_stat=False,
+                                                 backend=backend)
+
+        assert params == 3 * 3 * 2 * 3 + 2
+        assert macs == 3112706
+
     @pytest.mark.parametrize("backend", [FLOPS_BACKEND.PYTORCH, FLOPS_BACKEND.ATEN])
     def test_fc(self, backend: FLOPS_BACKEND):
         net = nn.Sequential(nn.Linear(3, 2, bias=True))