易妖游戏网
您的当前位置:首页Pytorch在dataloader类中设置shuffle的随机数种子方式

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

来源:易妖游戏网
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的随机数种⼦⽅式就是⼩编分享给⼤家的全部内容了,希望能给⼤家⼀个参考,也希望⼤家多多⽀持。

因篇幅问题不能全部显示,请点此查看更多更全内容