aanet
aanet copied to clipboard
FLOPs for AANet
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
.