pydoc update to match method signature

3909e4bd · Toby Boyd · 082e65c9 · 3909e4bd
Commit 3909e4bd authored Jun 08, 2017 by Toby Boyd
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tutorials/image/cifar10/cifar10_multi_gpu_train.py tutorials/image/cifar10/cifar10_multi_gpu_train.py +2 -0

No files found.
--- a/tutorials/image/cifar10/cifar10_multi_gpu_train.py
+++ b/tutorials/image/cifar10/cifar10_multi_gpu_train.py
@@ -67,6 +67,8 @@ 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.

  Returns:
     Tensor of shape [] containing the total loss for a batch of data