API Tensorflow Estimator не работает в распределенном режиме - PullRequest
Новая
       6

API Tensorflow Estimator не работает в распределенном режиме

0 голосов
/ 23 мая 2018

вот мой тестовый код `

from tensorflow.python.keras.layers import Conv1D, MaxPooling1D
from tensorflow.python.keras.models import Model
import logging
level = logging.getLevelName('INFO')
logging.getLogger().setLevel(level)
model = tf.keras.Sequential()
output = Dense(2, activation="softmax")
model.add(Dense(64, activation="relu", input_shape=(10,)))
model.add(output)
model.compile('rmsprop', 'categorical_crossentropy')
est_model = tf.keras.estimator.model_to_estimator(keras_model=model)
train_input_fn = tf.estimator.inputs.numpy_input_fn(
        x={"dense_2_input": np.random.randint(10, size=(320, 10))},
        y=np.random.rand(320, 2),
        num_epochs=10000,
        shuffle=False)
est_model.train(train_input_fn)

Мой TF_CONFIG:` 

TF_CONFIG={
"cluster": {"chief": ["localhost:2223"], 
"worker": ["localhost:2221"], 
"ps": ["lcoalhost:2222"]}, 
"task": {"index": "0", "type": "chief"}
}

Шеф застревает при ведении журнала Restoring paramater from ...... и ни один порт не слушает.

Есть предложения?

...