pytorch---之隨機種子初始化

zxyhhjs2017發表於2019-03-04
PyTorch隨機

隨機種子

在訓練開始時，引數的初始化是隨機的，為了讓每次的結果一致，我們需要設定隨機種子。例如（權重初始化等）在main函式中加入如下語句：

torch.manual_seed(args.seed)#為CPU設定隨機種子 
if cuda: 
    torch.cuda.manual_seed(seed)#為當前GPU設定隨機種子 
    torch.cuda.manual_seed_all(seed)#為所有GPU設定隨機種子

轉載：https://blog.csdn.net/taolusi/article/details/83182459

相關文章