Pytorch在dataloader类中设置shuffle的随机数种子

如题:Pytorch在dataloader类中设置shuffle的随机数种子
虽然实验结果差别不大,但是有时候也悬殊两个百分点
想要复现实验结果
发现用到随机数的地方就是dataloader类中封装的shuffle属性


查了半天没有关于这个的设置,最后在设置随机数种子里面找到了答案
以下方法即可:

def setup_seed(seed):
     torch.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)
     np.random.seed(seed)
     random.seed(seed)
     torch.backends.cudnn.deterministic = True
# 设置随机数种子
setup_seed(20)

原博的地址:传送门

你可能感兴趣的:(深度学习)