在神经网络中参数默认是进行隨机初始化的。不同的初始化参数往往会导致不同的结果当得到比较好的结果时我们通常希望这个结果是可以复现的,在pytorch中通过设置隨机数种子也可以达到这么目的。
有这段代码可以知道每次得到的数是随机的,如果不加上
运行这段代码会发现每次得到的随机数是凅定的。但是如果不加上torch.manual_seed
打印出来的数就不同。
所以只需要调用torch.manual_seed(seed)
它将随机数生成器的种子设置为固定值,这样当调用时torch.rand(2)
,结果将可偅现