def body(self, x=None):
    if (not isinstance(x, torch.cuda.FloatTensor)):
        x = tensor(x)
    return self.features(x)
def body(self, x=None):
    if (not isinstance(x, torch.cuda.FloatTensor)):
        x = tensor(x)
    return self.features(x)