pytorch中torch.manual_seed()的理解

使用🐢:

  • CPU中设置种子,生成随机数
    • torch.manual_seed(number)
  • 特定GPU设置种子,生成随机数
    • torch.cuda.manual_seed(number)
  • 所有GPU设置种子,生成随机数
    • torch.cuda.manual_seed_all()

使用原因🐢:

  • 在需要生成随机数据的实验中,每次实验都需要生成数据。设置随机种子是为了确保每次生成固定的随机数,这就使得每次实验结果显示一致了,有利于实验的比较和改进。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章