gpt4 book ai didi

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

转载 作者:qq735679552 更新时间:2022-09-29 22:32:09 26 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章Pytorch在dataloader类中设置shuffle的随机数种子方式由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

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

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

想要复现实验结果 。

发现用到随机数的地方就是dataloader类中封装的shuffle属性 。

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

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

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

以下方法即可:

?
1
2
3
4
5
6
7
8
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://blog.csdn.net/u014687517/article/details/94719910 。

最后此篇关于Pytorch在dataloader类中设置shuffle的随机数种子方式的文章就讲到这里了,如果你想了解更多关于Pytorch在dataloader类中设置shuffle的随机数种子方式的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com