pytorch中的torch.manual_seed(myseed) 和 torch.cuda.manual_seed(myseed)

2021-05-10 12:17:34 浏览数 (2)

myseed = 45216 使用方法: 为CPU中设置种子,生成随机数 torch.manual_seed(myseed) 为特定GPU设置种子,生成随机数 torch.cuda.manual_seed(myseed) 为所有GPU设置种子,生成随机数 torch.cuda.manual_seed_all(myseed) 解释: 在实验中需要生成随机数据的时候,每次实验都需要生成数据。设置随机种子是为了确保每次生成固定的随机数,这就使得每次实验结果显示一致了,有利于实验的比较和改进。

0 人点赞