pytorch---之随机种子初始化

随机种子

在训练开始时，参数的初始化是随机的，为了让每次的结果一致，我们需要设置随机种子。例如（权重初始化等）在main函数中加入如下语句：

torch.manual_seed(args.seed)#为CPU设置随机种子 
if cuda: 
    torch.cuda.manual_seed(seed)#为当前GPU设置随机种子 
    torch.cuda.manual_seed_all(seed)#为所有GPU设置随机种子

转载：https://blog.csdn.net/taolusi/article/details/83182459