源于Pytorch中的SEED

最新推荐文章于 2025-07-21 00:31:23 发布

写代码的二次猿

最新推荐文章于 2025-07-21 00:31:23 发布

阅读量897

点赞数

CC 4.0 BY-SA版权

文章标签： pytorch

本文链接：https://blog.csdn.net/imqyt/article/details/124021358

这段代码设置了随机数种子，确保在CPU和所有GPU上生成可复现的随机数序列。这对于实验和模型训练的可重复性至关重要。torch.backends.cudnn.deterministic=True进一步确保了每次运行结果的一致性。

    SEED = 1234
    random.seed(SEED)
    torch.manual_seed(SEED)#为CPU设置种子用于生成随机数，以使得结果是确定的
    #torch.cuda.manual_seed(SEED)  # 为当前GPU设置随机种子；
    torch.cuda.manual_seed_all(SEED)#如果使用多个GPU，应该使用torch.cuda.manual_seed_all()为所有的GPU设置种子。
    # torch.backends.cudnn.deterministic = True#保证每次运行结果一样