Skip to content

Commit

Permalink
Fixed the device specification for dequeue (#1480)
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
g21589 authored and sguada committed Jun 14, 2017
1 parent 2bb1baa commit 001a260
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion slim/train_image_classifier.py
Expand Up @@ -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)

#############################
Expand Down

0 comments on commit 001a260

Please sign in to comment.