pytorch实现保证每次运行使用的随机数都相同
更新时间：2020年02月20日 11:17:33 作者：慢行厚积
今天小编就为大家分享一篇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实现保证每次运行使用的随机数都相同就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
本文主要介绍了Python脚本修改Maya ASCII文件路径方法实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-02-02
这篇文章主要介绍了python 登录并爬取淘宝信息代码示例，具有一定借鉴价值，需要的朋友可以了解下。2017-12-12
这篇文章主要介绍了Django + Uwsgi + Nginx 的生产环境部署，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-06-06
这篇文章主要介绍了Pyecharts绘制全球流向图的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-01-01
今天小编就为大家分享一篇python 多维高斯分布数据生成方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了Python中bisect的使用方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-12-12
在Python编程中，get()函数是字典（Dictionary）对象中非常有用的函数，本文将详细介绍get()函数的用法及示例代码，感兴趣的可以了解一下2023-11-11
这篇文章主要介绍了Python使用apscheduler模块设置定时任务,APScheduler全称Advanced Python Scheduler 作用为在指定的时间规则执行指定的作业，本文对Python APScheduler 定时任务相关知识介绍的非常详细，需要的朋友参考下2022-05-05
python3使用libpcap库进行抓包及数据处理的操作方法
这篇文章主要介绍了python3使用libpcap库进行抓包及数据处理,需要的朋友可以参考下2022-10-10
这篇文章主要介绍了python实现不同数据库间数据同步功能,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-02-02
最新评论