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