热门

最新

红包

立Flag

投票

同城

我的

发布
peacefairy
peacefairy
5 年前
truepeacefairy

torch.manual_seed(1)的作用:用于设计随机种子初始化的。在深度学习中,我们在进行参数随机初始化的时候,它保证我们每次训练生成的随机数都是一样的。同时也就可以保证别人在测试你的模型时,和你的得到一个差不多的结果。
对 CPU设定种子torch.manual_seed(1)
对当前GPU设定种子torch.cuda.manual_seed(可随机生成)
对于多个GPU设定种子torch.cuda.manual_seed_all(可随机生成)

CSDN App 扫码分享
分享
评论
2
打赏
  • 复制链接
  • 举报
下一条:
亮点在 图​ ​4
立即登录