Reduce number of loaded batches

titu1994 · Jan 2, 2018 · 729d11a · 729d11a
1 parent 4b068cc
commit 729d11a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data_loader.py b/data_loader.py
@@ -67,7 +67,7 @@ def train_generator(batchsize):
 
         train_dataset = train_dataset.batch(batchsize)
         train_dataset = train_dataset.repeat()
-        train_dataset = train_dataset.shuffle(buffer_size=6)
+        train_dataset = train_dataset.shuffle(buffer_size=4)
         train_iterator = train_dataset.make_initializable_iterator()
 
         train_batch = train_iterator.get_next()
@@ -93,7 +93,7 @@ def val_generator(batchsize):
 
         val_dataset = val_dataset.batch(batchsize)
         val_dataset = val_dataset.repeat()
-        val_dataset = val_dataset.shuffle(buffer_size=6)
+        val_dataset = val_dataset.shuffle(buffer_size=4)
         val_iterator = val_dataset.make_initializable_iterator()
 
         val_batch = val_iterator.get_next()