Pytorch在dataloader類中設(shè)置shuffle的隨機數(shù)種子方式
如題:Pytorch在dataloader類中設(shè)置shuffle的隨機數(shù)種子方式
雖然實驗結(jié)果差別不大,但是有時候也懸殊兩個百分點
想要復(fù)現(xiàn)實驗結(jié)果
發(fā)現(xiàn)用到隨機數(shù)的地方就是dataloader類中封裝的shuffle屬性
查了半天沒有關(guān)于這個的設(shè)置,最后在設(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è)置隨機數(shù)種子 setup_seed(20)
以上這篇Pytorch在dataloader類中設(shè)置shuffle的隨機數(shù)種子方式就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python自動化爬取天眼查數(shù)據(jù)的實現(xiàn)
本文將結(jié)合實例代碼,介紹Python自動化爬取天眼查數(shù)據(jù)的實現(xiàn),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-06-06Python3中函數(shù)參數(shù)傳遞方式實例詳解
這篇文章主要介紹了Python3中函數(shù)參數(shù)傳遞方式,結(jié)合實例形式較為詳細(xì)的分析了Python3中函數(shù)參數(shù)傳遞的常見操作技巧,需要的朋友可以參考下2019-05-05PyTorch實現(xiàn)卷積神經(jīng)網(wǎng)絡(luò)的搭建詳解
這篇文章主要為大家介紹了PyTorch實現(xiàn)卷積神經(jīng)網(wǎng)絡(luò)的搭建詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2022-05-05Python3中在Anaconda環(huán)境下安裝basemap包
今天小編就為大家分享一篇關(guān)于Python3中在Anaconda環(huán)境下安裝basemap包的文章,小編覺得內(nèi)容挺不錯的,現(xiàn)在分享給大家,具有很好的參考價值,需要的朋友一起跟隨小編來看看吧2018-10-10keras自動編碼器實現(xiàn)系列之卷積自動編碼器操作
這篇文章主要介紹了keras自動編碼器實現(xiàn)系列之卷積自動編碼器操作,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-07-07