Skip to content

GitLab

Commit 72587577 authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change. 

PiperOrigin-RevId: 313798102
parent 8479cc7c
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
official/vision/detection/dataloader/input_reader.py
View file @ 72587577
...@@ -95,8 +95,6 @@ class InputFn(object): ...@@ -95,8 +95,6 @@ class InputFn(object):
num_parallel_calls=tf.data.experimental.AUTOTUNE) num_parallel_calls=tf.data.experimental.AUTOTUNE)
if self._is_training: if self._is_training:
# Large shuffle size is critical for 2vm input pipeline. Can use small
# value (e.g. 64) for 1vm.
dataset = dataset.shuffle(1000) dataset = dataset.shuffle(1000)
if self._num_examples > 0: if self._num_examples > 0:
dataset = dataset.take(self._num_examples) dataset = dataset.take(self._num_examples)
......
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