From a1a66fd4db9dfef370c89528ff98c9b6bb2753f9 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 10 Jul 2018 15:03:23 -0700 Subject: Fixing distributed version when not centering bias. PiperOrigin-RevId: 204019142 --- tensorflow/python/estimator/canned/boosted_trees.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'tensorflow/python/estimator/canned') 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) -- cgit v1.2.3