时间戳

首页 > 解决方案 > 如何修复 python 中的 rondom 生成器？每当我运行我的 CNN 时，我都会得到不同的结果

python - 如何修复 python 中的 rondom 生成器？每当我运行我的 CNN 时，我都会得到不同的结果

问题描述

我正在尝试训练和测试用于分类的 CNN 模型，每次在测试中运行代码时，我都会得到不同的准确度结果。

我怎样才能每次都得到相同的结果？python TensorFlow中是否有任何可能的解决方案来解决这个问题？

标签： pythontensorflowanacondaconv-neural-network

解决方案

试试这个：

import numpy as np
import tensorflow as tf
np.random.seed(1)
tf.set_random_seed(1)

只要您保持固定，您设置种子的值并不重要。

确保您还修复了任何第三方库的种子。随机性也可能由 GPU 库引起，如果您不使用 GPU，请不要担心。

编辑：假设您使用 TensorFlow 作为后端。对于 PyTorch（适用于 CPU 和 GPU btw），请使用：

import torch
torch.manual_seed(1)

推荐阅读