Commit 001a2602 authored by g21589's avatar g21589 Committed by Sergio Guadarrama
Browse files

Fixed the device specification for dequeue (#1480)

This patch assigns dequeue node to inputs_device. And nolonger shows
"Ignoring device specification /device:GPU:X for node
'clone_X/fifo_queue_Dequeue'" message.
parent 2bb1baad
......@@ -450,7 +450,8 @@ def main(_):
####################
def clone_fn(batch_queue):
"""Allows data parallelism by creating multiple clones of network_fn."""
images, labels = batch_queue.dequeue()
with tf.device(deploy_config.inputs_device()):
images, labels = batch_queue.dequeue()
logits, end_points = network_fn(images)
#############################
......
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