2017-05-12 6 views
0

tf.contrib.learn.Estimatorをカスタムモデル機能で使用して、分散トレーニングを実行するモデルを取得しようとしています。Estimator用の分散カスタムモデル機能

Tensorflow documentationは、ジョブ(psおよび/またはworker)をtf.device関数を使用してモデルに手動で配置する必要があることを指定します。ただし、このドキュメントでは、クラスとモデル関数Estimatorを使用せずに、モデルを宣言する従来の方法を参照しています。

新しいEstimatorが「モデルの訓練と評価の複雑な分散論理のほとんどを処理する」と述べたinformationが見つかりました。これは、いろいろなworker/psジョブを手動でモデルに配置する必要があることを意味しますか?そうであれば、Tensorflowの文書のコードから(例えばtf.deviceを使用して)宣言する必要があります。

+0

エスティメーターはTFに新たに追加されたもので、処理中の作業のようです。私は分散訓練のためにそれを使用する例を見つけることができませんでした、あなたがそれを有効にすることができれば、他の人がそれを見つけることができるようにここで答えを更新することができます –

答えて

0

ビット周り掘削した後、私はthis code sampleを発見し、手動でモデル関数における種々tf.deviceを配置することなくEstimatorで作業分散TFを得ることができました。

関連する問題