溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

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

torch.manual_seed( )怎么使用

發(fā)布時間:2021-12-27 14:18:25 來源:億速云 閱讀:359 作者:iii 欄目:大數(shù)據(jù)

這篇文章主要介紹“torch.manual_seed( )怎么使用”,在日常操作中,相信很多人在torch.manual_seed( )怎么使用問題上存在疑惑,小編查閱了各式資料,整理出簡單好用的操作方法,希望對大家解答”torch.manual_seed( )怎么使用”的疑惑有所幫助!接下來,請跟著小編一起來學(xué)習(xí)吧!

在神經(jīng)網(wǎng)絡(luò)中,參數(shù)默認(rèn)是進(jìn)行隨機(jī)初始化的。如果不設(shè)置的話每次訓(xùn)練時的初始化都是隨機(jī)的,導(dǎo)致結(jié)果不確定。如果設(shè)置初始化,則每次初始化都是固定的。

if args.seed is not None:
  random.seed(args.seed) #
  torch.manual_seed(args.seed) #為CPU設(shè)置種子用于生成隨機(jī)數(shù),以使得結(jié)果是確定的
   torch.cuda.manual_seed(args.seed) #為當(dāng)前GPU設(shè)置隨機(jī)種子;
   cudnn.deterministic = True

#如果使用多個GPU,應(yīng)該使用torch.cuda.manual_seed_all()為所有的GPU設(shè)置種子。

到此,關(guān)于“torch.manual_seed( )怎么使用”的學(xué)習(xí)就結(jié)束了,希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學(xué)習(xí),快去試試吧!若想繼續(xù)學(xué)習(xí)更多相關(guān)知識,請繼續(xù)關(guān)注億速云網(wǎng)站,小編會繼續(xù)努力為大家?guī)砀鄬嵱玫奈恼拢?/p>

向AI問一下細(xì)節(jié)

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

AI