torch.manual_seed()及其作用

含義

torch.manual_seed(parameter) #爲CPU設置種子用於生成隨機數,以使得結果是確定的
torch.cuda.manual_seed(parameter)#爲當前GPU設置隨機種
torch.cuda.manual_seed_all(parameter)#如果使用多個GPU,應該使用爲所有的GPU設置種子。

其中parameter爲int型變量

作用

設置唯一確定隨機數種子,確保隨機數種子不變,使得程序每次使用random函數均可獲得同一隨機值

參考By Florence_Janie

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章