def METHOD_NAME(self, batch, batch_idx):
    images, target = batch
    output = self.model(images)
    loss_train = F.cross_entropy(output, target)
    self.log("train_loss", loss_train)
    return loss_train