使用pytorch怎么保证每次运行的随机数相同-创新互联
使用pytorch怎么保证每次运行的随机数相同?针对这个问题,这篇文章详细介绍了相对应的分析和解答,希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。
创新互联服务项目包括台州网站建设、台州网站制作、台州网页制作以及台州网络营销策划等。多年来,我们专注于互联网行业,利用自身积累的技术优势、行业经验、深度合作伙伴关系等,向广大中小型企业、政府机构等提供互联网行业的解决方案,台州网站推广取得了明显的社会效益与经济效益。目前,我们服务的客户以成都为中心已经辐射到台州省份的部分城市,未来相信会继续扩大服务区域并继续获得客户的支持与信任!其实在代码的开头添加下面几句话即可:
# 保证训练时获取的随机数都是一样的 init_seed = 1 torch.manual_seed(init_seed) torch.cuda.manual_seed(init_seed) np.random.seed(init_seed) # 用于numpy的随机数
torch.manual_seed(seed)
为了生成随机数设置种子。返回一个torch.Generator对象
参数:
seed (int) – 期望的种子数
torch.cuda.manual_seed(seed)
为当前GPU生成随机数设置种子。如果CUDA不可用,调用该方法也是安全的;在这种情况下,该调用就会被忽略
参数:
seed (int) – 期望的种子数
如果你使用的是多GPU模型,就要调用manual_seed_all(seed).
关于使用pytorch怎么保证每次运行的随机数相同问题的解答就分享到这里了,希望以上内容可以对大家有一定的帮助,如果你还有很多疑惑没有解开,可以关注创新互联成都网站设计公司行业资讯频道了解更多相关知识。
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。
新闻标题:使用pytorch怎么保证每次运行的随机数相同-创新互联
当前地址:http://ybzwz.com/article/ceihgs.html