【Python】随机种子 random/numpy/pytorch/pytorch.cuda

设立随机种子能够确保随机数的初始值在每次实验时相同;
即能确保每次实验处于相同的初始状态;


import random, torch
import numpy as np

seed = 1

random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)

其中 seed = 1 表示编号为 1 的随机值,是一组随机值;
也可以换成别的整数数字,比如 123,456 之类的;
通常不同的随机种子值代表了不同的实验组;

猜你喜欢

转载自blog.csdn.net/ao1886/article/details/109989258