如何保证神经网络输出确定

最新推荐文章于 2024-05-25 01:57:10 发布

zhang_wenshuo

最新推荐文章于 2024-05-25 01:57:10 发布

阅读量98

点赞数 2

文章标签：神经网络人工智能深度学习

本文链接：https://blog.csdn.net/m0_55952025/article/details/135974390

版权

种子

def set_seed(seed):
    # python&numpy:
    random.seed(seed)
    np.random.seed(seed)
    # cpu&gpu:

    torch.manual_seed(seed)  # 为cpu设置随机种子
    torch.cuda.manual_seed(seed)  # 为当前GPU设置随机种子
    torch.cuda.manual_seed_all(seed)  # 为所有GPU设置随机种子
    # cudnn:（对结果影响不大，会影响性能）

    torch.backends.cudnn.deterministic = True
    torch.backends.benchmark = False

Mode：

将神经网络设置成eval模式