From 001a260214ba34f36e149bbd24f7f5d6a6634500 Mon Sep 17 00:00:00 2001 From: g21589 Date: Thu, 15 Jun 2017 02:12:54 +0800 Subject: [PATCH] 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. --- slim/train_image_classifier.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/slim/train_image_classifier.py b/slim/train_image_classifier.py index 57049a1a23..21180edb9e 100755 --- 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) #############################