Skip to content

GitLab

Commit 01db54e5 authored by xiangjinwu's avatar xiangjinwu Committed by GitHub
Browse files

To pass test, add VariableV2 in _PSDeviceChooser

parent 2fa4bb61
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
slim/deployment/model_deploy.py
View file @ 01db54e5
...@@ -663,7 +663,7 @@ class DeploymentConfig(object): ...@@ -663,7 +663,7 @@ class DeploymentConfig(object):
if op.device: if op.device:
return op.device return op.device
node_def = op if isinstance(op, tf.NodeDef) else op.node_def 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 t = self._task
self._task = (self._task + 1) % self._tasks self._task = (self._task + 1) % self._tasks
d = '%s/task:%d' % (self._device, t) d = '%s/task:%d' % (self._device, t)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment