Merge pull request #1530 from xiangjinwu/patch-1

To pass test, add VariableV2 in _PSDeviceChooser

Merge pull request #1530 from xiangjinwu/patch-1
To pass test, add VariableV2 in _PSDeviceChooser
cfdbdf10 · Neal Wu · GitHub · 2fa4bb61 · 01db54e5 · cfdbdf10
Commit cfdbdf10 authored Jun 05, 2017 by Neal Wu Committed by GitHub Jun 05, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

slim/deployment/model_deploy.py slim/deployment/model_deploy.py +1 -1

No files found.
--- a/slim/deployment/model_deploy.py
+++ b/slim/deployment/model_deploy.py
@@ -663,7 +663,7 @@ class DeploymentConfig(object):
        if op.device:
          return op.device
        node_def = op if isinstance(op, tf.NodeDef) else op.node_def
-        if node_def.op == 'Variable':
+        if node_def.op.startswith('Variable'):
          t = self._task
          self._task = (self._task + 1) % self._tasks
          d = '%s/task:%d' % (self._device, t)