Skip to content

Commit

Permalink
Merge pull request tensorflow#1542 from tensorflow/image-label-batch-…
Browse files Browse the repository at this point in the history
…rename

Rename the image/label batch variables
  • Loading branch information
nealwu authored Jun 9, 2017
2 parents 71e106f + 7d238c5 commit d4ae8d1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tutorials/image/cifar10/cifar10_multi_gpu_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,11 +171,11 @@ def train():
with tf.device('/gpu:%d' % i):
with tf.name_scope('%s_%d' % (cifar10.TOWER_NAME, i)) as scope:
# Dequeues one batch for the GPU
images, labels = batch_queue.dequeue()
image_batch, label_batch = batch_queue.dequeue()
# Calculate the loss for one tower of the CIFAR model. This function
# constructs the entire CIFAR model but shares the variables across
# all towers.
loss = tower_loss(scope, images, labels)
loss = tower_loss(scope, image_batch, label_batch)

# Reuse variables for the next tower.
tf.get_variable_scope().reuse_variables()
Expand Down

0 comments on commit d4ae8d1

Please sign in to comment.