Pytorch在dataloader类中设置shuffle的随机数种子方式
- 作者: 阳光T
- 来源: 51数据库
- 2020-08-06
今天小编就为大家分享一篇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的随机数种子方式就是小编分享给大家的全部内容了，希望能给大家一个参考，
推荐阅读
