pytorch实现保证每次运行使用的随机数都相同


Posted in Python onFebruary 20, 2020

其实在代码的开头添加下面几句话即可:

# 保证训练时获取的随机数都是一样的
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 相关文章推荐
python每次处理固定个数的字符的方法总结
Jan 29 Python
浅谈python抛出异常、自定义异常, 传递异常
Jun 20 Python
python实现决策树、随机森林的简单原理
Mar 26 Python
使用Flask集成bootstrap的方法
Jul 24 Python
python中partial()基础用法说明
Dec 30 Python
python3+selenium实现126邮箱登陆并发送邮件功能
Jan 23 Python
Python实现图片裁剪的两种方式(Pillow和OpenCV)
Oct 30 Python
Tensorflow训练模型越来越慢的2种解决方案
Feb 07 Python
如何写python的配置文件
Jun 07 Python
matplotlib设置颜色、标记、线条,让你的图像更加丰富(推荐)
Sep 25 Python
Django web自定义通用权限控制实现方法
Nov 24 Python
分享一个python的aes加密代码
Dec 22 Python
Python argparse模块使用方法解析
Feb 20 #Python
浅谈pytorch torch.backends.cudnn设置作用
Feb 20 #Python
Python sqlite3查询操作过程解析
Feb 20 #Python
python利用datetime模块计算程序运行时间问题
Feb 20 #Python
pytorch数据预处理错误的解决
Feb 20 #Python
Python异常继承关系和自定义异常实现代码实例
Feb 20 #Python
Python安装与卸载流程详细步骤(图解)
Feb 20 #Python
You might like
php实现通过stomp协议连接ActiveMQ操作示例
2020/02/23 PHP
告诉大家什么是JSON
2008/06/10 Javascript
javascript 遍历验证所有文本框的值
2009/08/27 Javascript
在JavaScript中获取请求的URL参数
2010/12/22 Javascript
jQuery常见开发技巧详细整理
2013/01/02 Javascript
关于JavaScript中name的意义冲突示例介绍
2014/05/29 Javascript
Javascript中的Callback方法浅析
2015/03/15 Javascript
jQuery插件EnPlaceholder实现输入框提示文字
2015/06/05 Javascript
javascript实现别踩白块儿小游戏程序
2015/11/22 Javascript
jquery实现两边飘浮可关闭的对联广告
2015/11/27 Javascript
详解javascript的变量与标识符
2016/01/04 Javascript
javascript每日必学之多态
2016/02/23 Javascript
js实现页面跳转的几种方法小结
2016/05/16 Javascript
JS实现关闭当前页而不弹出提示框的方法
2016/06/22 Javascript
Javascript 基础---Ajax入门必看
2016/07/06 Javascript
jquery.serialize() 函数语法及简单实例
2016/07/08 Javascript
js 动态给元素添加、移除事件的实现方法
2016/07/19 Javascript
mui 打开新窗口的方式总结及注意事项
2017/08/20 Javascript
通过jquery获取上传文件名称、类型和大小的实现代码
2018/04/19 jQuery
JavaScript实现无限轮播效果
2020/11/19 Javascript
利用python的socket发送http(s)请求方法示例
2018/05/07 Python
python抓取网站的图片并下载到本地的方法
2018/05/22 Python
Python中存取文件的4种不同操作
2018/07/02 Python
python控制台实现tab补全和清屏的例子
2019/08/20 Python
Python模拟FTP文件服务器的操作方法
2020/02/18 Python
萨克斯第五大道英国:Saks Fifth Avenue英国
2019/04/01 全球购物
印尼购物网站:iLOTTE
2019/10/16 全球购物
2019年Java面试必问之经典试题
2012/09/12 面试题
在职党员进社区活动总结
2014/07/05 职场文书
师德师风学习材料
2014/12/19 职场文书
年度考核个人总结
2015/03/06 职场文书
2015年感恩母亲节的演讲稿
2015/03/18 职场文书
小学大队委竞选口号
2015/12/25 职场文书
2016年法制宣传月活动总结
2016/04/01 职场文书
医生行业员工的辞职信
2019/06/24 职场文书
MySQL锁机制
2021/04/05 MySQL