def forward(self, x, return_feature=False):
    f = self.backbone(x)
    if (self.head is not None):
        f = self.head(f)
    if (self.classifier is None):
        return f
    y = self.classifier(f)
    if return_feature:
        return (y, f)
    return y
def forward(self, x, return_feature=False):
    f = self.backbone(x)
    if (self.head is not None):
        f = self.head(f)
    if (self.classifier is None):
        return f
    y = self.classifier(f)
    if return_feature:
        return (y, f)
    return y