diff options
Diffstat (limited to 'tensorflow/python/estimator/training.py')
-rw-r--r-- | tensorflow/python/estimator/training.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/estimator/training.py b/tensorflow/python/estimator/training.py index 5730101092..f5ac79ced2 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) |