"docs/git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "d678357f06ff45eaf020f31c0147cd4c113347de"
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(_): ...@@ -450,7 +450,8 @@ def main(_):
#################### ####################
def clone_fn(batch_queue): def clone_fn(batch_queue):
"""Allows data parallelism by creating multiple clones of network_fn.""" """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) 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