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)文章
對numpy中數(shù)組轉(zhuǎn)置的求解以及向量內(nèi)積計算方法
今天小編就為大家分享一篇對numpy中數(shù)組轉(zhuǎn)置的求解以及向量內(nèi)積計算方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10
python實現(xiàn)在函數(shù)中修改變量值的方法
今天小編就為大家分享一篇python實現(xiàn)在函數(shù)中修改變量值的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
Python基于datetime或time模塊分別獲取當(dāng)前時間戳的方法實例
今天小編就為大家分享一篇關(guān)于Python基于datetime或time模塊分別獲取當(dāng)前時間戳的方法實例,小編覺得內(nèi)容挺不錯的,現(xiàn)在分享給大家,具有很好的參考價值,需要的朋友一起跟隨小編來看看吧2019-02-02
使用Python的Twisted框架構(gòu)建非阻塞下載程序的實例教程
Twisted的異步工作模式使其在非阻塞情況下可以擁有較高的性能,這里我們來看一下使用Python的Twisted框架構(gòu)建非阻塞下載程序的實例教程,包括服務(wù)器端與客戶端的實踐.2016-05-05
Python+matplotlib繪制不同大小和顏色散點圖實例
這篇文章主要介紹了Python+matplotlib繪制不同大小和顏色散點圖實例,matplotlib的用法是比較多種多樣的,下面一起看看其中的一個實例吧2018-01-01
地圖可視化神器kepler.gl python接口的使用方法
這篇文章主要介紹了python 地圖可視化神器kepler.gl近期重要更新的的相關(guān)資料,幫助大家利用python實現(xiàn)地圖可視化,感興趣的朋友可以了解下2020-12-12

