您好,登錄后才能下訂單哦!
這篇文章主要介紹“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>
免責(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)容。