diff options
author | 2018-07-10 15:03:23 -0700 | |
---|---|---|
committer | 2018-07-10 15:18:01 -0700 | |
commit | a1a66fd4db9dfef370c89528ff98c9b6bb2753f9 (patch) | |
tree | 3ded3eb42eb1cfde61cd93dc655e8d8d01f5d92a /tensorflow/python/estimator/canned | |
parent | 0e75ab944932e562c387ffca2e19bd347ca63882 (diff) |
Fixing distributed version when not centering bias.
PiperOrigin-RevId: 204019142
Diffstat (limited to 'tensorflow/python/estimator/canned')
-rw-r--r-- | tensorflow/python/estimator/canned/boosted_trees.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/python/estimator/canned/boosted_trees.py b/tensorflow/python/estimator/canned/boosted_trees.py index a22e9745c1..3c832c7569 100644 --- a/tensorflow/python/estimator/canned/boosted_trees.py +++ b/tensorflow/python/estimator/canned/boosted_trees.py @@ -669,6 +669,8 @@ def _bt_model_fn( name='wait_until_n_batches_for_bias_accumulated') return center_bias_op + else: + return control_flow_ops.no_op() def grow_not_in_mem(): """Accumulates the data and grows a layer when ready.""" @@ -715,6 +717,8 @@ def _bt_model_fn( name='wait_until_n_batches_accumulated') return grow_model + else: + return control_flow_ops.no_op() update_model = control_flow_ops.cond( center_bias_var, center_bias_not_in_mem, grow_not_in_mem) |