lint/doc fixes

rockokw · Nov 6, 2017 · a97f5df · a97f5df
1 parent 1f6b3d7
commit a97f5df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/official/resnet/cifar10_main.py b/official/resnet/cifar10_main.py
@@ -142,6 +142,8 @@ def input_fn(is_training, data_dir, batch_size, num_epochs=1):
 
   Args:
     is_training: A boolean denoting whether the input is for training.
+    data_dir: The directory containing the input data.
+    batch_size: The number samples per batch.
     num_epochs: The number of epochs to repeat the dataset.
 
   Returns: