Skip to content

Commit

Permalink
Code cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
nealwu authored Jun 8, 2017
1 parent 9e8fd6d commit b5acc00
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions tutorials/image/cifar10/cifar10_multi_gpu_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,14 +67,13 @@ def tower_loss(scope, images, labels):
Args:
scope: unique prefix string identifying the CIFAR tower, e.g. 'tower_0'
images: Images. 4D tensor of [batch_size, height, width, 3] size.
labels: Labels. 1D tensor of [batch_size] size.
images: Images. 4D tensor of shape [batch_size, height, width, 3].
labels: Labels. 1D tensor of shape [batch_size].
Returns:
Tensor of shape [] containing the total loss for a batch of data
"""


# Build inference Graph.
logits = cifar10.inference(images)

Expand Down Expand Up @@ -138,7 +137,6 @@ def average_gradients(tower_grads):


def train():
print(FLAGS.batch_size)
"""Train CIFAR-10 for a number of steps."""
with tf.Graph().as_default(), tf.device('/cpu:0'):
# Create a variable to count the number of train() calls. This equals the
Expand Down

0 comments on commit b5acc00

Please sign in to comment.