def forward(self, x):
    features = self.features(x)
    out = F.relu(features, inplace=True)
    out = F.avg_pool2d(out, kernel_size=7, stride=1).view(features.size(0), (- 1))
    out = self.classifier(out)
    return out
def forward(self, x):
    features = self.features(x)
    out = F.relu(features, inplace=True)
    out = F.avg_pool2d(out, kernel_size=7, stride=1).view(features.size(0), (- 1))
    out = self.classifier(out)
    return out