作用
torch.manual_seed()
为CPU设置种子,保证每次的随机初始化都是相同的,从而保证结果可以复现。
torch.cuda.manual_seed()
为GPU设置种子,作用同上
torch.cuda.manual_seed_all()
为所有GPU设置种子(适用于多GPU时),作用同上
例子
import torch
torch.manual_seed(2)
print(torch.rand(2))
输出
0.4360
0.1851
[torch.FloatTensor of size 2]
如果不设置manual_seed,结果就会是随机的。
参考链接
https://discuss.pytorch.org/t/what-is-manual-seed/5939/3