def METHOD_NAME(self, batch: TorchData, epoch_idx: int, batch_idx: int):
    images, target = batch
    output = self.model(images)
    loss = self.criterion(output, target)
    acc1, acc5 = self.accuracy(output, target, topk=(1, 5))
    self.context.backward(loss)
    self.context.step_optimizer(self.optimizer)
    return {"loss": loss.item(), "top1": acc1[0], "top5": acc5[0]}