experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix wrong format of logging message.

Browse Source 
PiperOrigin-RevId: 168497373
This commit is contained in:
A. Unique TensorFlower 2017-09-12 22:55:40 -07:00 committed by TensorFlower Gardener
parent eec4f1b3ac
commit 83cbabb85c
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/contrib/learn/python/learn/estimators/head.py
View File

@ -1758,9 +1758,8 @@ def _weight_tensor(features, weight_column_name):
# We don't bother with expanding dims of non-staticly shaped tensors or
# scalars, and >1d is already in a good format.
if rank == 1:
logging.warning(
"Weights {} has shape {}, expanding to make it 2d.",
weight_column_name, shape)
logging.warning("Weights {} has shape {}, expanding to make it 2d.".
format(weight_column_name, shape))
return (
sparse_ops.sparse_reshape(weight_tensor, (-1, 1))
if isinstance(weight_tensor, sparse_tensor.SparseTensor) else