如何为tensorflow神经网络模型设定种子?

2024-04-25 18:56:10 发布

您现在位置:Python中文网/ 问答频道 /正文

当我为tensorflow(v1.14)对象检测api项目运行完全相同的实验设置时,对于相同的数据,我得到了不同的结果

因此,我猜在网络配置中一定存在某种随机化

这就是我正在使用的网络配置:https://github.com/tensorflow/models/blob/master/research/object_detection/samples/configs/faster_rcnn_inception_v2_pets.config

我的第一个猜测是初始值设定项:

initializer {
        truncated_normal_initializer {
          stddev: 0.01
        }

经过一些研究,我发现应该有一个“种子”参数:https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/initializers/truncated_normal

一旦我在配置中设置了种子,我就会得到错误: ParseError:38:11:消息类型“object_detection.protos.TruncatedNormalInitializer”没有名为“seed”的字段

如何在相同的实验中得出相同的结果,并返回相同的输出,有什么线索吗


Tags: 数据项目对象https网络githubcomapi