diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py b/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py index 96802a570c..d1113678a9 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py +++ b/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py @@ -191,6 +191,9 @@ def _dnn_linear_combined_model_fn(features, labels, mode, params, config=None): if not dnn_feature_columns: dnn_logits = None else: + if not dnn_hidden_units: + raise ValueError( + "dnn_hidden_units must be defined when dnn_feature_columns is specified.") dnn_partitioner = ( partitioned_variables.min_max_variable_partitioner( max_partitions=num_ps_replicas)) |