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

Another missed _ref in sync_replicas_optimizer.

Browse Source 
Change: 139491940
This commit is contained in:
A. Unique TensorFlower 2016-11-17 12:24:47 -08:00 committed by TensorFlower Gardener
parent 61229c2992
commit 4dcd9c50c7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/training/sync_replicas_optimizer.py
View File

@ -955,7 +955,7 @@ class SyncReplicasOptimizer(optimizer.Optimizer):
if num_tokens > 0:
with ops.device(self._global_step.device), ops.name_scope(""):
tokens = array_ops.fill([num_tokens],
self._global_step.ref())
self._global_step._ref()) # pylint: disable=protected-access
init_tokens = self._sync_token_queue.enqueue_many((tokens,))
else:
init_tokens = control_flow_ops.no_op(name="no_init_tokens")