diff options
| author | Volpeon <git@volpeon.ink> | 2022-11-14 17:09:58 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-11-14 17:09:58 +0100 |
| commit | 2ad46871e2ead985445da2848a4eb7072b6e48aa (patch) | |
| tree | 3137923e2c00fe1d3cd37ddcc93c8a847b0c0762 /data | |
| parent | Update (diff) | |
| download | textual-inversion-diff-2ad46871e2ead985445da2848a4eb7072b6e48aa.tar.gz textual-inversion-diff-2ad46871e2ead985445da2848a4eb7072b6e48aa.tar.bz2 textual-inversion-diff-2ad46871e2ead985445da2848a4eb7072b6e48aa.zip | |
Update
Diffstat (limited to 'data')
| -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 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) |