summaryrefslogtreecommitdiffstats
path: root/data
diff options
context:
space:
mode:
Diffstat (limited to 'data')
-rw-r--r--data/csv.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/data/csv.py b/data/csv.py
index 793fbf8..67ac43b 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -93,7 +93,7 @@ class CSVDataModule(pl.LightningDataModule):
93 items = [item for item in items if not "skip" in item or item["skip"] != True] 93 items = [item for item in items if not "skip" in item or item["skip"] != True]
94 num_images = len(items) 94 num_images = len(items)
95 95
96 valid_set_size = int(num_images * 0.2) 96 valid_set_size = int(num_images * 0.1)
97 if self.valid_set_size: 97 if self.valid_set_size:
98 valid_set_size = min(valid_set_size, self.valid_set_size) 98 valid_set_size = min(valid_set_size, self.valid_set_size)
99 valid_set_size = max(valid_set_size, 1) 99 valid_set_size = max(valid_set_size, 1)