从卷积神经网络的训练过程中的某个时刻开始,成本函数并没有变得更好。我想定义停止训练的条件。我在TensorFlow 1.1中找到了一个使用ValidationMonitor的解决方案。但是,在TensorFlow 1.4中无法做到这一点......
我的问题:如何在TensorFlow 1.4中使用early stopping,尤其是Experiment和Estimator?
请结帐我的code
答案 0 :(得分:0)
由于ValidationMonitor
对分布式培训效果不佳,我们决定不将其作为Hook实现。我们正在等待它的分布式友好版本。
作为一种解决方法,您可以将ValidationMonitor
包装为挂钩。以下代码显示了如何执行此操作:
validation_hook = tf.contrib.learn.monitors.replace_monitors_with_hooks([validation_monitor], estimator)[0]