pytorch中的torch.manual_seed()用于生成随机数种子理解

torch.manual_seed()

在神经网络中,参数默认是进行随机初始化的。如果不设置的话每次训练时的初始化都是随机的,导致结果不确定。

如果设置初始化,则每次初始化都是固定的。

实际上,计算机并不能产生真正的随机数,而是已经编写好的一些无规则排列的数字存储在电脑里,把这些数字划分为若干相等的N份,并为每份加上一个编号,编号固定的时候,获得的随机数也是固定的。

torch.manual_seed(1)用于设置随机初始化的种子,即上述的编号,编号固定,每次获取的随机数固定。

if args.seed is not None:
      random.seed(args.seed) 
     torch.manual_seed(args.seed)  #为CPU设置种子用于生成随机数,以使结果是确定的 
     torch.cuda.manual_seed(args.seed) #为当前GPU设置随机种子;
     cudnn.deterministic = True
     
#如果使用多个GPU,应该使用torch.cuda.manual_seed_all()为所有的GPU设置种子。
  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值