def METHOD_NAME(self, batch, optimizer):
    self.model.train()
    optimizer.zero_grad()
    model_result = self._batch_pass(batch, self.model)
    loss = self._loss(**model_result)
    loss.backward()
    optimizer.step()
    return loss.item()