Method scale_loss

codegeex/megatron/optimizer/optimizer.py:98–100

Simple scaling.

(self, loss)

        pass

    def scale_loss(self, loss):
        """Simple scaling."""
        return self.get_loss_scale() * loss

    @abstractmethod
    def step(self):