深度神经网络中seed函数
Posted Chaos
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了深度神经网络中seed函数相关的知识,希望对你有一定的参考价值。
背景
- 在阅读代码过程中经常发现以下几种代码,尤其常见于
pytorch
书写的深度神经网络代码中
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
np.random.seed(seed)
torch.cuda.manual_seed_all()
正文
- 经查询资料得出结论,该代码作用即为将模型在初始化过程中所用到的“随机数”全部固定下来,以保证每次重新训练模型需要初始化模型参数的时候能够得到相同的初始化参数,从而达到稳定复现训练结果的目的
参考资料
以上是关于深度神经网络中seed函数的主要内容,如果未能解决你的问题,请参考以下文章