From 09cebc802be38533a60ee047fbb4d955499a12a0 Mon Sep 17 00:00:00 2001
From: DLPerf <88604684+DLPerf@users.noreply.github.com>
Date: Thu, 26 Aug 2021 18:00:46 +0800
Subject: [PATCH] Update input_fn.py

---
 tensorflow/vision/model/input_fn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/vision/model/input_fn.py b/tensorflow/vision/model/input_fn.py
index 8f170d5..4c7899f 100644
--- a/tensorflow/vision/model/input_fn.py
+++ b/tensorflow/vision/model/input_fn.py
@@ -73,7 +73,7 @@ def input_fn(is_training, filenames, labels, params):
         )
     else:
         dataset = (tf.data.Dataset.from_tensor_slices((tf.constant(filenames), tf.constant(labels)))
-            .map(parse_fn)
+            .map(parse_fn,num_parallel_calls=tf.data.experimental.AUTOTUNE)
             .batch(params.batch_size)
             .prefetch(1)  # make sure you always have one batch ready to serve
         )