首页 > 解决方案 > 在张量流估计器中复制结果

问题描述

我正在使用 tensorflow 估计器构建分类器。

但是如果多次运行,我会为同一个实验得到两个不同的结果。

这些实验具有从批量数据到网络架构的所有相同参数。

我也在估算器的配置中设置了随机种子。

还有其他我不知道的随机种子吗?

我已经检查了实验的数据方面,我确保在这些实验中发送相同的批次和相同的顺序。

训练曲线看起来很接近,但验证曲线相距甚远。如下图所示。这些是两个相同实验的损失曲线,我在同一个数据集上运行它们,在每一步创建相同的批次

在此处输入图像描述

这段代码告诉我如何设置张量流估计器的随机种子。

config = tf.estimator.RunConfig(save_summary_steps=t.save_summary_steps,
                                    log_step_count_steps=t.log_step_count_steps,
                                    save_checkpoints_steps=t.save_checkpoints_steps,
                                    keep_checkpoint_max=t.keep_checkpoint_max,
                                    tf_random_seed=t.random_number
                                    )

标签: tensorflowtensorflow-estimator

解决方案


推荐阅读