Method _get_state

codegeex/megatron/optimizer/optimizer.py:125–126

(self)

    # Promote state so it can be retrieved or set via
    # "optimizer_instance.state"
    def _get_state(self):
        return self.optimizer.state

    def _set_state(self, value):
        self.optimizer.state = value