如何在TensorFlow 1.4中使用早期停止训练深度神经网络?

时间:2017-11-30 22:49:24

标签: tensorflow deep-learning conv-neural-network

从卷积神经网络的训练过程中的某个时刻开始,成本函数并没有变得更好。我想定义停止训练的条件。我在TensorFlow 1.1中找到了一个使用ValidationMonitor的解决方案。但是,在TensorFlow 1.4中无法做到这一点......

我的问题:如何在TensorFlow 1.4中使用early stopping,尤其是ExperimentEstimator

请结帐我的code

1 个答案:

答案 0 :(得分:0)

由于ValidationMonitor对分布式培训效果不佳,我们决定不将其作为Hook实现。我们正在等待它的分布式友好版本。

作为一种解决方法,您可以将ValidationMonitor包装为挂钩。以下代码显示了如何执行此操作: validation_hook = tf.contrib.learn.monitors.replace_monitors_with_hooks([validation_monitor], estimator)[0]