I want to build classification model with global average pooling.
Should I flatten before using Sigmoid() ?
        self.main = nn.Sequential(
            nn.Conv2d(in_channels=out_channel, out_channels=1, kernel_size=1),
            nn.AvgPool2d(28, 1)
            # flatten -> sigmoid
            # sigmoid
        )