def forward(self, x):
    if self.normalize:
        x = normalize_imagenet(x)
    net = self.features(x)
    out = self.fc(net)
    return out
def forward(self, x):
    if self.normalize:
        x = normalize_imagenet(x)
    net = self.features(x)
    out = self.fc(net)
    return out