Commit d4ae8d1e authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #1542 from tensorflow/image-label-batch-rename

Rename the image/label batch variables
parents 71e106fe 7d238c5e
...@@ -171,11 +171,11 @@ def train(): ...@@ -171,11 +171,11 @@ def train():
with tf.device('/gpu:%d' % i): with tf.device('/gpu:%d' % i):
with tf.name_scope('%s_%d' % (cifar10.TOWER_NAME, i)) as scope: with tf.name_scope('%s_%d' % (cifar10.TOWER_NAME, i)) as scope:
# Dequeues one batch for the GPU # 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 # Calculate the loss for one tower of the CIFAR model. This function
# constructs the entire CIFAR model but shares the variables across # constructs the entire CIFAR model but shares the variables across
# all towers. # all towers.
loss = tower_loss(scope, images, labels) loss = tower_loss(scope, image_batch, label_batch)
# Reuse variables for the next tower. # Reuse variables for the next tower.
tf.get_variable_scope().reuse_variables() tf.get_variable_scope().reuse_variables()
......
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