diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-03-13 14:01:00 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-13 15:34:18 -0700 |
| commit | ea963dd048749df525d2a1f19c31d7abcdc3268e (patch) | |
| tree | 470e45466c673d8a7dbba00166a5f113d3a6279b /tensorflow/examples/learn | |
| parent | 1bbf65f8e47ec5330a743702ce79376f580ac3ec (diff) | |
Fix global_step double-counting bug in some tutorials.
Change: 149995324
Diffstat (limited to 'tensorflow/examples/learn')
| -rw-r--r-- | tensorflow/examples/learn/wide_n_deep_tutorial.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/examples/learn/wide_n_deep_tutorial.py b/tensorflow/examples/learn/wide_n_deep_tutorial.py index 09af375314..c275f53af7 100644 --- a/tensorflow/examples/learn/wide_n_deep_tutorial.py +++ b/tensorflow/examples/learn/wide_n_deep_tutorial.py @@ -128,7 +128,8 @@ def build_estimator(model_dir, model_type): model_dir=model_dir, linear_feature_columns=wide_columns, dnn_feature_columns=deep_columns, - dnn_hidden_units=[100, 50]) + dnn_hidden_units=[100, 50], + fix_global_step_increment_bug=True) return m |