1 files changed, 1 insertions, 1 deletions

diff --git a/data/csv.py b/data/csv.py
index 619452e..fba5d4b 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -282,7 +282,7 @@ class VlpnDataModule():
         collate_fn_ = partial(collate_fn, self.dtype, self.tokenizer, self.num_class_images != 0)
 
         if valid_set_size == 0:
-            data_train, data_val = items, items[:self.batch_size]
+            data_train, data_val = items, items
         else:
             data_train, data_val = random_split(items, [train_set_size, valid_set_size], generator=generator)