pytorch中torch.manual_seed()的理解


使用 :

為CPU中設置種子,生成隨機數:

torch.manual_seed(number)

為特定GPU設置種子,生成隨機數:

torch.cuda.manual_seed(number)

為所有GPU設置種子,生成隨機數:

torch.cuda.manual_seed_all(number)

使用原因 :

在需要生成隨機數據的實驗中,每次實驗都需要生成數據。設置隨機種子是為了確保每次生成固定的隨機數,這就使得每次實驗結果顯示一致了,有利於實驗的比較和改進。使得每次運行該 .py 文件時生成的隨機數相同。

參考博客:https://zhuanlan.zhihu.com/p/161575780


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM