np.random.seed(), torch.manual_seed(args.seed)

np.random.seed()

np.random.seed()函数用于生成指定随机数

        seed()被设置了之后,np,random.random()可以按顺序产生一组固定的数组;

        如果使用相同的seed()值,则每次生成的随机数都相同,如果不设置这个值,那么每次生成的随机数不同;

        只在调用的时候seed()一下并不能使生成的随机数相同,需要每次调用都seed()一下,表示种子相同,从而生成的随机数相同。

>>> import random

# 随机数不一样
>>> random.seed()
>>> print('随机数1:',random.random())
>>> random.seed()
>>> print('随机数2:',random.random())

# 随机数一样
>>> random.seed(1)
>>> print('随机数3:',random.random())
>>> random.seed(1)
>>> print('随机数4:',random.random())
>>> random.seed(2)
>>> print('随机数5:',random.random())

'''
随机数1: 0.7643602170615428
随机数2: 0.31630323818329664
随机数3: 0.13436424411240122
随机数4: 0.13436424411240122
随机数5: 0.9560342718892494
'''

>>> import numpy as np

>>> np.random.seed(1)
>>> L1 = np.random.randn(3, 3)
>>> np.random.seed(1)
>>> L2 = np.random.randn(3, 3)
>>> print(L1)
>>> print(L2)
# 结果
[[ 1.62434536 -0.61175641 -0.52817175]
 [-1.07296862  0.86540763 -2.3015387 ]
 [ 1.74481176 -0.7612069   0.3190391 ]]
 
[[ 1.62434536 -0.61175641 -0.52817175]
 [-1.07296862  0.86540763 -2.3015387 ]
 [ 1.74481176 -0.7612069   0.3190391 ]]

torch.manual_seed(args.seed)

        torch.manual_seed(args.seed)  

        为CPU设置种子用于生成随机数,以使得结果是确定的, 方便下次复现;

        随机种子作用域是在设置时到下一次设置时

torch.manual_seed(seed) → torch._C.Generator
  1. seed,int类型,是种子 – CPU生成随机数的种子。取值范围为 [-0x8000000000000000, 0xffffffffffffffff] ,十进制是 [-9223372036854775808, 18446744073709551615] ,超出该范围将触发 RuntimeError 报错。

# 为CPU中设置种子,生成随机数:
torch.manual_seed(number)

#为特定GPU设置种子,生成随机数:
torch.cuda.manual_seed(number)

#为所有GPU设置种子,生成随机数:
# 如果使用多个GPU,应该使用torch.cuda.manual_seed_all()为所有的GPU设置种子。
torch.cuda.manual_seed_all(number)

参考:【PyTorch】torch.manual_seed() 详解_想变厉害的大白菜的博客-CSDN博客_torch.manual_seed

  • 5
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值