<>强如题:Pytorch在dataloader类中设置洗牌的随机数种子方式强>
虽然实验结果差别不大,但是有时候也悬殊两个百分点
想要复现实验结果
发现用到随机数的地方就是dataloader类中封装的洗牌属性
查了半天没有关于这个的设置,最后在设置随机数种子里面找到了答案
<强>以下方法即可:强>
def setup_seed(种子): torch.manual_seed(种子) torch.cuda.manual_seed_all(种子) np.random.seed(种子) random.seed(种子) torch.backends.cudnn.deterministic=True #设置随机数种子 setup_seed (20)
以上这篇Pytorch在dataloader类中设置洗牌的随机数种子方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。