在使用PyTorch时,如果希望通过设置随机数种子,在gpu或cpu上固定每一次的训练结果,则需要在程序执行的开始处添加以下代码:
def setup_seed(seed):
torch.manual_seed(seed) # CPU
torch.cuda.manual_seed_all(seed) # GPU
np.random.seed(seed) # numpy
在使用PyTorch时,如果希望通过设置随机数种子,在gpu或cpu上固定每一次的训练结果,则需要在程序执行的开始处添加以下代码:
def setup_seed(seed):
torch.manual_seed(seed) # CPU
torch.cuda.manual_seed_all(seed) # GPU
np.random.seed(seed) # numpy