diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-04 22:06:05 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-04 22:06:05 +0100 |
| commit | a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68 (patch) | |
| tree | 8bd97a745e1113b1035c504ec484e099f878aed0 /data/csv.py | |
| parent | Various updates (diff) | |
| download | textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.tar.gz textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.tar.bz2 textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.zip | |
Update
Diffstat (limited to 'data/csv.py')
| -rw-r--r-- | data/csv.py | 2 |
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 | ||