util.py 文件源码

def set_random_seed(seed):
    global random_seed
    random_seed = seed
    np.random.seed(seed)
    torch.manual_seed(random_seed)
    if torch.cuda.is_available():
        torch.cuda.manual_seed(random_seed)