Pytorch在dataloader类中设置shuffle的随机数种子方式
更新时间:2020年01月14日 09:26:58 作者:附近旳人
今天小编就为大家分享一篇Pytorch在dataloader类中设置shuffle的随机数种子方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
如题:Pytorch在dataloader类中设置shuffle的随机数种子方式
虽然实验结果差别不大,但是有时候也悬殊两个百分点
想要复现实验结果
发现用到随机数的地方就是dataloader类中封装的shuffle属性


查了半天没有关于这个的设置,最后在设置随机数种子里面找到了答案
以下方法即可:
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的随机数种子方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Python基于datetime或time模块分别获取当前时间戳的方法实例
今天小编就为大家分享一篇关于Python基于datetime或time模块分别获取当前时间戳的方法实例,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧2019-02-02
使用Python的Twisted框架构建非阻塞下载程序的实例教程
Twisted的异步工作模式使其在非阻塞情况下可以拥有较高的性能,这里我们来看一下使用Python的Twisted框架构建非阻塞下载程序的实例教程,包括服务器端与客户端的实践.2016-05-05
Python+matplotlib绘制不同大小和颜色散点图实例
这篇文章主要介绍了Python+matplotlib绘制不同大小和颜色散点图实例,matplotlib的用法是比较多种多样的,下面一起看看其中的一个实例吧2018-01-01
地图可视化神器kepler.gl python接口的使用方法
这篇文章主要介绍了python 地图可视化神器kepler.gl近期重要更新的的相关资料,帮助大家利用python实现地图可视化,感兴趣的朋友可以了解下2020-12-12


最新评论