diff options
-rw-r--r-- | tensorflow/contrib/tensor_forest/kernels/v4/grow_stats.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensor_forest/kernels/v4/grow_stats.cc b/tensorflow/contrib/tensor_forest/kernels/v4/grow_stats.cc index 81b4534f10..63bfc1aef1 100644 --- a/tensorflow/contrib/tensor_forest/kernels/v4/grow_stats.cc +++ b/tensorflow/contrib/tensor_forest/kernels/v4/grow_stats.cc @@ -207,7 +207,8 @@ void ClassificationStats::AddExample( } void ClassificationStats::CheckPrune() { - if (IsFinished() || weight_sum_ < prune_sample_epoch_ * prune_check_every_) { + if (params_.pruning_type().type() == SPLIT_PRUNE_NONE || IsFinished() || + weight_sum_ < prune_sample_epoch_ * prune_check_every_) { return; } ++prune_sample_epoch_; |