Horovod 및 Tensorflow 측정기

Question

Horovodtf.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 repo에 추가되었습니다.