random.seed(rand_seed)
随机数种子,当使用random.seed(rand_seed)设定好种子之后,其中rand_seed可以是任意数字,比如10,那么每次调用生成的随机数将会是同一个。
np.random.seed(rand_seed)
numpy形式下随机种子
torch.manual_seed(rand_seed)
pytorch形式下的随机种子
torch.cuda.manual_seed(rand_seed)
为当前GPU设置随机种子
torch.cuda.manual_seed_all(rand_seed)
当有多个GPU时,为所有的GPU设置种子。