Set the seed for reproducibility.

set_seed_torch.py

def set_seed(seed: int) -> np.random.Generator:
    """Set the seed for reproducibility.

    Parameters
    ----------
    seed: int
        The seed to set for reproducibility.

    Returns
    -------
    np.random.Generator
        The random number generator for numpy.

    """
    torch.manual_seed(seed)
    torch.cuda.manual_seed_all(seed)
    random.seed(seed)
    torch.backends.cudnn.benchmark = False
    torch.backends.cudnn.deterministic = True
    os.environ["PYTHONHASHSEED"] = str(seed)
    return np.random.default_rng(seed)