在循环中使用TensorFlow Estimator

时间:2018-11-18 21:35:47

标签: python tensorflow tensorflow-estimator

我已经实现了自定义估算器,现在我喜欢使用此估算器进行如下循环训练:

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

问题在于,每次在循环中调用tf.estimator.train时,都会创建一个新的MonitoredSession且永远不会将其关闭。这样,就永远不会释放分配的内存,并且有时由于内存不足异常,整个进程将被终止。无论如何,是否有必要强制关闭tf.estimator.train创建的会话?

0 个答案:

没有答案