senet improvement
Ageliss opened this issue · comments
GUO-QING JIANG commented
line 180 of senet.py:
x = F.avg_pool2d(x, 8, 1)
will be better fitting all image sizes if changed to
x = F.avg_pool2d(x, x.size(3), 1)
GUO-QING JIANG commented
@Ageliss that's right, would you mind create a merge request? 😸
Sure, here u go.