RepMLP
RepMLP copied to clipboard
Why the size after average pooling of Global Perceptron be (1, 1)
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)?