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的随机数种子方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
在Keras中利用np.random.shuffle()打乱数据集实例
这篇文章主要介绍了在Keras中利用np.random.shuffle()打乱数据集实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06Python cookbook(数据结构与算法)根据字段将记录分组操作示例
这篇文章主要介绍了Python cookbook(数据结构与算法)根据字段将记录分组操作,结合实例形式分析了itertools.groupby()函数针对字典进行分组操作的相关实现技巧,需要的朋友可以参考下2018-03-03
最新评论