python怎么设置随机数种子_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)

以上这篇Pytorch在DataLoader类中设置shuffle的随机数种子方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持我们。

总结

如果觉得编程之家网站内容还不错，欢迎将编程之家网站推荐给程序员好友。

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。

如您喜欢交流学习经验，点击链接加入交流1群：1065694478(已满)交流2群：163560250