2017-08-02 8 views
0

私のモデルのカプセル化オブジェクトとしてtf.estimator.Estimatorを使用するようにmyに書き直しています。 問題は次のとおりです。 典型的な入力パイプラインがどのようにピクチャに収まるかわかりません。QueensRunnerを使用したTensorflow tf.estimator.Estimator

私の入力パイプラインは、tf.train.Coordinatorによって調整されたキューを使用します。

私が呼び出すとき推定器に渡されるinit_fn機能では、すべての "入力グラフ" を作成tf.estimator.Estimator要件をsatisifyするには、次の

...create input graph... 
qr = tf.train.QueueRunner(queue, [operations...]) 
tf.train.add_queue_runner(qr) 

問題:

Estimator.train(...) 

それはこの input_fn(f)のように見えますこのようなシナリオでは、Estimator.train(...)の先頭と末尾に、どのようにしてqueue runnersを開始して停止することができますか?

開始私が見積もりに渡された足場オブジェクトにそれをしない私が渡すことができますキューとinit_fnを開始するために考え出した

スレッドを結合して正常に閉じる方法 - これはわかりません。

tf.estimatorを使用する場合、適切なスレッド化入力パイプラインの参照アーキテクチャはありますか?

Estimatorクラスでも、キューを処理する準備ができていますか?

答えて

1

Estimatorは、開始スレッドと結合スレッドを処理するtf.train.MonitoredTrainingSessionを使用します。例えば、 tf.estimator.inputs.*tf.contrib.learn.io.read*

など、いくつかの例の入力-fnを確認することができます
関連する問題