溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

Pytorch在dataloader類中設置shuffle的隨機數(shù)種子方式

發(fā)布時間：2020-09-16 07:52:24 來源：腳本之家 閱讀：685 作者：附近旳人 欄目：開發(fā)技術

如題：Pytorch在dataloader類中設置shuffle的隨機數(shù)種子方式

雖然實驗結果差別不大，但是有時候也懸殊兩個百分點

想要復現(xiàn)實驗結果

發(fā)現(xiàn)用到隨機數(shù)的地方就是dataloader類中封裝的shuffle屬性

Pytorch在dataloader類中設置shuffle的隨機數(shù)種子方式

Pytorch在dataloader類中設置shuffle的隨機數(shù)種子方式

查了半天沒有關于這個的設置，最后在設置隨機數(shù)種子里面找到了答案

以下方法即可：

def setup_seed(seed):
   torch.manual_seed(seed)
   torch.cuda.manual_seed_all(seed)
   np.random.seed(seed)
   random.seed(seed)
   torch.backends.cudnn.deterministic = True
# 設置隨機數(shù)種子
setup_seed(20)

以上這篇Pytorch在dataloader類中設置shuffle的隨機數(shù)種子方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持億速云。

向AI問一下細節(jié)

免責聲明：本站發(fā)布的內容（圖片、視頻和文字）以原創(chuàng)、轉載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權內容。

AI
助
手