def METHOD_NAME(self, batch: TorchData) -> Dict[str, Any]:
    input, target = batch
    logits = self.model(input)
    loss = self.model._loss(input, target)
    top1, top5 = accuracy(logits, target, topk=(1, 5))
    return {"loss": loss, "top1_accuracy": top1, "top5_accuracy": top5}