From 5c1dd0688db9b52de9544fdc11ec41f5fd919733 Mon Sep 17 00:00:00 2001
From: DLPerf <88604684+DLPerf@users.noreply.github.com>
Date: Fri, 27 Aug 2021 20:03:10 +0800
Subject: [PATCH] Update textClassiferModel.py

---
 lessonEight/textClassiferModel.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lessonEight/textClassiferModel.py b/lessonEight/textClassiferModel.py
index a21d560..5bc190f 100644
--- a/lessonEight/textClassiferModel.py
+++ b/lessonEight/textClassiferModel.py
@@ -113,7 +113,7 @@ def train_input_fn():
     dataset = tf.data.Dataset.from_tensor_slices((x_train, x_len_train, y_train))
     dataset = dataset.shuffle(buffer_size=len(x_train_variable))
     dataset = dataset.batch(100)
-    dataset = dataset.map(parser)
+    dataset = dataset.map(parser,num_parallel_calls=tf.data.experimental.AUTOTUNE)
     dataset = dataset.repeat()
     iterator = dataset.make_one_shot_iterator()
     return iterator.get_next()
@@ -121,7 +121,7 @@ def train_input_fn():
 def eval_input_fn():
     dataset = tf.data.Dataset.from_tensor_slices((x_test, x_len_test, y_test))
     dataset = dataset.batch(100)
-    dataset = dataset.map(parser)
+    dataset = dataset.map(parser,num_parallel_calls=tf.data.experimental.AUTOTUNE)
     iterator = dataset.make_one_shot_iterator()
     return iterator.get_next()