aanet icon indicating copy to clipboard operation
aanet copied to clipboard

FLOPs for AANet

Open gautamsreekumar opened this issue 3 years ago • 0 comments

Can you provide the FLOPs and the number of parameters in the AANet models (AANet and AANet+) for 576x960 resolution? Table 2 in the paper only gives these values of AA modules in other networks.

Also, shouldn't the below line:

https://github.com/haofeixu/aanet/blob/f689abf81ef072b4e50eb480d3a73db66eebda44/thop/count_hooks.py#L78-L83

be changed as shown below?

ops_per_element = kernel_ops* cin // m.groups + bias_ops

# total ops
# num_out_elements = y.numel()
output_elements = batch_size * out_w * out_h * cout
total_ops = output_elements * ops_per_element

This is because the bias_ops should not be multiplied by cin // m.groups.

gautamsreekumar avatar Aug 12 '21 19:08 gautamsreekumar