Method reset

codegeex/megatron/mpu/random.py:154–157

Set to the initial state (no tracker).

(self)

        self.seeds_ = set()

    def reset(self):
        """Set to the initial state (no tracker)."""
        self.states_ = {}
        self.seeds_ = set()

    def get_states(self):
        """Get rng states. Copy the dictionary so we have direct