如题: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的随机数种子方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我们。
原文地址:https://www.jb51.cc/python/535014.html
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。