fix lint errors

guptapriya · guptapriya · commit 2b47f77bfd26 · 2018-05-05T00:13:08.000-07:00
diff --git a/official/resnet/cifar10_main.py b/official/resnet/cifar10_main.py
@@ -106,7 +106,8 @@ def preprocess_image(image, is_training):
   return image
 
 
-def input_fn(is_training, data_dir, global_batch_size, num_epochs=1, num_gpus=1):
+def input_fn(is_training, data_dir, global_batch_size, num_epochs=1,
+             num_gpus=1, datasets_num_private_threads=None):
   """Input_fn using the tf.data input pipeline for CIFAR-10 dataset.
 
   Args:
@@ -115,7 +116,7 @@ def input_fn(is_training, data_dir, global_batch_size, num_epochs=1, num_gpus=1)
     global_batch_size: The number of samples per batch.
     num_epochs: The number of epochs to repeat the dataset.
     num_gpus: The number of GPUs.
-    datasets_num_private_threads: Number of threads for a private 
+    datasets_num_private_threads: Number of threads for a private
       threadpool created for all datasets computation.
 
 
diff --git a/official/resnet/imagenet_main.py b/official/resnet/imagenet_main.py
@@ -165,7 +165,7 @@ def input_fn(is_training, data_dir, global_batch_size, num_epochs=1,
     global_batch_size: The number of samples per batch.
     num_epochs: The number of epochs to repeat the dataset.
     num_gpus: The number of GPUs.
-    datasets_num_private_threads: Number of threads for a private 
+    datasets_num_private_threads: Number of threads for a private
       threadpool created for all datasets computation.
 
   Returns:
diff --git a/official/resnet/resnet_run_loop.py b/official/resnet/resnet_run_loop.py
@@ -72,7 +72,7 @@ def process_record_dataset(dataset, is_training, global_batch_size,
     # dataset for the appropriate number of epochs.
     # Using the fused shuffle_and_repeat method gives better performance.
     dataset = dataset.apply(tf.contrib.data.shuffle_and_repeat(
-        buffer_size=shuffle_buffer, num_epochs))
+        buffer_size=shuffle_buffer, count=num_epochs))
   else:
     dataset = dataset.repeat(num_epochs)
 
diff --git a/official/utils/flags/_performance.py b/official/utils/flags/_performance.py
@@ -135,7 +135,7 @@ def _check_loss_scale(loss_scale):  # pylint: disable=unused-variable
     flags.DEFINE_string(
         name="tf_gpu_thread_mode", short_name="gt_mode", default="global",
         help=help_wrap(
-          "Whether and how the GPU device uses its own threadpool.")
+             "Whether and how the GPU device uses its own threadpool.")
     )
 
   if tf_gpu_thread_count:
@@ -146,11 +146,11 @@ def _check_loss_scale(loss_scale):  # pylint: disable=unused-variable
 
   if datasets_num_private_threads:
     flags.DEFINE_integer(
-        name="datasets_num_private_threads", short_name="dataset_thread_count", 
+        name="datasets_num_private_threads", short_name="dataset_thread_count",
         default=None,
         help=help_wrap(
-          "Number of threads for a private threadpool created for all datasets"
-          "computation..")
+             "Number of threads for a private threadpool created for all"
+             "datasets computation..")
     )
 
   return key_flags
