diff options
Diffstat (limited to 'tensorflow/python/estimator/training.py')
| -rw-r--r-- | tensorflow/python/estimator/training.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/python/estimator/training.py b/tensorflow/python/estimator/training.py index 5730101092..a01b2300dd 100644 --- a/tensorflow/python/estimator/training.py +++ b/tensorflow/python/estimator/training.py @@ -312,10 +312,10 @@ def train_and_evaluate(estimator, train_spec, eval_spec): # hidden_units=[1024, 512, 256]) # Input pipeline for train and evaluate. - def train_input_fn: # returns x, y + def train_input_fn(): # returns x, y # please shuffle the data. pass - def eval_input_fn_eval: # returns x, y + def eval_input_fn(): # returns x, y pass train_spec = tf.estimator.TrainSpec(input_fn=train_input_fn, max_steps=1000) @@ -732,7 +732,8 @@ class _TrainingExecutor(object): job_name=config.task_type, task_index=config.task_id, config=session_config, - start=False) + start=False, + protocol=config.protocol) server.start() return server |