Function get_current_mode

internlm/core/context/random.py:95–100

Returns the current mode of the seed manager. Returns: :class:`torch.ByteTensor`: The current mode of the seed manager.

()

def get_current_mode():
    """Returns the current mode of the seed manager.
    Returns:
        :class:`torch.ByteTensor`: The current mode of the seed manager.
    """
    return _SEED_MANAGER.current_mode


def add_seed(parallel_mode: ParallelMode, seed: int, overwrite: bool = False):