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

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

概述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 @H_403_0@如题:Pytorch在DataLoader类中设置shuffle的随机数种子方式

@H_403_0@虽然实验结果差别不大,但是有时候也悬殊两个百分点

@H_403_0@想要复现实验结果

@H_403_0@发现用到随机数的地方就是DataLoader类中封装的shuffle属性

@H_403_0@


@H_403_0@


@H_403_0@查了半天没有关于这个的设置,最后在设置随机数种子里面找到了答案

@H_403_0@以下方法即可:

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)
@H_403_0@以上这篇Pytorch在DataLoader类中设置shuffle的随机数种子方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我们。

@H_502_35@ 总结

以上是内存溢出为你收集整理的Pytorch在dataloader类中设置shuffle的随机数种子方式全部内容,希望文章能够帮你解决Pytorch在dataloader类中设置shuffle的随机数种子方式所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/1199368.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-06-04
下一篇 2022-06-04

发表评论

登录后才能评论

评论列表(0条)

保存