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的随机数种子方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Python3中在Anaconda环境下安装basemap包
今天小编就为大家分享一篇关于Python3中在Anaconda环境下安装basemap包的文章,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧2018-10-10
最新评论