HorovodとTensorflowの見積もり

Question

Horovodの例をtf.train.MonitoredTrainingSessionに置き換えて、代わりにtf.estimator.Estimatorを使用する例を拡張することはできますか？私はTensorflow 1.4.0を使用しています。

Here is an example現在のコードとよく似ています。

私はhyperoptと一緒にこれを使用したい、と私は簡単に別のハイパー、hparamsで訓練する

tf.contrib.learn.learn_runner.run(
     experiment_fn=_create_my_experiment, 
     run_config=run_config, 
     schedule="train_and_evaluate", 
     hparams=hparams) 

ような何かを行うことができますどのように好きです。これにより、トレーニングセットと検証セット用の個別のTensorboardログディレクトリが得られます。これもHorovodソリューションでも同じです。私はtf.train.SingularMonitoredSession(hooks=hooks, config=config)で遊んでいましたが、hooksにはtf.train.SummarySaverHookが含まれていましたが、トレーニングセットでうまく動作させることができました。

Answer 1

TensorFlow Estimator exampleがHorovodレポに加えられている。