TensorFlow Estimator: как сделать прогноз при использовании сервера параметров? - PullRequest
Новая
       42

TensorFlow Estimator: как сделать прогноз при использовании сервера параметров?

0 голосов
/ 24 сентября 2019

TensorFlow Estimator прост в использовании для распределенного обучения со стратегией сервера параметров.Но я не могу делать предсказания со стратегией сервера параметров.Я не могу найти какой-либо ресурс для представления детали.

Пример кода прогнозирования: 

    run_config = tf.estimator.RunConfig()
    model = tf.estimator.Estimator(
        model_fn=self.model_fn,
        model_dir=self._config.model_path,
        config=run_config,
        params=self.params())
    results = model.predict(
        input_fn=lambda: test_data.build(
            batch_size=self._config.eval_batch_size,
            num_epochs=1))

TF_CONFIG: 

{'task': {'index': '0', 'type': 'ps'}, 'cluster': {'chief': ['127.0.0.1:2320'], 'ps': ['127.0.0.1:2220', '127.0.0.1:2221']}}
{'task': {'index': '1', 'type': 'ps'}, 'cluster': {'chief': ['127.0.0.1:2320'], 'ps': ['127.0.0.1:2220', '127.0.0.1:2221']}}
{'task': {'index': '0', 'type': 'chief'}, 'cluster': {'chief': ['127.0.0.1:2320'], 'ps': ['127.0.0.1:2220', '127.0.0.1:2221']}}

Результат: PS и Woker сделали прогноз.

Есть предложения?Большое спасибо.

...