Использование TensorFlow Estimator в цикле - PullRequest
0 голосов
/ 19 ноября 2018

Я реализовал пользовательский оценщик, и теперь мне нравится использовать этот оценщик для обучения в цикле следующим образом:

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

Проблема состоит в том, что каждый раз, когда tf.estimator.train вызывается вloop новый MonitoredSession создается и никогда не закрывается.Таким образом, выделенная память никогда не освобождается, и в какой-то момент весь процесс уничтожается из-за нехватки памяти.Есть ли способ принудительно закрыть сеанс, создаваемый tf.estimator.train?

...