Method backward

internlm/core/engine.py:138–145

Starts the backward propagation given the loss value computed by a loss function. Args: loss (torch.Tensor): The loss value computed by a loss function.

(self, loss: torch.Tensor)

        self._model.eval()

    def backward(self, loss: torch.Tensor):
        """
        Starts the backward propagation given the loss value computed by a loss function.

        Args:
            loss (torch.Tensor): The loss value computed by a loss function.
        """
        return self.optimizer.backward(loss)

    def backward_by_grad(self, tensor, grad):
        """