pytorch中torch.manual_seed()的理解

使用🐢:

  • CPU中設置種子,生成隨機數
    • torch.manual_seed(number)
  • 特定GPU設置種子,生成隨機數
    • torch.cuda.manual_seed(number)
  • 所有GPU設置種子,生成隨機數
    • torch.cuda.manual_seed_all()

使用原因🐢:

  • 在需要生成隨機數據的實驗中,每次實驗都需要生成數據。設置隨機種子是爲了確保每次生成固定的隨機數,這就使得每次實驗結果顯示一致了,有利於實驗的比較和改進。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章