背景⌗

  • 在阅读代码过程中经常发现以下几种代码，尤其常见于pytorch书写的深度神经网络代码中
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
np.random.seed(seed)
torch.cuda.manual_seed_all()

正文⌗

  • 经查询资料得出结论，该代码作用即为将模型在初始化过程中所用到的“随机数”全部固定下来，以保证每次重新训练模型需要初始化模型参数的时候能够得到相同的初始化参数，从而达到稳定复现训练结果的目的

参考资料⌗