Fixed the device specification for dequeue (tensorflow#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.
envin3 · Jun 14, 2017 · 001a260 · 001a260
1 parent 2bb1baa
commit 001a260
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/slim/train_image_classifier.py b/slim/train_image_classifier.py
@@ -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)
 
       #############################