count_include_pad=False

sail-sg / poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)

https://arxiv.org/abs/2111.11418

Apache License 2.0

1.29k stars 117 forks source link

Closed ydhongHIT closed 2 years ago

ydhongHIT commented 2 years ago

Hi, why do you set this argument to 'False'? The default is 'True'.

yuweihao commented 2 years ago

Hi @ydhongHIT ,

Happy Spring Festival!

Since the nn.AvgPool2d's padding is more than zero in PoolFormer, it might be more reasonable intuitively to set count_include_pad=False.