Count hook for convtranspose2d is wrong

[bl0]

Hi, thanks for your great work. I found an error in the following codes:

pytorch-OpCounter/thop/count_hooks.py

Lines 66 to 71 in 41d9d15

	ops_per_element = kernel_ops + bias_ops
	# total ops
	# num_out_elements = y.numel()
	# output_elements = batch_size * out_w * out_h * cout
	ops_per_element = m.weight.nelement()

It calculates ops_per_element with two different methods. The first is kh * kw * cin, while the second is kh * kw * cin * cout. Obviously, the first one is right, but it's overwrited by the second one.

[feiyangha]

Hi, I agree with you. The flops of convtranspose2d using this method is too huge.

[Lyken17]

Thanks for pointing out. I've fixed. Let me know if you find other errors in count hook.