1 files changed, 1 insertions, 1 deletions

diff --git a/data/csv.py b/data/csv.py
index c505230..a60733a 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -151,7 +151,7 @@ class CSVDataModule():
 
         num_images = len(items)
 
-        valid_set_size = self.valid_set_size if self.valid_set_size is not None else int(num_images * 0.2)
+        valid_set_size = self.valid_set_size if self.valid_set_size is not None else num_images // 10
         valid_set_size = max(valid_set_size, 1)
         train_set_size = num_images - valid_set_size