I want to use Dropout as a regularization.
I have ResNet Encoder and GAP Classifier.
If I add one Dropout(), where should I insert this?
class Classifier(nn.Module):
    def __init__(self):
        super(Classifier, self).__init__()
        self.conv = nn.Conv2d(64, kernel_size=1)
        self.avgpool = nn.AvgPool2d(8)

    def forward(self, x):
        x = self.conv(x)
        x = self.avgpool(x)
        x = x.view(x.size(0), -1)
        return x