pytorch中torch.manual_seed()的理解

2020/11/12 08:25
阅读数 1K

使用🐢:

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

使用原因🐢:

  • 在需要生成随机数据的实验中,每次实验都需要生成数据。设置随机种子是为了确保每次生成固定的随机数,这就使得每次实验结果显示一致了,有利于实验的比较和改进。
展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部