search

wutianyiRosun / CGNet

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]
MIT License
262 stars 51 forks source link

I think the codes about flops dosen't compute the AdaptiveAvgPool2d 's ops! #17

Closed monk42 closed 5 years ago

monk42 commented 5 years ago

if isinstance(net, torch.nn.MaxPool2d) or isinstance(net, torch.nn.AvgPool2d): net.register_forward_hook(pooling_hook)

wutianyiRosun commented 5 years ago

＠monk42 thx u,

ABDULMAWLA commented 4 years ago

i face the same problem