Method train

internlm/core/engine.py:128–131

Sets the model to training mode.

(self)

        return success, grad_norm

    def train(self):
        """Sets the model to training mode."""
        self.training = True
        self._model.train()

    def eval(self):
        """Sets the model to evaluation mode."""