@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的随机数种子方式所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)