def METHOD_NAME(self, batch, _):
    x, label = batch
    prediction = self.model(x)
    preds = torch.argmax(prediction, dim=1)
    acc = torch.sum(preds == label).float() / len(label)
    loss = torch.nn.functional.nll_loss(prediction, label)
    return loss, acc