diff options
author | 2017-04-11 14:05:42 -0800 | |
---|---|---|
committer | 2017-04-11 15:26:01 -0700 | |
commit | 43acf393020106400b7fc123d76aadea62a939dd (patch) | |
tree | 21d5f7941f141a4062eb010560e042994621a1eb | |
parent | 154b2169456cf61c630a58da4b90015a32562d1c (diff) |
Update the docstring of tuner.
Change: 152866121
-rw-r--r-- | tensorflow/contrib/training/python/training/tuner.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/training/python/training/tuner.py b/tensorflow/contrib/training/python/training/tuner.py index b724c84c54..8843632619 100644 --- a/tensorflow/contrib/training/python/training/tuner.py +++ b/tensorflow/contrib/training/python/training/tuner.py @@ -29,11 +29,11 @@ class Tuner(object): Example: ``` - def _create_my_experiment(config, hparams): + def _create_my_experiment(run_config, hparams): hidden_units = [hparams.unit_per_layer] * hparams.num_hidden_layers return tf.contrib.learn.Experiment( - estimator=DNNClassifier(config=config, hidden_units=hidden_units), + estimator=DNNClassifier(config=run_config, hidden_units=hidden_units), train_input_fn=my_train_input, eval_input_fn=my_eval_input) @@ -79,7 +79,7 @@ class Tuner(object): Args: experiment_fn: A function that creates an `Experiment`. It should accept - an argument `config` which should be used to create the `Estimator` + an argument `run_config` which should be used to create the `Estimator` (passed as `config` to its constructor), and an argument `hparams`, which should be used for hyper-parameters tuning. It must return an `Experiment`. |