RepMLP icon indicating copy to clipboard operation
RepMLP copied to clipboard

Why the size after average pooling of Global Perceptron be (1, 1)

Open Lloyd-Pottiger opened this issue 3 years ago • 0 comments

https://github.com/DingXiaoH/RepMLP/blob/main/repmlpnet.py#L49

def forward(self, inputs):
        x = F.adaptive_avg_pool2d(inputs, output_size=(1, 1))
        x = self.fc1(x)

according to the paper, it may should be (h, w)?

Lloyd-Pottiger avatar Apr 05 '22 10:04 Lloyd-Pottiger