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

Remove TODO(eddz) from TF warm-starting code.

Browse Source 
PiperOrigin-RevId: 252937921
This commit is contained in:
Eddie Zhou 2019-06-12 17:58:19 -07:00 committed by TensorFlower Gardener
parent 884a009807
commit 30b29861dc
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/training/warm_starting_util.py
View File

@ -256,7 +256,6 @@ def _warm_start_var_with_vocab(var,
# Assume tensor name remains the same. # Assume tensor name remains the same.
prev_tensor_name = _infer_var_name(var) prev_tensor_name = _infer_var_name(var)
# TODO(eddz): Fix functionality for rank-1 Variables (like FC biases).
total_v_first_axis = sum(v.get_shape().as_list()[0] for v in var) total_v_first_axis = sum(v.get_shape().as_list()[0] for v in var)
for v in var: for v in var:
v_shape = v.get_shape().as_list() v_shape = v.get_shape().as_list()