summaryrefslogtreecommitdiffstats
path: root/data
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-04 22:06:05 +0100
committerVolpeon <git@volpeon.ink>2023-01-04 22:06:05 +0100
commita5e45e2c0dab95589e5fbaa4fe87d18484fbbe68 (patch)
tree8bd97a745e1113b1035c504ec484e099f878aed0 /data
parentVarious updates (diff)
downloadtextual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.tar.gz
textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.tar.bz2
textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.zip
Update
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 c505230..a60733a 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -151,7 +151,7 @@ class CSVDataModule():
151 151
152 num_images = len(items) 152 num_images = len(items)
153 153
154 valid_set_size = self.valid_set_size if self.valid_set_size is not None else int(num_images * 0.2) 154 valid_set_size = self.valid_set_size if self.valid_set_size is not None else num_images // 10
155 valid_set_size = max(valid_set_size, 1) 155 valid_set_size = max(valid_set_size, 1)
156 train_set_size = num_images - valid_set_size 156 train_set_size = num_images - valid_set_size
157 157