From c6f6ae28ca1276dd0a1e437121b47b3c6302e0b0 Mon Sep 17 00:00:00 2001
From: mszmyd <erilf9@gmail.com>
Date: Sun, 5 May 2024 19:28:40 +0200
Subject: [PATCH] add batch

---
 dataset/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dataset/dataset.py b/dataset/dataset.py
index 0c9c035..e190384 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -34,6 +34,7 @@ class Dataset:
         self.dataset = self.__load_dataset()\
             .shuffle(self.shuffle_buffer_size, seed=self.seed)\
             .repeat(self.repeat)\
+            .batch(self.batch_size, drop_remainder=True)\
             .prefetch(tf.data.experimental.AUTOTUNE)
 
     def __load_dataset(self) -> tf.data.Dataset: