python - 在循环中使用 TensorFlow Estimator

Question

我已经实现了一个自定义估计器，现在我喜欢使用这个估计器进行循环训练，如下所示：

for _ in xrange(steps):
   tf.estimator.train(training_iteration)
   post_process_results(...)

问题是每次在循环中调用 tf.estimator.train 时，都会创建一个新的 MonitoredSession 并且永远不会关闭。因此，分配的内存永远不会被释放，并且在某些时候整个进程会因为内存不足异常而被杀死。反正有没有强制关闭由 tf.estimator.train 创建的会话？

标签： pythontensorflowtensorflow-estimator